pgdevkit 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

pgdevkit/db/complex_types.py

@@ -9,7 +9,18 @@ from psycopg.types.composite import CompositeInfo, register_composite
 from psycopg.types.enum import EnumInfo, register_enum
 from psycopg.types.json import Jsonb
 
-ComplexTypeInfo = CompositeInfo | EnumInfo | type[Jsonb] | None
+
+class JsonbArray:
+    """Sentinel for a `jsonb[]` column (a Postgres ARRAY of jsonb) — distinct
+    from `Jsonb` (a scalar jsonb column) so `recursive_convert` knows whether
+    an incoming Python list *is* the array (each element needs its own
+    `Jsonb(...)` wrapper) or is JSON array *content* for one scalar jsonb
+    value (the whole list gets wrapped once). Both cases arrive from
+    `information_schema.columns` looking identical (a plain Python list) —
+    only the column's own array-ness disambiguates them."""
+
+
+ComplexTypeInfo = CompositeInfo | EnumInfo | type[Jsonb] | type[JsonbArray] | None
 
 
 class ComplexHelper:
@@ -67,7 +78,7 @@ class ComplexHelper:
         if res["data_type"].lower() == "jsonb":
             return Jsonb
         if res["data_type"].upper() == "ARRAY" and res["udt_name"] == "_jsonb":
-            return Jsonb
+            return JsonbArray
         if (
             not res["is_enum"]
             and not res["is_user_defined"]
@@ -181,9 +192,16 @@ class ComplexHelper:
             return None
         if self.system_complex_type_dict is None:
             await self.load_complex_type_dict()
+        if info == JsonbArray:
+            # The value here *is* the jsonb[] array -- each element is its
+            # own scalar jsonb value, unlike the plain-Jsonb case below.
+            assert isinstance(value, list), f"Expected a list for a jsonb[] column, got {type(value)}"
+            return [Jsonb(item) for item in value]
         if info == Jsonb:
-            # A JSONB column's value is wrapped whole, even if it's a list —
-            # only an array-of-composite/enum column recurses per element.
+            # A scalar JSONB column's value is wrapped whole, even if it's a
+            # list — that list is JSON array *content* for the one jsonb
+            # value, not multiple array elements (see JsonbArray above for
+            # the jsonb[] column case).
             return Jsonb(value)
         if isinstance(value, list):
             return [await self.recursive_convert(item, info, con) for item in value]

{pgdevkit-0.2.3.dist-info → pgdevkit-0.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pgdevkit
-Version: 0.2.3
+Version: 0.2.4
 Summary: A helper for developing with Postgres
 Requires-Python: >=3.14
 Requires-Dist: docker>=7.1.0

{pgdevkit-0.2.3.dist-info → pgdevkit-0.2.4.dist-info}/RECORD

@@ -8,7 +8,7 @@ pgdevkit/lakebase.py,sha256=0PABOk4p-2klLTY8QAUibQf9QeyNHKywUWEWGdkUY8k,3165
 pgdevkit/models.py,sha256=dAZ8f7pgDsxoLVrDWsM4h0qwJ0pOzXDdWMGsnjrW80E,2103
 pgdevkit/parser.py,sha256=B3NRtcSBuEUcQNp5lxEMi6ApVedkbNUNdklCea-Ts40,12021
 pgdevkit/db/__init__.py,sha256=PlnMycSmQVqnzHue_uLQZuikbW6QAqs6p91IO5l6nfU,763
-pgdevkit/db/complex_types.py,sha256=t1y5_yIGLUtI_iuBUr7Es7e83OD9YVyvzInwLyaMfcM,10197
+pgdevkit/db/complex_types.py,sha256=bv3oMztSnrhz4-OJROBXfiouL2jJ8TxpSR_83HMLmTI,11197
 pgdevkit/db/connection.py,sha256=ylc_KVYsOvPeuW5GL1ebDDH0O3w93oe-EHwkUH4KoVs,5793
 pgdevkit/db/crud.py,sha256=RTRhum5GyfeRAw8p8kk_kFr9uKCeWe8wwrUZYLvzdZU,12103
 pgdevkit/db/loader.py,sha256=0Swmp0h0s7I4acG0Dv3cem2YD4lFvq8aFEbFQPPQrxg,616
@@ -25,7 +25,7 @@ pgdevkit/docs/database-layout.md,sha256=dNvDH956xSXuAnLqkOopyOfaIm_A-RtUd57VTaZ1
 pgdevkit/skills/pgdevkit/SKILL.md,sha256=6uxe6BQwOLexGiBpf00VTFTHOiWJo5k2Qqzu172rb4I,12156
 pgdevkit/skills/pgdevkit/references/dynamic-sql.md,sha256=aTR1sx81moiTwaMaiOecaqk-9rT8y5Zxz7W1BoC3vK0,2774
 pgdevkit/skills/pgdevkit/references/temporal-tables.md,sha256=LwZnItvQWI7bB0oRHxIfUc0ynnq7N0V5Mnv-QHvFN3Q,1364
-pgdevkit-0.2.3.dist-info/METADATA,sha256=TLAMgx_QXdQmoqQyalKNEQcwd9VupvVH3ZPn4e3BZQk,5915
-pgdevkit-0.2.3.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
-pgdevkit-0.2.3.dist-info/entry_points.txt,sha256=FK2P55lf0WaGlOZJIINOGgCt8WB7ep0OnPsgnrp80bg,42
-pgdevkit-0.2.3.dist-info/RECORD,,
+pgdevkit-0.2.4.dist-info/METADATA,sha256=vM_8DEMsPyq7CSplDGBhDdWpPJLD_wrMNJk4GSvWRVA,5915
+pgdevkit-0.2.4.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+pgdevkit-0.2.4.dist-info/entry_points.txt,sha256=FK2P55lf0WaGlOZJIINOGgCt8WB7ep0OnPsgnrp80bg,42
+pgdevkit-0.2.4.dist-info/RECORD,,