python - 如何在访问日志中添加 Flask 全局 'g' 对象的 id？

Question

我正在使用 Gunicorn 来运行我的 Flask 应用程序。

我在 Flask 的全局 g 对象上有一个 user_id ，我想用它来登录每个经过身份验证的请求。

有人知道有什么好方法吗？我尝试添加这样的自定义过滤器，但执行记录器时请求似乎不在范围内:

class MyFilter(logging.Filter):

    def filter(self, record):
        if hasattr(g, 'user_id'):
            record.msg = "User ID: {0} ".format(g.user_id) + record.msg
        return 1

我期待这样的事情:

2014-11-08 21:52:15 [INFO] REQUESTLOG 127.0.0.1 User ID: 12345 "GET /v1.0/healthcheck HTTP/1.1" 200 51 "-" "curl/7.35.0"

谢谢!

Answer 1

您可以子类化 Werkzeug 请求处理程序并提供您自己版本的 log_request() 方法。这是一个例子:

from werkzeug.serving import BaseRequestHandler
from flask import Flask
app = Flask(__name__)

@app.route('/')
def index():
    return "hi"

class MyRequestHandler(BaseRequestHandler):
    def log_request(self, code='-', size='-'):
        self.log('info', 'custom log! "%s" %s %s', self.requestline, code, size)

if __name__ == '__main__':
    app.run(debug=True, request_handler=MyRequestHandler)