thoth-dbmanager 0.5.9__py3-none-any.whl → 0.5.10__py3-none-any.whl

thoth_dbmanager/adapters/postgresql.py

@@ -176,33 +176,33 @@ class PostgreSQLAdapter(DbAdapter):
    
     
     def get_columns_as_documents(self, table_name: str) -> List[ColumnDocument]:
-        """Get columns as document objects"""
+        """Get columns as document objects using pg_catalog to avoid SELECT restrictions"""
         query = """
-        SELECT 
-            c.column_name,
-            c.data_type,
-            c.is_nullable,
-            c.column_default,
-            c.character_maximum_length,
-            COALESCE(pgd.description, '') as comment,
-            CASE WHEN pk.column_name IS NOT NULL THEN true ELSE false END as is_pk,
-            c.table_schema as schema_name
-        FROM information_schema.columns c
-        LEFT JOIN pg_class pgc ON pgc.relname = c.table_name
-        LEFT JOIN pg_namespace pgn ON pgn.oid = pgc.relnamespace AND pgn.nspname = c.table_schema
-        LEFT JOIN pg_description pgd ON pgd.objoid = pgc.oid AND pgd.objsubid = c.ordinal_position
-        LEFT JOIN (
-            SELECT ku.column_name, ku.table_name, ku.table_schema
-            FROM information_schema.table_constraints tc
-            JOIN information_schema.key_column_usage ku ON tc.constraint_name = ku.constraint_name
-            WHERE tc.constraint_type = 'PRIMARY KEY'
-        ) pk ON pk.column_name = c.column_name AND pk.table_name = c.table_name AND pk.table_schema = c.table_schema
-        WHERE c.table_name = :table_name
-        AND c.table_schema NOT IN ('information_schema', 'pg_catalog')
-        AND c.table_schema = :schema
-        ORDER BY c.ordinal_position
+        SELECT
+            a.attname AS column_name,
+            format_type(a.atttypid, a.atttypmod) AS data_type,
+            (NOT a.attnotnull) AS is_nullable,
+            pg_get_expr(d.adbin, d.adrelid) AS column_default,
+            NULL::int AS character_maximum_length,
+            COALESCE(pgd.description, '') AS comment,
+            EXISTS (
+                SELECT 1
+                FROM pg_index i
+                WHERE i.indrelid = c.oid AND i.indisprimary AND a.attnum = ANY(i.indkey)
+            ) AS is_pk,
+            n.nspname AS schema_name
+        FROM pg_class c
+        JOIN pg_namespace n ON n.oid = c.relnamespace
+        JOIN pg_attribute a ON a.attrelid = c.oid
+        LEFT JOIN pg_attrdef d ON d.adrelid = c.oid AND d.adnum = a.attnum
+        LEFT JOIN pg_description pgd ON pgd.objoid = c.oid AND pgd.objsubid = a.attnum
+        WHERE c.relname = :table_name
+          AND n.nspname = :schema
+          AND a.attnum > 0
+          AND NOT a.attisdropped
+        ORDER BY a.attnum
         """
-        
+
         results = self.execute_query(query, {"table_name": table_name, "schema": self.schema})
         documents = []
         
@@ -212,8 +212,8 @@ class PostgreSQLAdapter(DbAdapter):
                 column_name=row.column_name,
                 data_type=row.data_type,
                 comment=row.comment or "",
-                is_pk=row.is_pk,
-                is_nullable=row.is_nullable == 'YES',
+                is_pk=bool(row.is_pk),
+                is_nullable=bool(row.is_nullable),
                 default_value=row.column_default,
                 max_length=row.character_maximum_length,
                 schema_name=row.schema_name

