Chore: Adjust database adapter for cratedb-sqlparse 0.0.6

crate · Aug 14, 2024 · e9dd340 · e9dd340
1 parent 882ba94
commit e9dd340
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cratedb_toolkit/util/database.py b/cratedb_toolkit/util/database.py
@@ -392,5 +392,8 @@ def get_table_names(sql: str) -> t.List[t.List[str]]:
     names = []
     statements = sqlparse_cratedb(sql)
     for statement in statements:
-        names.append([statement.metadata.table_name])
+        local_names = []
+        for table in statement.metadata.tables:
+            local_names.append(table.name)
+        names.append(local_names)
     return names