refactor: add NoneType to abc module

diff --git a/psycopg/psycopg/_transform.py b/psycopg/psycopg/_transform.py
index 93a4435d93c1455df096bf682e8c423713d28bb4..44f6b93bac2c2429f739ca478066d922b169c1fa 100644 (file)
--- a/psycopg/psycopg/_transform.py
+++ b/psycopg/psycopg/_transform.py
@@ -5,13 +5,13 @@ Helper object to transform values between Python and PostgreSQL
 # Copyright (C) 2020 The Psycopg Team
 
 from typing import Any, Dict, List, Optional, Sequence, Tuple
-from typing import DefaultDict, Type, TYPE_CHECKING
+from typing import DefaultDict, TYPE_CHECKING
 from collections import defaultdict
 
 from . import pq
 from . import postgres
 from . import errors as e
-from .abc import Buffer, LoadFunc, AdaptContext, PyFormat, DumperKey
+from .abc import Buffer, LoadFunc, AdaptContext, PyFormat, DumperKey, NoneType
 from .rows import Row, RowMaker
 from ._compat import TypeAlias
 from .postgres import INVALID_OID, TEXT_OID
@@ -23,7 +23,6 @@ if TYPE_CHECKING:
     from .pq.abc import PGresult
     from .connection import BaseConnection
 
-NoneType: Type[None] = type(None)
 DumperCache: TypeAlias = Dict[DumperKey, "Dumper"]
 OidDumperCache: TypeAlias = Dict[int, "Dumper"]
 LoaderCache: TypeAlias = Dict[int, "Loader"]

diff --git a/psycopg/psycopg/abc.py b/psycopg/psycopg/abc.py
index 0da76e4d08f4a2385ddd49664bc2045e11c08adc..06c7b69db2e756399406c787ed0df1ec04c3660f 100644 (file)
--- a/psycopg/psycopg/abc.py
+++ b/psycopg/psycopg/abc.py
@@ -20,6 +20,8 @@ if TYPE_CHECKING:
     from .connection import BaseConnection
     from ._adapters_map import AdaptersMap
 
+NoneType: type = type(None)
+
 # An object implementing the buffer protocol
 Buffer: TypeAlias = Union[bytes, bytearray, memoryview]
 

diff --git a/psycopg/psycopg/crdb.py b/psycopg/psycopg/crdb.py
index 2b0e60cd91ee264b157ac592426c37c69e91ab8b..cf61d7a72751b6efaee1b7b6be157324840230b6 100644 (file)
--- a/psycopg/psycopg/crdb.py
+++ b/psycopg/psycopg/crdb.py
@@ -10,7 +10,7 @@ from typing import Any, Optional, Type, Union, overload, TYPE_CHECKING
 from ._typeinfo import TypeInfo, TypesRegistry
 
 from . import errors as e
-from .abc import AdaptContext
+from .abc import AdaptContext, NoneType
 from .rows import Row, RowFactory, AsyncRowFactory, TupleRow
 from .postgres import TEXT_OID
 from .conninfo import ConnectionInfo
@@ -263,7 +263,7 @@ def register_crdb_net_adapters(context: AdaptContext) -> None:
 
 
 def register_crdb_none_adapters(context: AdaptContext) -> None:
-    context.adapters.register_dumper(type(None), CrdbNoneDumper)
+    context.adapters.register_dumper(NoneType, CrdbNoneDumper)
 
 
 for t in [

diff --git a/psycopg/psycopg/types/array.py b/psycopg/psycopg/types/array.py
index ef9429b1575ed0fada12c2251326bb3a8164f10e..569e61f66d0e07bbf94478f31d2ae84951fdd97b 100644 (file)
--- a/psycopg/psycopg/types/array.py
+++ b/psycopg/psycopg/types/array.py
@@ -13,7 +13,7 @@ from functools import lru_cache
 from .. import pq
 from .. import errors as e
 from .. import postgres
-from ..abc import AdaptContext, Buffer, Dumper, DumperKey
+from ..abc import AdaptContext, Buffer, Dumper, DumperKey, NoneType
 from ..adapt import RecursiveDumper, RecursiveLoader, PyFormat
 from .._struct import pack_len, unpack_len
 from ..postgres import TEXT_OID, INVALID_OID
@@ -27,7 +27,6 @@ _pack_dim = cast(Callable[[int, int], bytes], _struct_dim.pack)
 _unpack_dim = cast(Callable[[bytes, int], Tuple[int, int]], _struct_dim.unpack_from)
 
 TEXT_ARRAY_OID = postgres.types["text"].array_oid
-NoneType: type = type(None)
 
 
 class BaseListDumper(RecursiveDumper):

diff --git a/psycopg/psycopg/types/none.py b/psycopg/psycopg/types/none.py
index 6a80fab37069cb8554047721bd2a3d0bc203d524..2ab857ccabf0ead816ff115b206a3a56dd8ce4cf 100644 (file)
--- a/psycopg/psycopg/types/none.py
+++ b/psycopg/psycopg/types/none.py
@@ -4,7 +4,7 @@ Adapters for None.
 
 # Copyright (C) 2020 The Psycopg Team
 
-from ..abc import AdaptContext
+from ..abc import AdaptContext, NoneType
 from ..adapt import Dumper
 
 
@@ -22,4 +22,4 @@ class NoneDumper(Dumper):
 
 
 def register_default_adapters(context: AdaptContext) -> None:
-    context.adapters.register_dumper(type(None), NoneDumper)
+    context.adapters.register_dumper(NoneType, NoneDumper)