{thoth_dbmanager-0.5.9.dist-info → thoth_dbmanager-0.5.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: thoth_dbmanager
-Version: 0.5.9
+Version: 0.5.10
 Summary: A Python library for managing SQL databases with support for multiple database types, LSH-based similarity search, and a modern plugin architecture.
 Author-email: Marco Pancotti <mp@tylconsulting.it>
 Project-URL: Homepage, https://github.com/mptyl/thoth_dbmanager

{thoth_dbmanager-0.5.9.dist-info → thoth_dbmanager-0.5.10.dist-info}/RECORD

@@ -4,7 +4,7 @@ thoth_dbmanager/documents.py,sha256=BnYL9MR9eOl9cNOrY58SgHgXACnHCk_UzKjpfV8RjgY,
 thoth_dbmanager/dynamic_imports.py,sha256=nKCwg3Y2e5f4oP_-9X53t4Z9InbIGDMgidQjeN1Xgoo,7592
 thoth_dbmanager/adapters/__init__.py,sha256=rfXo8IxWPx2TQV2OwFrZqpOzrkKuO6HR0q8ftkuGhW0,1578
 thoth_dbmanager/adapters/mariadb.py,sha256=Bc31B4LNzlqu1QwhBP49Ia1jlh-XWh14Q_ya-KqWvsQ,15636
-thoth_dbmanager/adapters/postgresql.py,sha256=7NiYHGWQpiZ3NMT6G3tqLeCJztxNZ4E0jB-aeT5FhK0,18453
+thoth_dbmanager/adapters/postgresql.py,sha256=xUu5_U6guDx27DH6G4txADfQCp0EOuWPxkFgnnRQGJI,18259
 thoth_dbmanager/adapters/sqlite.py,sha256=sg9ifwBL3L_aBf6i3Z0wq_Z7mTjnVeWhCY1VK6Cfh9E,15147
 thoth_dbmanager/adapters/sqlserver.py,sha256=VcWW5xudK4_3BuPwX0WcBAxgN6j9seRssvwsWPpJudo,25320
 thoth_dbmanager/core/__init__.py,sha256=22dsupbGskkjGgS9IzeCwaf5o7nIKH_og1qARZURAFM,848
@@ -26,9 +26,9 @@ thoth_dbmanager/plugins/mariadb.py,sha256=OW_u_x3mfaQYH5dUEreUbelPE6KEQU4kiqT3_D
 thoth_dbmanager/plugins/postgresql.py,sha256=VFGiGa1qAWRR00DkoHIAFS-U9uMLMwb5gKa_NB-cQA8,6097
 thoth_dbmanager/plugins/sqlite.py,sha256=yqqa_lToLabOv_ut2uRTXxhjyEQSVbB8kIcUwy7HNZk,9436
 thoth_dbmanager/plugins/sqlserver.py,sha256=AxEXwe21FRh6L1lX6-gu3VzcxDQ7dO5YLvLXTrIyaRk,5868
-thoth_dbmanager-0.5.9.dist-info/licenses/LICENSE,sha256=81-BOzGgwtY1XdYfkwMQB87AkOGXI9OMq0kjNcZA4UE,1071
-thoth_dbmanager-0.5.9.dist-info/licenses/LICENSE.md,sha256=TtKT2ej3GRki3W8FmSeYzRuQtZ1oMGWchpWykZ4vA7c,1070
-thoth_dbmanager-0.5.9.dist-info/METADATA,sha256=Bic4w9wYxCM8SKiJVKVwvJJTrFo9XC96BftJivljIcw,14777
-thoth_dbmanager-0.5.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-thoth_dbmanager-0.5.9.dist-info/top_level.txt,sha256=b9ttxm9RUc0KUCASEKRx6FqoREYJ1-KZWSpNuaM0uQ4,16
-thoth_dbmanager-0.5.9.dist-info/RECORD,,
+thoth_dbmanager-0.5.10.dist-info/licenses/LICENSE,sha256=81-BOzGgwtY1XdYfkwMQB87AkOGXI9OMq0kjNcZA4UE,1071
+thoth_dbmanager-0.5.10.dist-info/licenses/LICENSE.md,sha256=TtKT2ej3GRki3W8FmSeYzRuQtZ1oMGWchpWykZ4vA7c,1070
+thoth_dbmanager-0.5.10.dist-info/METADATA,sha256=AGjfF7xBRTDs8SnHzND6VuFvRCzUav0cjaiV2Tdw7r4,14778
+thoth_dbmanager-0.5.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+thoth_dbmanager-0.5.10.dist-info/top_level.txt,sha256=b9ttxm9RUc0KUCASEKRx6FqoREYJ1-KZWSpNuaM0uQ4,16
+thoth_dbmanager-0.5.10.dist-info/RECORD,,