@@ -15,6 +15,7 @@ from psycopg2._psycopg import (
1515 ReplicationConnection as _replicationConnection ,
1616 ReplicationCursor as _replicationCursor ,
1717 ReplicationMessage as ReplicationMessage ,
18+ _Vars ,
1819 connection as _connection ,
1920 cursor as _cursor ,
2021 quote_ident as quote_ident ,
@@ -29,6 +30,7 @@ from psycopg2._range import (
2930 RangeCaster as RangeCaster ,
3031 register_range as register_range ,
3132)
33+ from psycopg2 .sql import Composable
3234
3335_T_cur = TypeVar ("_T_cur" , bound = _cursor )
3436
@@ -106,7 +108,7 @@ class RealDictConnection(_connection):
106108class RealDictCursor (DictCursorBase ):
107109 def __init__ (self , * args , ** kwargs ) -> None : ...
108110 column_mapping : Any
109- def execute (self , query , vars = None ): ...
111+ def execute (self , query : str | bytes | Composable , vars : _Vars = None ) -> None : ...
110112 def callproc (self , procname , vars = None ): ...
111113 def fetchone (self ) -> RealDictRow | None : ... # type: ignore[override]
112114 def fetchmany (self , size : int | None = None ) -> list [RealDictRow ]: ... # type: ignore[override]
0 commit comments