Merge pull request #31 from ActianCorp/hab6-pkconstraint-fix

Fix get_primary_keys return type
ActianCorp · Feb 20, 2024 · 89119af · 89119af
2 parents 1be79a4 + 62c75e3
commit 89119af
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/sqlalchemy_ingres/base.py b/lib/sqlalchemy_ingres/base.py
@@ -376,8 +376,10 @@ def get_primary_keys(self, connection, table_name, schema=None, **kw):
 
         try:
             rs = connection.exec_driver_sql(sqltext, params)
-
-            return [row[0].rstrip() for row in rs.fetchall()]
+
+            cols = [row[0].rstrip() for row in rs.fetchall()]
+            return {"constrained_columns": [] if cols is None else cols, "name": None}
+
         finally:
             if rs:
                 rs.close()