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")

View file

@ -75,21 +75,25 @@ class BaseView(AsgiView):
async def render(self, templates, request, context):
template = self.ds.jinja_env.select_template(templates)
template_context = {
**context,
**{
"database_url": self.database_url,
"database_color": self.database_color,
},
}
if request and request.args.get("_context") and self.ds.config("template_debug"):
**context,
**{
"database_url": self.database_url,
"database_color": self.database_color,
},
}
if (
request
and request.args.get("_context")
and self.ds.config("template_debug")
):
return Response.html(
"<pre>{}</pre>".format(
jinja2.escape(json.dumps(template_context, default=repr, indent=4))
)
)
return Response.html(await self.ds.render_template(
template, template_context, request=request
))
return Response.html(
await self.ds.render_template(template, template_context, request=request)
)
class DataView(BaseView):