Allow whitespace prefix for sql queries

This commit is contained in:
HelloZeroNet 2016-04-25 02:25:54 +02:00
parent 23265861c9
commit 34ead0aec2

View file

@ -437,7 +437,7 @@ class UiWebsocket(object):
def actionDbQuery(self, to, query, params=None, wait_for=None): def actionDbQuery(self, to, query, params=None, wait_for=None):
rows = [] rows = []
try: try:
assert query.upper().startswith("SELECT"), "Only SELECT query supported" assert query.strip().upper().startswith("SELECT"), "Only SELECT query supported"
res = self.site.storage.query(query, params) res = self.site.storage.query(query, params)
except Exception, err: # Response the error to client except Exception, err: # Response the error to client
return self.response(to, {"error": str(err)}) return self.response(to, {"error": str(err)})