Skip to content

Commit 8e25bda

Browse files
authored
psycopg2: Add types to RealDictCursor.execute (#14427)
1 parent 58b1130 commit 8e25bda

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

stubs/psycopg2/psycopg2/extras.pyi

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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):
106108
class 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

Comments
 (0)