使用Flask.g轻松实现请求日志记录的技巧

Flask是一个 Python Web框架，提供了轻量级的解决方案来构建Web应用。Flask和其他Web框架一样，也具有请求和响应的功能。对于日志记录来说，Flask可以使用Flask.g上下文变量来实现请求日志记录。

Flask.g是一个特殊的全局变量，它在每个请求生命周期中都是独立的。通过Flask.g，我们可以在请求处理函数和路由之间共享数据。在实现请求日志记录时，我们可以使用Flask.g来存储请求相关的信息，比如请求时间、请求者IP等。

下面是一个使用Flask.g实现请求日志记录的示例：

from flask import Flask, request, g
import logging

app = Flask(__name__)
app.config['LOG_FILENAME'] = 'requests.log'

# 配置日志记录
logging.basicConfig(filename=app.config['LOG_FILENAME'], level=logging.INFO)

# 请求开始时的预处理
@app.before_request
def before_request():
    # 存储请求时间
    g.start_time = time.time()

# 请求结束时的后处理
@app.after_request
def after_request(response):
    # 计算请求处理时间
    process_time = time.time() - g.start_time

    # 记录请求日志
    log_message = f"{request.method} {request.path} - {request.remote_addr} - {response.status_code} - {process_time:.2f}s"
    app.logger.info(log_message)

    return response

# 路由和请求处理函数
@app.route('/')
def index():
    # 处理请求
    return 'Hello, World!'

if __name__ == '__main__':
    app.run()

在上面的示例中，我们使用了before_request装饰器来定义一个预处理函数，该函数会在每个请求开始时执行。在该函数中，我们将请求开始的时间存储在g.start_time变量中。

然后，我们使用after_request装饰器来定义一个后处理函数，该函数会在每个请求结束时执行。在该函数中，我们通过计算g.start_time和当前时间的差来得到请求处理时间。然后，我们使用Flask的日志记录器(app.logger)将请求的相关信息写入日志文件。

最后，我们定义了一个路由和请求处理函数(index)，该函数会在访问根路径时执行。在这个示例中，我们只是简单地返回了一个字符串作为响应。

需要注意的是，我们在Flask的配置中设置了LOG_FILENAME变量，用于指定日志文件的文件名。你可以根据自己的需求进行修改。另外，为了使日志记录生效，你需要确保日志文件的目录具有写入权限。

总结起来，使用Flask.g来实现请求日志记录非常简单。通过在请求开始时预处理和请求结束时后处理，我们可以轻松地记录请求的相关信息。这样的日志记录对于调试、性能分析和安全审计非常有用。