@@ -15,6 +15,7 @@ from psycopg2._psycopg import (
15
15
ReplicationConnection as _replicationConnection ,
16
16
ReplicationCursor as _replicationCursor ,
17
17
ReplicationMessage as ReplicationMessage ,
18
+ _Vars ,
18
19
connection as _connection ,
19
20
cursor as _cursor ,
20
21
quote_ident as quote_ident ,
@@ -29,6 +30,7 @@ from psycopg2._range import (
29
30
RangeCaster as RangeCaster ,
30
31
register_range as register_range ,
31
32
)
33
+ from psycopg2 .sql import Composable
32
34
33
35
_T_cur = TypeVar ("_T_cur" , bound = _cursor )
34
36
@@ -106,7 +108,7 @@ class RealDictConnection(_connection):
106
108
class RealDictCursor (DictCursorBase ):
107
109
def __init__ (self , * args , ** kwargs ) -> None : ...
108
110
column_mapping : Any
109
- def execute (self , query , vars = None ): ...
111
+ def execute (self , query : str | bytes | Composable , vars : _Vars = None ) -> None : ...
110
112
def callproc (self , procname , vars = None ): ...
111
113
def fetchone (self ) -> RealDictRow | None : ... # type: ignore[override]
112
114
def fetchmany (self , size : int | None = None ) -> list [RealDictRow ]: ... # type: ignore[override]
0 commit comments