Handle cases with cur.description is None (#130)

* Handle cases with cur.description is None --------- Co-authored-by: Matts966 <[email protected]>
treasure-data · Mar 28, 2024 · 708e8c5 · 708e8c5
1 parent 3b298d1
commit 708e8c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytd/query_engine.py b/pytd/query_engine.py
@@ -95,7 +95,8 @@ def execute(self, query, **kwargs):
         cur = self.cursor(**kwargs)
         self.executed = cur.execute(query)
         rows = cur.fetchall()
-        columns = [desc[0] for desc in cur.description]
+        # cur.description is None for CREATE and DROP statements in recent version of Trino
+        columns = [desc[0] for desc in cur.description] if cur.description else None
         return {"data": rows, "columns": columns}
 
     def create_header(self, extra_lines=[]):