Access cursor before closing the db connection

sphinx-contrib · May 13, 2024 · a6da716 · a6da716
1 parent 6610826
commit a6da716
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sphinxcontrib/sqltable.py b/sphinxcontrib/sqltable.py
@@ -71,6 +71,8 @@ def run(self):
             LOG.info('Running query %r' % query)
             with engine.connect() as conn:
                 results = conn.execute(sqlalchemy.text(query))
+                table_headers = results.keys()
+                table_body = results.fetchall()
         except Exception as err:
             error = self.state_machine.reporter.error(
                 u'Error with query %s for sqltable: %s' % (
@@ -83,8 +85,6 @@ def run(self):
             return [error]
 
         # Extract some values we need for building the table.
-        table_headers = results.keys()
-        table_body = results
         max_cols = len(table_headers)
 
         # Handle the width settings and title