Skip to content

Commit 70488fc

Browse files
committed
Fix Py3 support for bytes SQL queries
Fixes #64
1 parent 4901d68 commit 70488fc

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

flask_debugtoolbar/panels/sqlalchemy.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,13 @@ def query_signer():
2121
salt='fdt-sql-query')
2222

2323

24+
def is_select(statement):
25+
prefix = b'select' if isinstance(statement, bytes) else 'select'
26+
return statement.lower().strip().startswith(prefix)
27+
28+
2429
def dump_query(statement, params):
25-
if not params or not statement.lower().strip().startswith('select'):
30+
if not params or not is_select(statement):
2631
return None
2732

2833
try:
@@ -38,7 +43,7 @@ def load_query(data):
3843
abort(406)
3944

4045
# Make sure it is a select statement
41-
if not statement.lower().strip().startswith('select'):
46+
if not is_select(statement):
4247
abort(406)
4348

4449
return statement, params

0 commit comments

Comments
 (0)