[fix] Sanitize blacklisted functions

This commit is contained in:
Saurabh 2018-05-30 15:07:18 +05:30
parent 418d654331
commit 55eaba90f5

View file

@ -203,7 +203,10 @@ class DatabaseQuery(object):
if any(keyword in field.lower().split() for keyword in blacklisted_keywords):
_raise_exception()
if any("{0}(".format(keyword) in field.lower().split() for keyword in blacklisted_functions):
if any("({0}".format(keyword) in field.lower() for keyword in blacklisted_keywords):
_raise_exception()
if any("{0}(".format(keyword) in field.lower() for keyword in blacklisted_functions):
_raise_exception()
def extract_tables(self):