更新文档
This commit is contained in:
@@ -4,38 +4,21 @@
|
||||
import os
|
||||
import logging
|
||||
from flask import Flask, request
|
||||
from . import console
|
||||
|
||||
# 获取插件所在的目录(结尾没有分割符号)
|
||||
dir_path = os.path.dirname(__file__).replace("\\", "/")
|
||||
folder_name = dir_path[dir_path.rfind("/") + 1:] # 插件文件夹名称
|
||||
|
||||
|
||||
def event_init(app: Flask):
|
||||
"""初始化完成时会调用这里"""
|
||||
# 移除原有的输出日志
|
||||
app.logger = None
|
||||
log = logging.getLogger('werkzeug')
|
||||
log.setLevel(logging.ERROR)
|
||||
|
||||
|
||||
# 更改IP地址,只有在最新版的flask中才能生效
|
||||
@app.before_request
|
||||
def before_request():
|
||||
request.remote_addr = get_user_ip(request)
|
||||
|
||||
|
||||
# 使用自定义的日志输出
|
||||
@app.after_request
|
||||
def after_request(rep):
|
||||
if rep.status_code == 200:
|
||||
console.success(f"{request.remote_addr} -- {request.full_path} 200")
|
||||
elif rep.status_code == 404:
|
||||
console.error(f"{request.remote_addr} -- {request.full_path} 404")
|
||||
elif rep.status_code == 500:
|
||||
console.warning(f"{request.remote_addr} -- {request.full_path} 500")
|
||||
else:
|
||||
console.info(f"{request.remote_addr} -- {request.full_path} {rep.status_code}")
|
||||
return rep
|
||||
|
||||
|
||||
|
||||
def get_user_ip(request):
|
||||
"""获取用户真实IP"""
|
||||
if 'HTTP_X_FORWARDED_FOR' in request.headers:
|
||||
@@ -54,4 +37,4 @@ def get_user_ip(request):
|
||||
return request.headers['REMOTE_ADDR']
|
||||
elif 'X-Forwarded-For' in request.headers:
|
||||
return request.headers['X-Forwarded-For']
|
||||
return request.remote_addr
|
||||
return request.remote_addr
|
||||
|
||||
Reference in New Issue
Block a user