Annotate Error and Diagnostics (#7671)

Move cursor class to top of file so it can be used as base class

Author: srittau

stubs/psycopg2/psycopg2/_psycopg.pyi

@@ -68,6 +68,52 @@ threadsafety: int
 
 __libpq_version__: int
 
+class cursor:
+    arraysize: int
+    binary_types: Any
+    closed: Any
+    connection: Any
+    description: Any
+    itersize: Any
+    lastrowid: Any
+    name: Any
+    pgresult_ptr: Any
+    query: Any
+    row_factory: Any
+    rowcount: int
+    rownumber: int
+    scrollable: Any
+    statusmessage: Any
+    string_types: Any
+    typecaster: Any
+    tzinfo_factory: Any
+    withhold: Any
+    def __init__(self, *args, **kwargs) -> None: ...
+    def callproc(self, procname, parameters=...): ...
+    def cast(self, oid, s): ...
+    def close(self): ...
+    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 | 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, ...] | None: ...
+    def mogrify(self, *args, **kwargs): ...
+    def nextset(self): ...
+    def scroll(self, value, mode=...): ...
+    def setinputsizes(self, sizes): ...
+    def setoutputsize(self, size, column=...): ...
+    def __enter__(self: Self) -> Self: ...
+    def __exit__(
+        self, type: type[BaseException] | None, value: BaseException | None, traceback: TracebackType | None
+    ) -> None: ...
+    def __iter__(self: Self) -> Self: ...
+    def __next__(self) -> tuple[Any, ...]: ...
+
+_cursor: TypeAlias = cursor
+
 class AsIs:
     adapted: Any
     def __init__(self, *args, **kwargs) -> None: ...
@@ -143,31 +189,31 @@ class Decimal:
     def __conform__(self, *args, **kwargs): ...
 
 class Diagnostics:
-    column_name: Any
-    constraint_name: Any
-    context: Any
-    datatype_name: Any
-    internal_position: Any
-    internal_query: Any
-    message_detail: Any
-    message_hint: Any
-    message_primary: Any
-    schema_name: Any
-    severity: Any
-    severity_nonlocalized: Any
-    source_file: Any
-    source_function: Any
-    source_line: Any
-    sqlstate: Any
-    statement_position: Any
-    table_name: Any
-    def __init__(self, *args, **kwargs) -> None: ...
+    column_name: str | None
+    constraint_name: str | None
+    context: str | None
+    datatype_name: str | None
+    internal_position: str | None
+    internal_query: str | None
+    message_detail: str | None
+    message_hint: str | None
+    message_primary: str | None
+    schema_name: str | None
+    severity: str | None
+    severity_nonlocalized: str | None
+    source_file: str | None
+    source_function: str | None
+    source_line: str | None
+    sqlstate: str | None
+    statement_position: str | None
+    table_name: str | None
+    def __init__(self, __err: Error) -> None: ...
 
 class Error(Exception):
-    cursor: Any
-    diag: Any
-    pgcode: Any
-    pgerror: Any
+    cursor: _cursor | None
+    diag: Diagnostics
+    pgcode: str | None
+    pgerror: str | None
     def __init__(self, *args, **kwargs) -> None: ...
     def __reduce__(self): ...
     def __setstate__(self, state): ...
@@ -241,7 +287,7 @@ class ReplicationConnection(psycopg2.extensions.connection):
     set_session: Any
     def __init__(self, *args, **kwargs) -> None: ...
 
-class ReplicationCursor(psycopg2.extensions.cursor):
+class ReplicationCursor(cursor):
     feedback_timestamp: Any
     io_timestamp: Any
     wal_end: Any
@@ -275,7 +321,6 @@ class Xid:
     def __getitem__(self, __index): ...
     def __len__(self): ...
 
-_cursor: TypeAlias = cursor
 _T_cur = TypeVar("_T_cur", bound=_cursor)
 
 class connection:
@@ -338,50 +383,6 @@ class connection:
     def __enter__(self): ...
     def __exit__(self, type, value, traceback): ...
 
-class cursor:
-    arraysize: int
-    binary_types: Any
-    closed: Any
-    connection: Any
-    description: Any
-    itersize: Any
-    lastrowid: Any
-    name: Any
-    pgresult_ptr: Any
-    query: Any
-    row_factory: Any
-    rowcount: int
-    rownumber: int
-    scrollable: Any
-    statusmessage: Any
-    string_types: Any
-    typecaster: Any
-    tzinfo_factory: Any
-    withhold: Any
-    def __init__(self, *args, **kwargs) -> None: ...
-    def callproc(self, procname, parameters=...): ...
-    def cast(self, oid, s): ...
-    def close(self): ...
-    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 | 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, ...] | None: ...
-    def mogrify(self, *args, **kwargs): ...
-    def nextset(self): ...
-    def scroll(self, value, mode=...): ...
-    def setinputsizes(self, sizes): ...
-    def setoutputsize(self, size, column=...): ...
-    def __enter__(self: Self) -> Self: ...
-    def __exit__(
-        self, type: type[BaseException] | None, value: BaseException | None, traceback: TracebackType | None
-    ) -> None: ...
-    def __iter__(self: Self) -> Self: ...
-    def __next__(self) -> tuple[Any, ...]: ...
-
 class lobject:
     closed: Any
     mode: Any