diff --git a/stubs/psycopg2/psycopg2/_psycopg.pyi b/stubs/psycopg2/psycopg2/_psycopg.pyi index b4983f65d..02c018b13 100644 --- a/stubs/psycopg2/psycopg2/_psycopg.pyi +++ b/stubs/psycopg2/psycopg2/_psycopg.pyi @@ -5,6 +5,7 @@ from typing import Any, TypeVar, overload import psycopg2 import psycopg2.extensions +from psycopg2.sql import Composable _Vars = Sequence[Any] | Mapping[str, Any] | None @@ -360,11 +361,11 @@ class cursor: def callproc(self, procname, parameters=...): ... def cast(self, oid, s): ... def close(self): ... - def copy_expert(self, sql, file, size=...): ... + def copy_expert(self, sql: str | bytes | Composable, file, size=...): ... def copy_from(self, file, table, sep=..., null=..., size=..., columns=...): ... def copy_to(self, file, table, sep=..., null=..., columns=...): ... - def execute(self, query: str | bytes, vars: _Vars = ...) -> None: ... - def executemany(self, query: str | bytes, vars_list: Iterable[_Vars]) -> None: ... + def execute(self, query: str | bytes | Composable, vars: _Vars = ...) -> None: ... + def executemany(self, query: str | bytes | Composable, vars_list: Iterable[_Vars]) -> None: ... def fetchall(self) -> list[tuple[Any, ...]]: ... def fetchmany(self, size=...) -> list[tuple[Any, ...]]: ... def fetchone(self) -> tuple[Any, ...] | Any: ...