fix issue with _AsyncSessionProtocol (#157)

Author: CaselIT

sqlalchemy-stubs/ext/asyncio/session.pyi

@@ -134,7 +134,8 @@ class _AsyncSessionProtocol(
     async def close_all(cls) -> None: ...  # NOTE: Deprecated.
 
 class _AsyncSessionTypingCommon(
-    _SessionNoIoTypingCommon, _SessionClassMethodNoIoTypingCommon
+    _SessionNoIoTypingCommon[Union[AsyncConnection, AsyncEngine]],
+    _SessionClassMethodNoIoTypingCommon,
 ):
     bind: Any = ...
     def begin(self, **kw: Any) -> AsyncSessionTransaction: ...

sqlalchemy-stubs/orm/session.pyi

@@ -290,7 +290,7 @@ class SessionTransaction:
     def __enter__(self: _TSessionTransaction) -> _TSessionTransaction: ...
     def __exit__(self, type_: Any, value: Any, traceback: Any) -> None: ...
 
-class _SessionNoIoTypingCommon:
+class _SessionNoIoTypingCommon(Generic[_T]):
     @property
     def dirty(self) -> util.IdentitySet[Any]: ...
     @property
@@ -321,15 +321,17 @@ class _SessionNoIoTypingCommon:
         self,
         mapper: Optional[Any] = ...,
         clause: Optional[ClauseElement] = ...,
-        bind: Optional[Union[Connection, Engine]] = ...,
+        bind: Optional[_T] = ...,
         _sa_skip_events: Optional[Any] = ...,
         _sa_skip_for_implicit_returning: bool = ...,
-    ) -> Union[Connection, Engine]: ...
+    ) -> _T: ...
     def is_modified(
         self, instance: Any, include_collections: bool = ...
     ) -> bool: ...
 
-class _SessionTypingCommon(_SessionNoIoTypingCommon):
+class _SessionTypingCommon(
+    _SessionNoIoTypingCommon[Union[Connection, Engine]]
+):
     bind: Optional[Union[Connection, Engine]]
     autocommit: bool
     def begin(

test/files/async_stuff.py

@@ -0,0 +1,7 @@
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.ext.asyncio import create_async_engine
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.orm import Session
+
+engine = create_async_engine(...)
+async_session = sessionmaker(engine, class_=AsyncSession)