@@ -14,6 +14,7 @@ def callable_(c: object) -> TypeIs[Callable[..., object]]: ...
1414multi_types : tuple [type [Incomplete ], ...]
1515
1616_T = TypeVar ("_T" )
17+ _R = TypeVar ("_R" )
1718_VT = TypeVar ("_VT" )
1819_F = TypeVar ("_F" , bound = Callable [..., Any ])
1920_Model : TypeAlias = Model
@@ -54,6 +55,11 @@ class _FKKwargs(_FieldKwargs, total=False):
5455 related_name : str | Callable [[Field [Any ]], str ] | None
5556 rel_model : type [Model ] | None
5657
58+ class classmethod_only (Generic [_T , _R ]):
59+ fn : Callable [[type [_T ]], _R ]
60+ def __init__ (self , fn : Callable [[type [_T ]], _R ]) -> None : ...
61+ def __get__ (self , instance : None , instance_type : type [_T ] | None = None ) -> Callable [[], _R ]: ...
62+
5763class attrdict (dict [str , _VT ]):
5864 def __getattr__ (self , attr : str ) -> _VT : ...
5965 def __setattr__ (self , attr : str , value : _VT ) -> None : ...
@@ -980,8 +986,6 @@ class _BasePsycopgAdapter:
980986 @overload
981987 def isolation_level_str (self , isolation_level : _T ) -> _T : ...
982988
983- def server_side_cursor (self , conn ): ...
984-
985989class Psycopg2Adapter (_BasePsycopgAdapter ):
986990 json_type : Incomplete
987991 jsonb_type : Incomplete
@@ -994,6 +998,7 @@ class Psycopg2Adapter(_BasePsycopgAdapter):
994998 def is_connection_usable (self , conn ) -> bool : ...
995999 def is_connection_reusable (self , conn ) -> bool : ...
9961000 def is_connection_closed (self , conn ) -> bool : ...
1001+ def server_side_cursor (self , conn ): ...
9971002
9981003class Psycopg3Adapter (_BasePsycopgAdapter ):
9991004 json_type : Incomplete
@@ -1007,6 +1012,7 @@ class Psycopg3Adapter(_BasePsycopgAdapter):
10071012 def is_connection_usable (self , conn ) -> bool : ...
10081013 def is_connection_reusable (self , conn ) -> bool : ...
10091014 def is_connection_closed (self , conn ) -> bool : ...
1015+ def server_side_cursor (self , conn ): ...
10101016
10111017class PostgresqlDatabase (Database ):
10121018 field_types : Incomplete
@@ -1029,6 +1035,8 @@ class PostgresqlDatabase(Database):
10291035 prefer_psycopg3 : bool = False ,
10301036 ** kwargs ,
10311037 ) -> None : ...
1038+ @property
1039+ def index_value_literals (self ) -> bool : ... # type: ignore[override]
10321040 def is_connection_usable (self ) -> bool : ...
10331041 def begin (self , isolation_level : str | None = None ) -> None : ...
10341042 def get_tables (self , schema : str | None = None ) -> list [str ]: ...
@@ -1816,6 +1824,7 @@ class Metadata:
18161824 def fields_to_index (self ) -> list [Incomplete ]: ...
18171825 def set_database (self , database : _DatabaseType ) -> None : ...
18181826 def set_table_name (self , table_name ) -> None : ...
1827+ def get_database_instance (self ) -> _DatabaseType | None : ...
18191828
18201829class SubclassAwareMetadata (Metadata ):
18211830 models : Incomplete
@@ -1872,8 +1881,7 @@ class Model(metaclass=ModelBase):
18721881 def replace_many (cls , rows , fields = None ): ...
18731882 @classmethod
18741883 def raw (cls , sql , * params ) -> ModelRaw : ...
1875- @classmethod
1876- def delete (cls ) -> ModelDelete : ...
1884+ delete : classmethod_only [Self , ModelDelete ]
18771885 @classmethod
18781886 def create (cls , ** query ) -> Self : ...
18791887 @classmethod
0 commit comments