dev

Author: zzzeek

sqlalchemy-stubs/engine/create.pyi

@@ -1,13 +1,16 @@
 from typing import Any
+from typing import Union
 
 from . import base as base
+from .base import Engine
 from .mock import create_mock_engine as create_mock_engine
+from .url import URL
 from .. import event as event
 from .. import exc as exc
 from .. import util as util
 from ..sql import compiler as compiler
 
-def create_engine(url: Any, **kwargs: Any): ...
+def create_engine(url: Union[str, URL], **kwargs: Any) -> Engine: ...
 def engine_from_config(
     configuration: Any, prefix: str = ..., **kwargs: Any
-): ...
+) -> Engine: ...

sqlalchemy-stubs/engine/url.pyi

@@ -59,4 +59,4 @@ class URL:
     def get_dialect(self): ...
     def translate_connect_args(self, names: Any = ..., **kw: Any): ...
 
-def make_url(name_or_url: Any): ...
+def make_url(name_or_url: Union[str, URL]) -> URL: ...

sqlalchemy-stubs/inspection.pyi

@@ -1,6 +1,23 @@
 from typing import Any
+from typing import overload
+from typing import Type
 
 from . import exc as exc
 from . import util as util
+from .engine.base import Connection
+from .engine.base import Engine
+from .engine.reflection import Inspector
+from .orm import InstanceState
+from .orm import Mapper
 
-def inspect(subject: Any, raiseerr: bool = ...): ...
+# this doesn't really seem to work...
+@overload
+def inspect(subject: Any, raiseerr: bool = ...) -> Any: ...
+@overload
+def inspect(subject: Engine, raiseerr: bool = ...) -> Inspector: ...
+@overload
+def inspect(subject: Connection, raiseerr: bool = ...) -> Inspector: ...
+@overload
+def inspect(subject: object, raiseerr: bool = ...) -> InstanceState: ...
+@overload
+def inspect(subject: Type, raiseerr: bool = ...) -> Mapper: ...