@@ -2,9 +2,9 @@ import itertools
2
2
import logging
3
3
import threading
4
4
from _typeshed import Incomplete , SupportsKeysAndGetItem
5
- from collections .abc import Generator , Iterable
5
+ from collections .abc import Callable , Generator , Iterable
6
6
from types import TracebackType
7
- from typing import ClassVar , NamedTuple , TypeVar
7
+ from typing import Any , ClassVar , NamedTuple , TypeVar
8
8
from typing_extensions import Self
9
9
10
10
class NullHandler (logging .Handler ):
@@ -18,6 +18,7 @@ long = int
18
18
izip_longest = itertools .zip_longest
19
19
20
20
_VT = TypeVar ("_VT" )
21
+ _F = TypeVar ("_F" , bound = Callable [..., Any ])
21
22
22
23
class attrdict (dict [str , _VT ]):
23
24
def __getattr__ (self , attr : str ) -> _VT : ...
@@ -116,9 +117,10 @@ class Node:
116
117
def unwrap (self ): ...
117
118
118
119
class ColumnFactory :
119
- node : Incomplete
120
- def __init__ (self , node ) -> None : ...
121
- def __getattr__ (self , attr : str ): ...
120
+ node : Node
121
+ def __init__ (self , node : Node ) -> None : ...
122
+ def __getattr__ (self , attr : str ) -> Column : ...
123
+ def __getitem__ (self , attr : str ) -> Column : ...
122
124
123
125
class _DynamicColumn :
124
126
def __get__ (self , instance , instance_type : Incomplete | None = ...): ...
@@ -880,8 +882,8 @@ class SqliteDatabase(Database):
880
882
def aggregate (self , name : Incomplete | None = ..., num_params : int = ...): ...
881
883
def register_collation (self , fn , name : Incomplete | None = ...): ...
882
884
def collation (self , name : Incomplete | None = ...): ...
883
- def register_function (self , fn , name : Incomplete | None = ... , num_params : int = ... ) -> None : ...
884
- def func (self , name : Incomplete | None = ... , num_params : int = ...) : ...
885
+ def register_function (self , fn , name : str | None = None , num_params : int = - 1 , deterministic : bool | None = None ) -> None : ...
886
+ def func (self , name : str | None = None , num_params : int = - 1 , deterministic : bool | None = None ) -> Callable [[ _F ], _F ] : ...
885
887
def register_window_function (self , klass , name : Incomplete | None = ..., num_params : int = ...) -> None : ...
886
888
def window_function (self , name : Incomplete | None = ..., num_params : int = ...): ...
887
889
def register_table_function (self , klass , name : Incomplete | None = ...) -> None : ...
@@ -934,6 +936,7 @@ class PostgresqlDatabase(Database):
934
936
def is_connection_usable (self ): ...
935
937
def last_insert_id (self , cursor , query_type : Incomplete | None = ...): ...
936
938
def rows_affected (self , cursor ): ...
939
+ def begin (self , isolation_level : str | None = None ) -> None : ...
937
940
def get_tables (self , schema : Incomplete | None = ...): ...
938
941
def get_views (self , schema : Incomplete | None = ...): ...
939
942
def get_indexes (self , table , schema : Incomplete | None = ...): ...
@@ -967,6 +970,7 @@ class MySQLDatabase(Database):
967
970
def init (self , database , ** kwargs ) -> None : ...
968
971
def is_connection_usable (self ): ...
969
972
def default_values_insert (self , ctx ): ...
973
+ def begin (self , isolation_level : str | None = None ) -> None : ...
970
974
def get_tables (self , schema : Incomplete | None = ...): ...
971
975
def get_views (self , schema : Incomplete | None = ...): ...
972
976
def get_indexes (self , table , schema : Incomplete | None = ...): ...
0 commit comments