mirror/flask-debugtoolbar
1
0
Fork 0
mirror of https://github.com/pallets-eco/flask-debugtoolbar.git synced 2025-12-30 18:19:31 -06:00
Code Issues Packages Projects Releases Wiki Activity

Fix Py3 support for bytes SQL queries

Browse Source 
Fixes #64
This commit is contained in:
Matt Good
2014-01-14 17:56:47 -08:00
parent 4901d68b01
commit 70488fc14a
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
flask_debugtoolbar/panels/sqlalchemy.py
View File

@@ -21,8 +21,13 @@ def query_signer():
salt='fdt-sql-query')
def is_select(statement):
prefix = b'select' if isinstance(statement, bytes) else 'select'
return statement.lower().strip().startswith(prefix)
def dump_query(statement, params):
if not params or not statement.lower().strip().startswith('select'):
if not params or not is_select(statement):
return None
try:
@@ -38,7 +43,7 @@ def load_query(data):
abort(406)
# Make sure it is a select statement
if not statement.lower().strip().startswith('select'):
if not is_select(statement):
abort(406)
return statement, params