Add support for engine-per-query as well as pulling out psycopg2's isolation level and transaction status

1 files changed, 8 insertions, 2 deletions

diff --git a/debug_toolbar/templates/debug_toolbar/panels/sql.html b/debug_toolbar/templates/debug_toolbar/panels/sql.html
index bec97a4..436f3ce 100644
--- a/debug_toolbar/templates/debug_toolbar/panels/sql.html
+++ b/debug_toolbar/templates/debug_toolbar/panels/sql.html
@@ -44,9 +44,9 @@
 						{% if query.is_select %}
 							<a class="remoteCall" href="/__debug__/sql_select/?sql={{ query.raw_sql|urlencode }}&amp;params={{ query.params|urlencode }}&amp;duration={{ query.duration|floatformat:"2"|urlencode }}&amp;hash={{ query.hash }}">Sel</a>
 							<a class="remoteCall" href="/__debug__/sql_explain/?sql={{ query.raw_sql|urlencode }}&amp;params={{ query.params|urlencode }}&amp;duration={{ query.duration|floatformat:"2"|urlencode }}&amp;hash={{ query.hash }}">Expl</a>
-							{% if is_mysql %}
+							{% ifequal query.engine 'mysql' %}
 								<a class="remoteCall" href="/__debug__/sql_profile/?sql={{ query.raw_sql|urlencode }}&amp;params={{ query.params|urlencode }}&amp;duration={{ query.duration|floatformat:"2"|urlencode }}&amp;hash={{ query.hash }}">Prof</a>
-							{% endif %}
+							{% endifequal %}
 						{% endif %}
 					{% endif %}
 					</td>
@@ -56,6 +56,12 @@
 					<td colspan="4">
 						<div class="djSQLDetailsDiv">
 							<p><strong>Connection:</strong> {{ query.alias }}</p>
+							{% if query.iso_level %}
+								<p><strong>Isolation Level:</strong> {{ query.iso_level }}</p>
+							{% endif %}
+							{% if query.trans_status %}
+								<p><strong>Transaction Status:</strong> {{ query.trans_status }}</p>
+							{% endif %}
 							{% if query.stacktrace %}
 								<pre>{{ query.stacktrace }}</pre>
 							{% endif %}