From 0f492ed4032b9fe279c469bdb1acd71a57ee3e6d Mon Sep 17 00:00:00 2001
From: Rikki <rikki@member.fsf.org>
Date: Sun, 9 Mar 2025 02:39:58 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=20BASE=5FURL/KEY=20?=
 =?UTF-8?q?=E7=BB=84=E5=90=88=E6=A3=80=E6=9F=A5=E4=B8=AD=E8=A2=AB=20GPG=5F?=
 =?UTF-8?q?KEY=20=E5=B9=B2=E6=89=B0=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bot.py | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/bot.py b/bot.py
index 716d0f4ba..5b25d75f9 100644
--- a/bot.py
+++ b/bot.py
@@ -6,7 +6,7 @@ from dotenv import load_dotenv
 from loguru import logger
 
 # 获取所有环境变量
-env_config = {key: os.getenv(key) for key in os.environ}
+env_mask = {key: os.getenv(key) for key in os.environ}
 
 # 设置基础配置
 base_config = {
@@ -83,9 +83,17 @@ def load_env():
         logger.error(f"ENVIRONMENT 配置错误，请检查 .env 文件中的 ENVIRONMENT 变量及对应 .env.{env} 是否存在")
         exit(1)
 
-provider = {}
 
-def scan_provider():
+
+def scan_provider(env_config: dict):
+    provider = {}
+    
+    # 利用未初始化 env 时获取的 env_mask 来对新的环境变量集去重
+    # 避免 GPG_KEY 这样的变量干扰检查
+    for key in env_config:
+        if key in env_mask:
+            del env_config[key]
+
     # 遍历 env_config 的所有键
     for key in env_config:
         # 检查键是否符合 {provider}_BASE_URL 或 {provider}_KEY 的格式
@@ -106,13 +114,19 @@ def scan_provider():
     # 检查每个 provider 是否同时存在 url 和 key
     for provider_name, config in provider.items():
         if config["url"] is None or config["key"] is None:
+            logger.error(
+                f"provider 内容：{config}\n"
+                f"env_config 内容：{env_config}"
+            )
             raise ValueError(f"请检查 '{provider_name}' 提供商配置是否丢失 BASE_URL 或 KEY 环境变量")
 
 if __name__ == "__main__":
     easter_egg()
     init_config()
     init_env()
-    scan_provider()
+
+    env_config = {key: os.getenv(key) for key in os.environ}
+    scan_provider(env_config)
 
     # 合并配置
     nonebot.init(**base_config, **env_config)