From 2c13ce090816e6a5e09e46c0d6427faefe0a0af4 Mon Sep 17 00:00:00 2001
From: xaoyaoo <xaoyaoo@163.com>
Date: Sat, 3 Aug 2024 18:42:28 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=97=A5=E5=BF=97=E7=AE=A1?=
 =?UTF-8?q?=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pywxdump/server.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/pywxdump/server.py b/pywxdump/server.py
index b046460..f77f19e 100644
--- a/pywxdump/server.py
+++ b/pywxdump/server.py
@@ -25,7 +25,7 @@ def is_port_in_use(_host, _port):
 
 
 def start_falsk(merge_path="", wx_path="", key="", my_wxid="", port=5000, online=False, debug=False,
-                isopenBrowser=True):
+                isopenBrowser=True, loger_handler=None):
     """
     启动flask
     :param merge_path:  合并后的数据库路径
@@ -63,6 +63,13 @@ def start_falsk(merge_path="", wx_path="", key="", my_wxid="", port=5000, online
     app.secret_key = 'secret_key'
 
     app.logger.setLevel(logging.WARNING)
+    if loger_handler:
+        app.logger.addHandler(loger_handler)
+        # 获取 Werkzeug 的日志记录器
+        werkzeug_logger = logging.getLogger('werkzeug')
+        # 将自定义格式器应用到 Werkzeug 的日志记录器
+        werkzeug_logger.addHandler(loger_handler)
+        werkzeug_logger.setLevel(logging.DEBUG)
 
     CORS(app, resources={r"/*": {"origins": "*"}}, supports_credentials=True)  # 允许所有域名跨域