]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
refactor: use Self to represent the return from `BaseConnection._connect_gen()`
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Wed, 3 Jan 2024 00:59:30 +0000 (01:59 +0100)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Wed, 3 Jan 2024 03:07:17 +0000 (04:07 +0100)
psycopg/psycopg/_connection_base.py

index cf391998bd39ae16bb0900a67e65c7c9f2f5d869..90b0068388fe459986de3a191d8a74fc10c19d9c 100644 (file)
@@ -6,7 +6,7 @@ psycopg connection objects
 
 import logging
 from typing import Callable, Generic
-from typing import List, NamedTuple, Optional, Type, TypeVar, Tuple, Union
+from typing import List, NamedTuple, Optional, Tuple, TypeVar, Union
 from typing import TYPE_CHECKING
 from weakref import ref, ReferenceType
 from warnings import warn
@@ -17,13 +17,13 @@ from . import pq
 from . import errors as e
 from . import postgres
 from . import generators
-from .abc import ConnectionType, PQGen, PQGenConn, Query
+from .abc import PQGen, PQGenConn, Query
 from .sql import Composable, SQL
 from ._tpc import Xid
 from .rows import Row
 from .adapt import AdaptersMap
 from ._enums import IsolationLevel
-from ._compat import LiteralString
+from ._compat import LiteralString, Self
 from .pq.misc import connection_summary
 from .conninfo import ConnectionInfo
 from ._pipeline import BasePipeline
@@ -421,9 +421,9 @@ class BaseConnection(Generic[Row]):
 
     @classmethod
     def _connect_gen(
-        cls: Type[ConnectionType],
+        cls,
         conninfo: str = "",
-    ) -> PQGenConn[ConnectionType]:
+    ) -> PQGenConn[Self]:
         """Generator to connect to the database and create a new instance."""
         pgconn = yield from generators.connect(conninfo)
         conn = cls(pgconn)