Update api.txt

api.txt

@@ -18,6 +18,53 @@ from cryptography.fernet import Fernet
 from cryptography.hazmat.primitives import hashes
 from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
 
+
+
+# ==============================================================================
+# 依赖检查与自动安装 (Dependency Check and Auto-Installation)
+# ==============================================================================
+import sys
+import subprocess
+import importlib.util
+
+print("--- 正在检查并安装必要的依赖库 ---")
+
+# 定义需要检查和安装的库列表
+# 格式为 '库名': '导入时使用的模块名'
+required_packages = {
+    'fastapi': 'fastapi',
+    'uvicorn': 'uvicorn',
+    'httpx': 'httpx',
+    'websockets': 'websockets',
+    'loguru': 'loguru',
+    'cryptography': 'cryptography',
+}
+
+def install_package(package_name):
+    """使用 pip 安装指定的库"""
+    try:
+        print(f"正在尝试安装 {package_name}...")
+        # 使用 sys.executable 确保使用当前 Python 解释器对应的 pip
+        subprocess.check_call([sys.executable, "-m", "pip", "install", package_name])
+        print(f"✅ 成功安装 {package_name}")
+    except subprocess.CalledProcessError as e:
+        print(f"❌ 安装 {package_name} 失败: {e}")
+        # 根据需要可以决定是否在安装失败时退出程序
+        # sys.exit(1)
+
+# 遍历并检查每个库
+for package_name, module_name in required_packages.items():
+    if importlib.util.find_spec(module_name) is None:
+        print(f"⚠️ 未找到库: {package_name}。")
+        install_package(package_name)
+    else:
+        print(f"✔️ 库 {package_name} 已安装。")
+
+print("--- 所有依赖库均已准备就绪 ---")
+print("\n" * 2)
+
+
+
 app = FastAPI()
 
 # 加密配置