Get columns before fetching all rows

pennsignals · Feb 11, 2022 · 0f04ed7 · 0f04ed7
1 parent e1ab7b2
commit 0f04ed7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dsdk/persistor.py b/src/dsdk/persistor.py
@@ -138,8 +138,8 @@ def df_from_query_by_keys(
         with NamedTemporaryFile("w", delete=False, suffix=".sql") as fout:
             fout.write(rendered)
         cur.execute(rendered)
-        rows = cur.fetchall()
         columns = (each[0] for each in cur.description)
+        rows = cur.fetchall()
         if rows:
             df = DataFrame(rows)
             df.columns = columns