database-wrapper-pgsql 0.2.7__py3-none-any.whl → 0.2.9__py3-none-any.whl

database_wrapper_pgsql/__init__.py

@@ -24,6 +24,7 @@ from .connector import (
     PgConnectionTypeAsync,
     PgCursorTypeAsync,
 )
+from .pg_introspector import PostgresIntrospector
 
 # Set the logger to a quiet default, can be enabled if needed
 logger = logging.getLogger("database_wrapper_pgsql")
@@ -47,4 +48,5 @@ __all__ = [
     "PgCursorTypeAsync",
     # Helpers
     "PgConfig",
+    "PostgresIntrospector",
 ]

database_wrapper_pgsql/pg_introspector.py

@@ -0,0 +1,72 @@
+import psycopg
+
+from database_wrapper import ColumnMetaIntrospector, DBIntrospector
+from .type_mapping import map_db_type
+
+
+class PostgresIntrospector(DBIntrospector):
+    conn: psycopg.Connection
+
+    def map_db_type(self, db_type: str) -> str:
+        return map_db_type(db_type)
+
+    def get_table_columns(self, schema: str, table: str) -> list[ColumnMetaIntrospector]:
+        """
+        Returns ColumnMeta including enum labels if column type is enum.
+        """
+        q = """
+        WITH cols AS (
+          SELECT
+            a.attname                             AS col_name,
+            a.attnotnull                          AS not_null,
+            pg_catalog.format_type(a.atttypid, a.atttypmod) AS fmt_type,
+            t.typname                             AS typname,
+            t.typcategory                         AS typcategory,
+            a.atttypid                            AS typid,
+            pg_catalog.pg_get_expr(ad.adbin, ad.adrelid) AS default_expr
+          FROM pg_attribute a
+          JOIN pg_class c       ON c.oid = a.attrelid
+          JOIN pg_namespace n   ON n.oid = c.relnamespace
+          JOIN pg_type t        ON t.oid = a.atttypid
+          LEFT JOIN pg_attrdef ad
+               ON ad.adrelid = a.attrelid
+              AND ad.adnum   = a.attnum
+          WHERE n.nspname = %s
+            AND c.relname = %s
+            AND a.attnum > 0
+            AND NOT a.attisdropped
+          ORDER BY a.attnum
+        )
+        SELECT
+          cols.col_name,
+          cols.typname,
+          cols.fmt_type,
+          NOT not_null AS is_nullable,
+          default_expr,
+          CASE WHEN typcategory = 'E' THEN (
+            SELECT array_agg(e.enumlabel ORDER BY e.enumsortorder)
+            FROM pg_enum e
+            WHERE e.enumtypid = cols.typid
+          )
+          ELSE NULL
+          END AS enum_labels
+        FROM cols;
+        """
+        with self.conn.cursor() as cur:
+            cur.execute(q, (schema, table))
+            rows = cur.fetchall()
+
+        out: list[ColumnMetaIntrospector] = []
+        for row in rows:
+            out.append(
+                ColumnMetaIntrospector(
+                    col_name=row["col_name"],
+                    db_type=row["typname"],
+                    is_nullable=bool(row["is_nullable"]),
+                    has_default=(row["default_expr"] is not None),
+                    default_expr=row["default_expr"],
+                    # enum_labels=list(row["enum_labels"]) if row["enum_labels"] else None,
+                )
+            )
+
+        return out

database_wrapper_pgsql/type_mapping.py

