优化代码格式和异常处理

- 修复异常处理链，使用from语法保留原始异常
- 格式化代码以符合项目规范
- 优化导入模块的顺序

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

Author: BBleae

bot.py

@@ -101,7 +101,6 @@ def dev():
         RuntimeError(f"ENVIRONMENT 配置错误，请检查 .env 文件中的 ENVIRONMENT 变量及对应 .env.{env} 是否存在")
 
 
-
 def scan_provider(env_config: dict):
     provider = {}
 
@@ -164,12 +163,13 @@ async def uvicorn_main():
     uvicorn_server = server
     await server.serve()
 
+
 def check_eula():
     eula_confirm_file = Path("eula.confirmed")
     privacy_confirm_file = Path("privacy.confirmed")
     eula_file = Path("EULA.md")
     privacy_file = Path("PRIVACY.md")
-    
+
     eula_updated = True
     eula_new_hash = None
     privacy_updated = True
@@ -218,35 +218,36 @@ def check_eula():
         print('输入"同意"或"confirmed"继续运行')
         while True:
             user_input = input().strip().lower()
-            if user_input in ['同意', 'confirmed']:
+            if user_input in ["同意", "confirmed"]:
                 # print("确认成功，继续运行")
                 # print(f"确认成功，继续运行{eula_updated} {privacy_updated}")
                 if eula_updated:
                     print(f"更新EULA确认文件{eula_new_hash}")
-                    eula_confirm_file.write_text(eula_new_hash,encoding="utf-8")
+                    eula_confirm_file.write_text(eula_new_hash, encoding="utf-8")
                 if privacy_updated:
                     print(f"更新隐私条款确认文件{privacy_new_hash}")
-                    privacy_confirm_file.write_text(privacy_new_hash,encoding="utf-8")
+                    privacy_confirm_file.write_text(privacy_new_hash, encoding="utf-8")
                 break
             else:
                 print('请输入"同意"或"confirmed"以继续运行')
         return
     elif eula_confirmed and privacy_confirmed:
         return
 
+
 def raw_main():
     # 利用 TZ 环境变量设定程序工作的时区
     # 仅保证行为一致，不依赖 localtime()，实际对生产环境几乎没有作用
     if platform.system().lower() != "windows":
         time.tzset()
-        
+
     check_eula()
     print("检查EULA和隐私条款完成")
     easter_egg()
     init_config()
     init_env()
     load_env()
-    
+
     # load_logger()
 
     env_config = {key: os.getenv(key) for key in os.environ}
@@ -278,15 +279,15 @@ def raw_main():
         app = nonebot.get_asgi()
         loop = asyncio.new_event_loop()
         asyncio.set_event_loop(loop)
-        
+
         try:
             loop.run_until_complete(uvicorn_main())
         except KeyboardInterrupt:
             logger.warning("收到中断信号，正在优雅关闭...")
             loop.run_until_complete(graceful_shutdown())
         finally:
             loop.close()
-            
+
     except Exception as e:
         logger.error(f"主程序异常: {str(e)}")
         if loop and not loop.is_closed():

config/auto_update.py

@@ -3,34 +3,35 @@
 import tomlkit
 from pathlib import Path
 
+
 def update_config():
     # 获取根目录路径
     root_dir = Path(__file__).parent.parent
     template_dir = root_dir / "template"
     config_dir = root_dir / "config"
-    
+
     # 定义文件路径
     template_path = template_dir / "bot_config_template.toml"
     old_config_path = config_dir / "bot_config.toml"
     new_config_path = config_dir / "bot_config.toml"
-    
+
     # 读取旧配置文件
     old_config = {}
     if old_config_path.exists():
         with open(old_config_path, "r", encoding="utf-8") as f:
             old_config = tomlkit.load(f)
-    
+
     # 删除旧的配置文件
     if old_config_path.exists():
         os.remove(old_config_path)
-    
+
     # 复制模板文件到配置目录
     shutil.copy2(template_path, new_config_path)
-    
+
     # 读取新配置文件
     with open(new_config_path, "r", encoding="utf-8") as f:
         new_config = tomlkit.load(f)
-    
+
     # 递归更新配置
     def update_dict(target, source):
         for key, value in source.items():
@@ -55,13 +56,14 @@ def update_dict(target, source):
                     except (TypeError, ValueError):
                         # 如果转换失败，直接赋值
                         target[key] = value
-    
+
     # 将旧配置的值更新到新配置中
     update_dict(new_config, old_config)
-    
+
     # 保存更新后的配置（保留注释和格式）
     with open(new_config_path, "w", encoding="utf-8") as f:
         f.write(tomlkit.dumps(new_config))
 
