de-duplicate colnames if they are identical

defog-ai · May 16, 2024 · aea5406 · aea5406
1 parent 958af22
commit aea5406
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/defog/query.py b/defog/query.py
@@ -43,6 +43,10 @@ def execute_query_once(db_type: str, db_creds, query: str):
 
         cur.execute(query)
         colnames = [desc[0] for desc in cur.description]
+
+        # if there are any column names that are the same, we need to deduplicate them
+        colnames = [f"{col}_{i}" if colnames.count(col) > 1 else col for i, col in enumerate(colnames)]
+
         results = cur.fetchall()
         cur.close()
         conn.close()