@@ -0,0 +1,54 @@
+# type_mapping_pg.py
+from typing import Any, Optional, Tuple
+from decimal import Decimal
+import datetime
+from database_wrapper import SerializeType
+
+# Flip this if you want lossless decimals
+USE_DECIMAL = False
+
+_PG_TO_PY_BASE: dict[str, Tuple[type, Optional[SerializeType]]] = {
+    # integers
+    "int2": (int, None),
+    "int4": (int, None),
+    "int8": (int, None),
+    # floats
+    "float4": (float, None),
+    "float8": (float, None),
+    # bool
+    "bool": (bool, None),
+    "boolean": (bool, None),
+    # text
+    "text": (str, None),
+    "varchar": (str, None),
+    "bpchar": (str, None),
+    "citext": (str, None),
+    # json
+    "json": (dict[str, Any], SerializeType.JSON),
+    "jsonb": (dict[str, Any], SerializeType.JSON),
+    # temporal
+    "timestamptz": (datetime.datetime, SerializeType.DATETIME),
+    "timestamp": (datetime.datetime, SerializeType.DATETIME),
+    "date": (datetime.date, SerializeType.DATE),
+    "time": (datetime.time, SerializeType.TIME),
+    "timetz": (datetime.time, SerializeType.TIME),
+    # UUID
+    "uuid": (str, None),  # or python's uuid.UUID if you prefer
+}
+
+
+def map_db_type(
+    db_type: str, *, length: int | None = None, precision: int | None = None, scale: int | None = None
+) -> tuple[type, Optional[SerializeType]]:
+    t = db_type.lower()
+    if t == "numeric":
+        if USE_DECIMAL:
+            return (Decimal, SerializeType.DECIMAL)
+        return (float, None)
+
+    # money: tends to need Decimal
+    if t in ("money", "smallmoney"):
+        return (Decimal, SerializeType.DECIMAL) if USE_DECIMAL else (float, None)
+
+    # fall back to base
+    return _PG_TO_PY_BASE.get(t, (str, None))

{database_wrapper_pgsql-0.2.7.dist-info → database_wrapper_pgsql-0.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: database_wrapper_pgsql
-Version: 0.2.7
+Version: 0.2.9
 Summary: database_wrapper for PostgreSQL database
 Author-email: Gints Murans <gm@gm.lv>
 License: GNU General Public License v3.0 (GPL-3.0)
@@ -32,7 +32,7 @@ Classifier: Topic :: Software Development
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-Requires-Dist: database_wrapper==0.2.7
+Requires-Dist: database_wrapper==0.2.9
 Requires-Dist: psycopg[binary]>=3.2.0
 Requires-Dist: psycopg[pool]>=3.2.0
 

{database_wrapper_pgsql-0.2.7.dist-info → database_wrapper_pgsql-0.2.9.dist-info}/RECORD

@@ -1,10 +1,12 @@
-database_wrapper_pgsql/__init__.py,sha256=9QvHYUq5EgEpqIaZArrm30v_USM5zIaf0UN-z6cHWhs,1058
+database_wrapper_pgsql/__init__.py,sha256=MKa2gKPeRh-nQxp_Jd1BxbmZtcow_M3LBtBe7hQhrWQ,1136
 database_wrapper_pgsql/connector.py,sha256=Hicf9DHxOjv6H77ySdTi26YmKjlSTzOIWQrjETq3LRg,25027
 database_wrapper_pgsql/db_wrapper_pgsql.py,sha256=Tm18dzSOINz4Z8xHWynLCPf-HmD_SFP0PYVE8ZPd_A4,1921
 database_wrapper_pgsql/db_wrapper_pgsql_async.py,sha256=d4AZp_dHdmob7oJlEst2DnhMqpeXeG6WOzI8k5OldKg,2024
 database_wrapper_pgsql/db_wrapper_pgsql_mixin.py,sha256=b4m7YSk5OjukuCMky_VwFT8TYCqjU2goNnSC2HOM0HA,5452
+database_wrapper_pgsql/pg_introspector.py,sha256=MIhDFsvQ4a1bLQaCADOmvyV-Kr1cddlKFb9novf7bhw,2556
 database_wrapper_pgsql/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-database_wrapper_pgsql-0.2.7.dist-info/METADATA,sha256=kgUU9s_hDPvWuvg-YC-86i8XTTCGkd3nQGA86JMgGdc,3229
-database_wrapper_pgsql-0.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-database_wrapper_pgsql-0.2.7.dist-info/top_level.txt,sha256=EQhZLk12wRdsSp-Lo3Jc4cXmNfG8y5EouNv_7OBCSGo,23
-database_wrapper_pgsql-0.2.7.dist-info/RECORD,,
+database_wrapper_pgsql/type_mapping.py,sha256=TD1gEPNEidFfbofTYKRuuNNqHqPG0eiijoE9ICvV6SQ,1647
+database_wrapper_pgsql-0.2.9.dist-info/METADATA,sha256=J0u0g_HR9YgRB2eTsgllHqsKVA6kE01vLXawmuI81Z0,3229
+database_wrapper_pgsql-0.2.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+database_wrapper_pgsql-0.2.9.dist-info/top_level.txt,sha256=EQhZLk12wRdsSp-Lo3Jc4cXmNfG8y5EouNv_7OBCSGo,23
+database_wrapper_pgsql-0.2.9.dist-info/RECORD,,