]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
register_array() renamed to array.register()
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Wed, 8 Apr 2020 09:59:47 +0000 (21:59 +1200)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Wed, 8 Apr 2020 09:59:47 +0000 (21:59 +1200)
psycopg3/types/array.py
psycopg3/types/composite.py
tests/types/test_array.py

index 13318cd1760f2ccab4ea731314386029dad1f389..ad4697c3eca614d8b6dcd7fee49957a7cc3d4ec2 100644 (file)
@@ -274,7 +274,7 @@ class ArrayCasterBinary(ArrayCasterBase):
         return agg(dims)
 
 
-def register_array(
+def register(
     array_oid: int,
     base_oid: int,
     context: AdaptContext = None,
@@ -287,7 +287,7 @@ def register_array(
         (Format.TEXT, ArrayCasterText),
         (Format.BINARY, ArrayCasterBinary),
     ):
-        tcname = f"{name}_array_{format.name.lower()}"
+        tcname = f"{name.title()}Array{format.name.title()}Caster"
         t = type(tcname, (base,), {"base_oid": base_oid})
         TypeCaster.register(array_oid, t, context=context, format=format)
 
@@ -304,4 +304,4 @@ def register_all_arrays() -> None:
             (t.oid, Format.TEXT) in TypeCaster.globals
             or (t.oid, Format.BINARY) in TypeCaster.globals
         ):
-            register_array(t.array_oid, t.oid, name=t.name)
+            register(t.array_oid, t.oid, name=t.name)
index f5d67db6f559ddd5f39f4b94231f0f05a516752b..d7e5161ab37cfac19b50b835ebdb503b621e9f91 100644 (file)
@@ -8,9 +8,9 @@ from collections import namedtuple
 from typing import Any, Callable, Generator, List, Sequence, Tuple, Union
 from typing import Optional, TYPE_CHECKING
 
+from . import array
 from ..adapt import Format, TypeCaster, Transformer, AdaptContext
 from .oids import builtins, TypeInfo
-from .array import register_array
 
 if TYPE_CHECKING:
     from ..connection import Connection
@@ -86,7 +86,7 @@ def register(
     )
 
     if info.array_oid:
-        register_array(
+        array.register(
             info.array_oid, info.oid, context=context, name=info.name
         )
 
index 4feb496be026252c295f0b82e5803b4747e680d6..4067d2f5359a8390e67836dbc7c726ba8df5cb8f 100644 (file)
@@ -1,8 +1,7 @@
 import pytest
 import psycopg3
-from psycopg3.types import builtins
+from psycopg3.types import builtins, array
 from psycopg3.adapt import Format, Transformer
-from psycopg3.types.array import register_array
 
 
 tests_str = [
@@ -107,7 +106,7 @@ def test_array_register(conn):
     res = cur.fetchone()[0]
     assert res == "{postgres=arwdDxt/postgres}"
 
-    register_array(
+    array.register(
         builtins["aclitem"].array_oid, builtins["aclitem"].oid, context=conn
     )
     cur.execute("select '{postgres=arwdDxt/postgres}'::aclitem[]")