bad-exit-annotation (PYI036)

Derived from the flake8-pyi linter.

Fix is sometimes available.

What it does

Checks for incorrect function signatures on __exit__ and __aexit__ methods.

Why is this bad?

Improperly-annotated __exit__ and __aexit__ methods can cause unexpected behavior when interacting with type checkers.

Example

class Foo:
    def __exit__(self, typ, exc, tb, extra_arg) -> None:
        ...

Use instead:

class Foo:
    def __exit__(
        self,
        typ: type[BaseException] | None,
        exc: BaseException | None,
        tb: TracebackType | None,
        extra_arg: int = 0,
    ) -> None:
        ...