diff --git a/depobs/website/do.py b/depobs/website/do.py index 9cf0150d..37031f16 100644 --- a/depobs/website/do.py +++ b/depobs/website/do.py @@ -3,7 +3,7 @@ import logging.config import celery -from flask import Flask +from flask import Flask, request from dockerflow.flask import Dockerflow from dockerflow.logging import JsonLogFormatter @@ -18,6 +18,16 @@ flasklog.setLevel(logging.ERROR) +# override the summary logger from the Dockerflow class to add +# logging of query string +# https://github.com/mozilla-services/python-dockerflow/issues/44 +class Customflow(Dockerflow): + def summary_extra(self): + out = super().summary_extra() + out["query_string"] = request.query_string.decode("utf-8") + return out + + def create_app(test_config=None): # reimport to pick up changes for testing and autoreload import depobs.website.models as models @@ -38,7 +48,7 @@ def create_app(test_config=None): if app.config["INIT_DB"]: models.create_tables_and_views(app) - dockerflow = Dockerflow(app, db=models.db) + dockerflow = Customflow(app, db=models.db) dockerflow.init_app(app) app.register_blueprint(scans_blueprint) app.register_blueprint(views_blueprint)