Add missing symbols for .all()

Author: zzzeek

sqlalchemy-stubs/engine/__init__.pyi

@@ -1,3 +1,52 @@
-from .create import create_engine as create_engine
-from .create import engine_from_config as engine_from_config
+from . import events as events, util as util
+from ..sql import ddl as ddl
+from .base import (
+    Connection as Connection,
+    Engine as Engine,
+    NestedTransaction as NestedTransaction,
+    RootTransaction as RootTransaction,
+    Transaction as Transaction,
+    TwoPhaseTransaction as TwoPhaseTransaction,
+)
+from .create import (
+    create_engine as create_engine,
+    engine_from_config as engine_from_config,
+)
+from .cursor import (
+    BaseCursorResult as BaseCursorResult,
+    BufferedColumnResultProxy as BufferedColumnResultProxy,
+    BufferedColumnRow as BufferedColumnRow,
+    BufferedRowResultProxy as BufferedRowResultProxy,
+    CursorResult as CursorResult,
+    FullyBufferedResultProxy as FullyBufferedResultProxy,
+    LegacyCursorResult as LegacyCursorResult,
+)
+from .interfaces import (
+    Compiled as Compiled,
+    Connectable as Connectable,
+    CreateEnginePlugin as CreateEnginePlugin,
+    Dialect as Dialect,
+    ExceptionContext as ExceptionContext,
+    ExecutionContext as ExecutionContext,
+    TypeCompiler as TypeCompiler,
+)
 from .mock import create_mock_engine as create_mock_engine
+from .reflection import Inspector as Inspector
+from .result import (
+    ChunkedIteratorResult as ChunkedIteratorResult,
+    FrozenResult as FrozenResult,
+    IteratorResult as IteratorResult,
+    MappingResult as MappingResult,
+    MergedResult as MergedResult,
+    Result as Result,
+    ScalarResult as ScalarResult,
+    result_tuple as result_tuple,
+)
+from .row import (
+    BaseRow as BaseRow,
+    LegacyRow as LegacyRow,
+    Row as Row,
+    RowMapping as RowMapping,
+)
+from .url import URL as URL, make_url as make_url
+from .util import connection_memoize as connection_memoize

sqlalchemy-stubs/orm/query.pyi

@@ -1,5 +1,6 @@
 from typing import Any
 from typing import Optional
+from typing import List
 
 from . import interfaces
 from .context import QueryContext as QueryContext
@@ -113,7 +114,7 @@ class Query(
     def limit(self, limit: Any) -> None: ...
     def offset(self, offset: Any) -> None: ...
     def distinct(self, *expr: Any) -> None: ...
-    def all(self): ...
+    def all(self) -> List[Any]: ...
     def from_statement(self, statement: Any) -> None: ...
     def first(self): ...
     def one_or_none(self): ...

sqlalchemy-stubs/orm/session.pyi

@@ -1,5 +1,6 @@
 from .. import util
 from typing import Any, Optional
+from ..engine import Result
 
 class _SessionClassMethods:
     @classmethod
@@ -151,15 +152,15 @@ class Session(_SessionClassMethods):
         _parent_execute_state: Optional[Any] = ...,
         _add_event: Optional[Any] = ...,
         **kw: Any,
-    ): ...
+    ) -> Result: ...
     def scalar(
         self,
         statement: Any,
         params: Optional[Any] = ...,
         execution_options: Any = ...,
         bind_arguments: Optional[Any] = ...,
         **kw: Any,
-    ): ...
+    ) -> Any: ...
     def close(self) -> None: ...
     def invalidate(self) -> None: ...
     def expunge_all(self) -> None: ...