更新插件功能

plugins/realip/__init__.json

@@ -0,0 +1,9 @@
+{
+  "plugin_name": "真实IP插件",
+  "plugin_version": "1.0.0.1",
+  "plugin_description": "在上游更改访客IP地址，并采用自定义日志输出。",
+  "developer_name": "Yishang",
+  "developer_website": "https://lovepikachu.top",
+  "developer_email": "422880152@qq.com",
+  "developer_phone": "-"
+}

plugins/realip/__init__.py

@@ -0,0 +1,66 @@
+"""
+初始化插件
+"""
+import os
+import logging
+from flask import Flask, request
+from applications.dev import console
+
+# 获取插件所在的目录（结尾没有分割符号）
+dir_path = os.path.dirname(__file__).replace("\\", "/")
+folder_name = dir_path[dir_path.rfind("/") + 1:]  # 插件文件夹名称
+
+def event_enable():
+    """当此插件被启用时会调用此处"""
+    print(f"启用插件，dir_path: {dir_path} ; folder_name: {folder_name}")
+
+
+def event_disable():
+    """当此插件被禁用时会调用此处"""
+    print(f"禁用插件，dir_path: {dir_path} ; folder_name: {folder_name}")
+
+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