57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
"""
|
|
初始化插件
|
|
"""
|
|
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:
|
|
arr = request.headers['HTTP_X_FORWARDED_FOR'].strip().split(",")
|
|
i = 0
|
|
while i < len(arr):
|
|
if arr[i].find("unknown") != -1:
|
|
del arr[i]
|
|
else:
|
|
i += 1
|
|
if len(arr) != 0:
|
|
return arr[0].strip()
|
|
elif 'HTTP_CLIENT_IP' in request.headers:
|
|
return request.headers['HTTP_CLIENT_IP']
|
|
elif 'REMOTE_ADDR' in request.headers:
|
|
return request.headers['REMOTE_ADDR']
|
|
elif 'X-Forwarded-For' in request.headers:
|
|
return request.headers['X-Forwarded-For']
|
|
return request.remote_addr |