fix: make get_pk_constraint adhere to SqlAlchemy specs (#9)

Lordshinjo · Mar 19, 2020 · 466faee · 466faee
1 parent c925457
commit 466faee
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sqla_vertica_python/vertica_python.py b/sqla_vertica_python/vertica_python.py
@@ -372,12 +372,12 @@ def get_pk_constraint(self, connection, table_name, schema=None, **kw):
             query += " AND table_schema = '" + schema + "' \n"
 
         cols = set()
-        name = set()
+        name = None
         for row in connection.execute(query):
-             name.add(row[1])
+             name = row[1] if name is None else name
              cols.add(row[2])
 
-        return {"constrained_columns": cols, "name": name}
+        return {"constrained_columns": list(cols), "name": name}
 
 
     def get_foreign_keys(self, connection, table_name, schema, **kw):