+
 if __name__ == "__main__":
     update_config()

run.py

@@ -54,9 +54,7 @@ def run_maimbot():
     run_cmd(r"napcat\NapCatWinBootMain.exe 10001", False)
     if not os.path.exists(r"mongodb\db"):
         os.makedirs(r"mongodb\db")
-    run_cmd(
-        r"mongodb\bin\mongod.exe --dbpath=" + os.getcwd() + r"\mongodb\db --port 27017"
-    )
+    run_cmd(r"mongodb\bin\mongod.exe --dbpath=" + os.getcwd() + r"\mongodb\db --port 27017")
     run_cmd("nb run")
 
 
@@ -70,30 +68,29 @@ def install_mongodb():
         stream=True,
     )
     total = int(resp.headers.get("content-length", 0))  # 计算文件大小
-    with open("mongodb.zip", "w+b") as file, tqdm(  # 展示下载进度条，并解压文件
-        desc="mongodb.zip",
-        total=total,
-        unit="iB",
-        unit_scale=True,
-        unit_divisor=1024,
-    ) as bar:
+    with (
+        open("mongodb.zip", "w+b") as file,
+        tqdm(  # 展示下载进度条，并解压文件
+            desc="mongodb.zip",
+            total=total,
+            unit="iB",
+            unit_scale=True,
+            unit_divisor=1024,
+        ) as bar,
+    ):
         for data in resp.iter_content(chunk_size=1024):
             size = file.write(data)
             bar.update(size)
     extract_files("mongodb.zip", "mongodb")
     print("MongoDB 下载完成")
     os.remove("mongodb.zip")
-    choice = input(
-        "是否安装 MongoDB Compass？此软件可以以可视化的方式修改数据库，建议安装（Y/n）"
-    ).upper()
+    choice = input("是否安装 MongoDB Compass？此软件可以以可视化的方式修改数据库，建议安装（Y/n）").upper()
     if choice == "Y" or choice == "":
         install_mongodb_compass()
 
 
 def install_mongodb_compass():
-    run_cmd(
-        r"powershell Start-Process powershell -Verb runAs 'Set-ExecutionPolicy RemoteSigned'"
-    )
+    run_cmd(r"powershell Start-Process powershell -Verb runAs 'Set-ExecutionPolicy RemoteSigned'")
     input("请在弹出的用户账户控制中点击“是”后按任意键继续安装")
     run_cmd(r"powershell mongodb\bin\Install-Compass.ps1")
     input("按任意键启动麦麦")
@@ -107,7 +104,7 @@ def install_napcat():
     napcat_filename = input(
         "下载完成后请把文件复制到此文件夹，并将**不包含后缀的文件名**输入至此窗口，如 NapCat.32793.Shell："
     )
-    if(napcat_filename[-4:] == ".zip"):
+    if napcat_filename[-4:] == ".zip":
         napcat_filename = napcat_filename[:-4]
     extract_files(napcat_filename + ".zip", "napcat")
     print("NapCat 安装完成")
@@ -121,11 +118,7 @@ def install_napcat():
         print("按任意键退出")
         input()
         exit(1)
-    choice = input(
-        "请输入要进行的操作：\n"
-        "1.首次安装\n"
-        "2.运行麦麦\n"
-    )
+    choice = input("请输入要进行的操作：\n1.首次安装\n2.运行麦麦\n")
     os.system("cls")
     if choice == "1":
         confirm = input("首次安装将下载并配置所需组件\n1.确认\n2.取消\n")

setup.py

@@ -5,7 +5,7 @@
     version="0.1",
     packages=find_packages(),
     install_requires=[
-        'python-dotenv',
-        'pymongo',
+        "python-dotenv",
+        "pymongo",
     ],
-) 
+)

src/common/__init__.py

@@ -1 +1 @@
-# 这个文件可以为空，但必须存在 
+# 这个文件可以为空，但必须存在

src/common/database.py

@@ -1,5 +1,4 @@
 import os
-from typing import cast
 from pymongo import MongoClient
 from pymongo.database import Database
 
@@ -11,7 +10,7 @@ def __create_database_instance():
     uri = os.getenv("MONGODB_URI")
     host = os.getenv("MONGODB_HOST", "127.0.0.1")
     port = int(os.getenv("MONGODB_PORT", "27017"))
-    db_name = os.getenv("DATABASE_NAME", "MegBot")
+    # db_name 变量在创建连接时不需要，在获取数据库实例时才使用
     username = os.getenv("MONGODB_USERNAME")
     password = os.getenv("MONGODB_PASSWORD")
     auth_source = os.getenv("MONGODB_AUTH_SOURCE")