From c80fb67165e1a07afb7c5732c171e889e26bac3e Mon Sep 17 00:00:00 2001
From: ninemine <1371605831@qq.com>
Date: Wed, 29 Oct 2025 12:14:50 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dwebhook=20url=E6=8C=87?=
 =?UTF-8?q?=E5=AE=9A?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app.py    | 10 ++--------
 config.py | 15 ++++++++++-----
 2 files changed, 12 insertions(+), 13 deletions(-)

diff --git a/app.py b/app.py
index d12c7f8..506f9c6 100644
--- a/app.py
+++ b/app.py
@@ -7,7 +7,7 @@ from fastapi.responses import JSONResponse
 from contextlib import asynccontextmanager
 import asyncio
 
-from config import APP_CONFIG, SESSION_TIMEOUT
+from config import APP_CONFIG, SESSION_TIMEOUT, SetWebhookURL, GetWebhookURL
 from core.middleware import ConcurrencyLimitMiddleware
 from core.database import get_db
 from routers import callback, health
@@ -125,14 +125,8 @@ if __name__ == "__main__":
     
     # 如果提供了webhook URL，设置环境变量
     if args.webhook_url:
-        os.environ['WEBHOOK_URL'] = args.webhook_url
+        SetWebhookURL(args.webhook_url)
         logger.info(f"设置Webhook URL: {args.webhook_url}")
-        
-        # 重新导入配置模块以更新WEBHOOK_URL
-        import importlib
-        import config
-        importlib.reload(config)
-        logger.info(f"更新后的Webhook URL: {config.WEBHOOK_URL}")
     
     # 启动服务器
     uvicorn.run(
diff --git a/config.py b/config.py
index 43c9be0..9848e51 100644
--- a/config.py
+++ b/config.py
@@ -9,15 +9,20 @@ load_dotenv()
 # 项目根目录
 BASE_DIR = Path(__file__).resolve().parent
 
-def get_webhook_url():
-    """动态获取Webhook URL"""
-    return os.getenv(
+# WPS Webhook配置 - 使用函数动态获取
+WEBHOOK_URL = os.getenv(
         "WEBHOOK_URL",
         "https://xz.wps.cn/api/v1/webhook/send?key=da86927e491f2aef4b964223687c2c80"
     )
 
-# WPS Webhook配置 - 使用函数动态获取
-WEBHOOK_URL = get_webhook_url()
+def SetWebhookURL(url: str):
+    """设置Webhook URL"""
+    global WEBHOOK_URL
+    WEBHOOK_URL = url
+
+def GetWebhookURL() -> str:
+    """获取Webhook URL"""
+    return WEBHOOK_URL
 
 # 数据库配置
 DATABASE_PATH = os.getenv("DATABASE_PATH", str(BASE_DIR / "data" / "bot.db"))