from typing import reveal_type, Optional, Self
from types import TracebackType
from contextlib import AbstractContextManager

class A1(AbstractContextManager):
    def __exit__(  # pylint: disable=useless-return
            self, exc_type: Optional[type[BaseException]],
            exc_val: Optional[BaseException],
            exc_tb: Optional[TracebackType]) -> Optional[bool]:
        return None

class A2(AbstractContextManager):
    def __exit__(  # pylint: disable=useless-return
            self, exc_type: Optional[type[BaseException]],
            exc_val: Optional[BaseException],
            exc_tb: Optional[TracebackType]) -> Optional[bool]:
        return None

    def __enter__(self) -> Self:
        return self

class B1(A1):
    pass

class B2(A2):
    pass

with B1() as d1:
    reveal_type(d1)

with B2() as d2:
    reveal_type(d2)