Reformatted with black

This commit is contained in:
Simon Willison 2020-02-12 22:05:46 -08:00
commit 298a899e79
2 changed files with 17 additions and 11 deletions

View file

@ -174,7 +174,9 @@ disallawed_sql_res = [(re.compile("pragma"), "Statement may not contain PRAGMA")
def validate_sql_select(sql):
sql = "\n".join(line for line in sql.split('\n') if not line.strip().startswith('--'))
sql = "\n".join(
line for line in sql.split("\n") if not line.strip().startswith("--")
)
sql = sql.strip().lower()
if not any(r.match(sql) for r in allowed_sql_res):
raise InvalidSql("Statement must be a SELECT")