From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Wed, 3 Jan 2024 00:59:30 +0000 (+0100)
Subject: refactor: use Self to represent the return from `BaseConnection._connect_gen()`
X-Git-Tag: 3.2.0~107
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=4324a38be33b0420fd4ed661e3f9dc86efdedd94;p=thirdparty%2Fpsycopg.git

refactor: use Self to represent the return from `BaseConnection._connect_gen()`
---

diff --git a/psycopg/psycopg/_connection_base.py b/psycopg/psycopg/_connection_base.py
index cf391998b..90b006838 100644
--- a/psycopg/psycopg/_connection_base.py
+++ b/psycopg/psycopg/_connection_base.py
@@ -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)