From 8c7692c385d9bbde3032282fcdb8f2baba5561c2 Mon Sep 17 00:00:00 2001 From: Michael van Tellingen Date: Thu, 24 Feb 2011 21:55:52 +0100 Subject: [PATCH] Add sqlalchemy headers for the result sets --- flaskext/debugtoolbar/views.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/flaskext/debugtoolbar/views.py b/flaskext/debugtoolbar/views.py index 91624af..86f8a23 100644 --- a/flaskext/debugtoolbar/views.py +++ b/flaskext/debugtoolbar/views.py @@ -29,9 +29,10 @@ def sql_select(render): db = SQLAlchemy(current_app) - result = db.engine.execute(statement, params).fetchall() + result = db.engine.execute(statement, params) return render('panels/sqlalchemy_select.html', { - 'result': result, + 'result': result.fetchall(), + 'headers': result.keys(), 'sql': format_sql(statement, params), 'duration': float(request.args['duration']), }) @@ -59,9 +60,10 @@ def sql_explain(render): else: query = 'EXPLAIN %s' % statement - result = db.engine.execute(query, params).fetchall() + result = db.engine.execute(query, params) return render('panels/sqlalchemy_explain.html', { - 'result': result, + 'result': result.fetchall(), + 'headers': result.keys(), 'sql': format_sql(statement, params), 'duration': float(request.args['duration']), })