日志中添加Referrer信息

在Flask中使用默认的logging模块进行日志记录时，可以通过在日志记录器中添加Referrer字段来包含请求的Referrer信息。您可以通过以下方式进行修改：

import logging
from logging.handlers import RotatingFileHandler
from flask import Flask, request

app = Flask(__name__)

# 创建日志记录器
logger = logging.getLogger('my_logger')
logger.setLevel(logging.INFO)

# 设置日志格式
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')


# 创建错误日志处理器
error_handler = RotatingFileHandler('logs/error.log', encoding='utf-8', maxBytes=10485760, backupCount=10)
error_handler.setLevel(logging.ERROR)
error_handler.setFormatter(formatter)
logger.addHandler(error_handler)

# 创建普通日志处理器
handler = RotatingFileHandler('logs/app.log', encoding='utf-8-sig', maxBytes=10485760, backupCount=10)
handler.setLevel(logging.INFO)
handler.setFormatter(formatter)
logger.addHandler(handler)


# 设置文件编码为 UTF-8
log_file = codecs.open('logs/app.log', mode='a', encoding='utf-8')
handler.stream = log_file



# 记录带有Referrer的日志
@app.after_request
def log_request(response):
referrer = request.referrer
app.logger.info('Referrer: %s', referrer)
return response
 

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

在上述代码中，通过request.referrer获取请求的Referrer信息，并将其添加到日志记录中。您可以在适当的位置使用logger.info('Referrer: %s', referrer)语句记录带有Referrer字段的日志。

请注意，要正确使用request.referrer，您需要确保请求中包含Referer（注意拼写）头部，并且该头部的值有效。