dev

Author: zzzeek

sqlalchemy-stubs/orm/__init__.pyi

@@ -95,8 +95,8 @@ column_property = ColumnProperty
 composite = CompositeProperty
 
 def backref(name: Any, **kwargs: Any): ...
-def deferred(*columns: Any, **kw: Any): ...
-def query_expression(default_expr: Any = ...): ...
+def deferred(*columns: Any, **kw: Any) -> ColumnProperty: ...
+def query_expression(default_expr: Any = ...) -> ColumnProperty: ...
 
 mapper = Mapper
 synonym = SynonymProperty

sqlalchemy-stubs/orm/decl_api.pyi

@@ -21,6 +21,7 @@ class DeclarativeMeta(type):
     def __setattr__(cls, key: Any, value: Any) -> None: ...
     def __delattr__(cls, key: Any) -> None: ...
     metadata: MetaData
+    registry: "registry"
 
 def synonym_for(name: Any, map_column: bool = ...): ...
 

sqlalchemy-stubs/orm/relationships.pyi

@@ -1,5 +1,6 @@
 from typing import Any
 from typing import Optional
+from typing import Type
 from typing import Union
 
 from . import attributes as attributes
@@ -44,7 +45,7 @@ class RelationshipProperty(StrategizedProperty):
     sync_backref: Any = ...
     lazy: Any = ...
     single_parent: Any = ...
-    collection_class: Any = ...
+    collection_class: Optional[Type] = ...
     passive_deletes: Any = ...
     cascade_backrefs: Any = ...
     passive_updates: Any = ...

sqlalchemy-stubs/sql/functions.pyi

@@ -96,8 +96,8 @@ class _FunctionGenerator:
     def __getattr__(self, name: Any): ...
     def __call__(self, *c: Any, **kwargs: Any): ...
 
-func: Any
-modifier: Any
+func: _FunctionGenerator = ...
+modifier: _FunctionGenerator = ...
 
 class Function(FunctionElement):
     __visit_name__: str = ...