dbt-cube-sync 0.1.0a6__py3-none-any.whl → 0.1.0a7__py3-none-any.whl

dbt_cube_sync/core/db_inspector.py

@@ -2,7 +2,7 @@
 Database inspector - fetches column types using SQLAlchemy
 """
 from typing import Dict, Optional
-from sqlalchemy import create_engine, inspect, MetaData, Table
+from sqlalchemy import create_engine, inspect, MetaData, Table, text
 from sqlalchemy.engine import Engine
 
 
@@ -16,8 +16,17 @@ class DatabaseInspector:
         Args:
             sqlalchemy_uri: SQLAlchemy connection URI (e.g., postgresql://user:pass@host:port/db)
         """
-        self.engine: Engine = create_engine(sqlalchemy_uri)
+        # Add connect_args for Redshift compatibility
+        if 'redshift' in sqlalchemy_uri:
+            self.engine: Engine = create_engine(
+                sqlalchemy_uri,
+                connect_args={'sslmode': 'prefer'}
+            )
+        else:
+            self.engine: Engine = create_engine(sqlalchemy_uri)
+
         self.inspector = inspect(self.engine)
+        self.is_redshift = 'redshift' in sqlalchemy_uri.lower()
 
     def get_table_columns(self, schema: str, table_name: str) -> Dict[str, str]:
         """
@@ -33,19 +42,71 @@ class DatabaseInspector:
         columns = {}
 
         try:
-            # Get columns from the database
-            table_columns = self.inspector.get_columns(table_name, schema=schema)
+            # For Redshift, use direct SQL query to avoid pg_catalog issues
+            if self.is_redshift:
+                columns = self._get_redshift_columns(schema, table_name)
+            else:
+                # Get columns from the database using inspector
+                table_columns = self.inspector.get_columns(table_name, schema=schema)
 
-            for column in table_columns:
-                col_name = column['name']
-                col_type = str(column['type'])
-                columns[col_name] = col_type
+                for column in table_columns:
+                    col_name = column['name']
+                    col_type = str(column['type'])
+                    columns[col_name] = col_type
 
         except Exception as e:
             print(f"Warning: Could not inspect table {schema}.{table_name}: {e}")
 
         return columns
 
+    def _get_redshift_columns(self, schema: str, table_name: str) -> Dict[str, str]:
+        """
+        Get columns for Redshift using direct SQL query
+
+        Args:
+            schema: Database schema name
+            table_name: Table name
+
+        Returns:
+            Dictionary mapping column names to data types
+        """
+        columns = {}
+
+        try:
+            # Query Redshift's pg_table_def view which is more reliable
+            query = text("""
+                SELECT column_name, data_type
+                FROM pg_table_def
+                WHERE schemaname = :schema
+                AND tablename = :table_name
+                ORDER BY column_name
+            """)
+
+            with self.engine.connect() as conn:
+                result = conn.execute(query, {"schema": schema, "table_name": table_name})
+                for row in result:
+                    columns[row[0]] = row[1]
+
+        except Exception as e:
+            # Fallback to information_schema if pg_table_def fails
+            try:
+                query = text("""
+                    SELECT column_name, data_type
+                    FROM information_schema.columns
+                    WHERE table_schema = :schema
+                    AND table_name = :table_name
+                    ORDER BY ordinal_position
+                """)
+
+                with self.engine.connect() as conn:
+                    result = conn.execute(query, {"schema": schema, "table_name": table_name})
+                    for row in result:
+                        columns[row[0]] = row[1]
+            except Exception as fallback_error:
+                print(f"Warning: Could not query Redshift table {schema}.{table_name}: {fallback_error}")
+
+        return columns
+
     def close(self):
         """Close the database connection"""
         self.engine.dispose()

{dbt_cube_sync-0.1.0a6.dist-info → dbt_cube_sync-0.1.0a7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbt-cube-sync
-Version: 0.1.0a6
+Version: 0.1.0a7
 Summary: Synchronization tool for dbt models to Cube.js schemas and BI tools
 Author: Ponder
 Requires-Python: >=3.9,<4.0

{dbt_cube_sync-0.1.0a6.dist-info → dbt_cube_sync-0.1.0a7.dist-info}/RECORD

@@ -8,10 +8,10 @@ dbt_cube_sync/connectors/superset.py,sha256=5YEqadVZRPFAJkgvhqkse3JuGJkQHfyvT88j
 dbt_cube_sync/connectors/tableau.py,sha256=jKve1zErzTbgPOtmPB92ZwZl4I6uEySedM51JiwlGrE,1261
 dbt_cube_sync/core/__init__.py,sha256=kgsawtU5dqEvnHz6dU8qwJbH3rtIV7QlK2MhtYVDCaY,46
 dbt_cube_sync/core/cube_generator.py,sha256=o_-fa09F3RQADueIgou8EFhmxKd7PbQ-hCJmXvRuvWM,10839
-dbt_cube_sync/core/db_inspector.py,sha256=eoJl7XG3dPcKg22SEX2dehC8Hvj5hgLR8sUgKiPCIGI,1540
+dbt_cube_sync/core/db_inspector.py,sha256=HK7hpU56X5ED-i_vXGB9rVs79eAwgEXS_SMdk08PHs0,3850
 dbt_cube_sync/core/dbt_parser.py,sha256=vQEUO19WYdeFNnulU2_PD4hdHUtTO-Y9BXfHuH6ZVnM,10192
 dbt_cube_sync/core/models.py,sha256=JjiFAO0vbfVZkKOd6NcZb_JMGSVMTMfQiYjHcZbKtnI,2811
-dbt_cube_sync-0.1.0a6.dist-info/METADATA,sha256=SgI2Sm6jQ748KKn3ZnFGz1qEoIaJmSfMm_Owv9hE5Hc,8274
-dbt_cube_sync-0.1.0a6.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-dbt_cube_sync-0.1.0a6.dist-info/entry_points.txt,sha256=iEAB_nZ1AoSeFwSHPY2tr02xmTHLVFKp5CJeFh0AfCw,56
-dbt_cube_sync-0.1.0a6.dist-info/RECORD,,
+dbt_cube_sync-0.1.0a7.dist-info/METADATA,sha256=2cn4bF41UTmo9nv817Xmd2I5LjGmzNYC9gNX_tlmz3Y,8274
+dbt_cube_sync-0.1.0a7.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+dbt_cube_sync-0.1.0a7.dist-info/entry_points.txt,sha256=iEAB_nZ1AoSeFwSHPY2tr02xmTHLVFKp5CJeFh0AfCw,56
+dbt_cube_sync-0.1.0a7.dist-info/RECORD,,