K2cr2O1
9f54a98c17
添加抖音视频解析插件，支持自动解析抖音分享链接并提取视频信息。优化现有代码结构，包括： - 重构单例模式实现 - 移除未使用的导入和文件 - 修复性能测试脚本中的异步调用 - 优化消息事件模型中的权限常量定义 - 改进编译脚本的错误处理 - 增强B站解析插件的稳定性 同时清理了多个废弃脚本和临时文件，提升代码可维护性。
119 lines
4.5 KiB
Python
119 lines
4.5 KiB
Python
"""
|
||
Mypyc 编译脚本
|
||
|
||
用于将核心 Python 模块编译为 C 扩展，以提升性能。
|
||
使用方法:
|
||
python setup_mypyc.py build_ext --inplace
|
||
|
||
注意:
|
||
1. 需要安装 C 编译器 (Windows 上需要 Visual Studio Build Tools, Linux 上需要 GCC)。
|
||
2. 编译后的文件 (.pyd 或 .so) 是平台相关的，不能跨平台复制。
|
||
3. 建议在部署的目标环境 (Linux) 上运行此脚本。
|
||
"""
|
||
import os
|
||
import sys
|
||
import subprocess
|
||
|
||
# 基础模块列表
|
||
# 注意：Mypyc 对动态特性支持有限，只选择计算密集或类型明确的模块
|
||
modules = [
|
||
# 工具模块
|
||
'core/utils/json_utils.py', # JSON 处理
|
||
'core/utils/executor.py', # 代码执行引擎
|
||
'core/utils/singleton.py', # 单例模式基类
|
||
'core/utils/exceptions.py', # 自定义异常
|
||
'core/utils/logger.py', # 日志模块
|
||
|
||
# 核心管理模块
|
||
'core/managers/command_manager.py', # 指令匹配和分发
|
||
'core/managers/admin_manager.py', # 管理员管理
|
||
'core/managers/permission_manager.py', # 权限管理
|
||
'core/managers/plugin_manager.py', # 插件管理器
|
||
|
||
# 核心基础模块
|
||
'core/ws.py', # WebSocket 核心
|
||
'core/bot.py', # Bot 核心抽象
|
||
'core/config_loader.py', # 配置加载
|
||
'core/config_models.py', # 配置模型
|
||
'core/permission.py', # 权限枚举
|
||
|
||
# API 基础模块
|
||
'core/api/base.py', # API 基础类
|
||
|
||
# 数据模型（适合编译的高频使用数据类）
|
||
'models/message.py', # 消息段模型
|
||
'models/sender.py', # 发送者模型
|
||
'models/objects.py', # API 响应数据模型
|
||
]
|
||
|
||
# 注意：事件模型文件暂时不编译，因为它们与 mypyc 存在兼容性问题
|
||
# mypyc 对某些数据类特性和继承结构的支持有限，会导致运行时错误
|
||
# event_models = glob.glob('models/events/*.py')
|
||
# event_models = [m for m in event_models if not m.endswith('__init__.py')]
|
||
# modules.extend(event_models)
|
||
|
||
# 确保文件存在
|
||
valid_modules = []
|
||
for m in modules:
|
||
if os.path.exists(m):
|
||
valid_modules.append(m)
|
||
else:
|
||
print(f"Warning: Module {m} not found, skipping.")
|
||
|
||
if not valid_modules:
|
||
print("No valid modules found to compile.")
|
||
sys.exit(1)
|
||
|
||
print(f"Compiling the following modules with mypyc: {valid_modules}")
|
||
|
||
# 使用 mypyc 命令行工具单独编译每个模块，确保位置正确
|
||
success_count = 0
|
||
for module_path in valid_modules:
|
||
print(f"\nCompiling {module_path}...")
|
||
try:
|
||
# 直接调用 mypyc 命令行工具
|
||
result = subprocess.run(
|
||
[sys.executable, '-m', 'mypyc', module_path],
|
||
capture_output=True,
|
||
text=True,
|
||
check=True
|
||
)
|
||
|
||
# 验证编译产物是否在正确位置
|
||
module_name = module_path.replace('.py', '')
|
||
pyd_path = module_name + '.cp314-win_amd64.pyd'
|
||
mypyc_path = module_name + '__mypyc.cp314-win_amd64.pyd'
|
||
|
||
if os.path.exists(pyd_path):
|
||
print(f" ✓ Compiled successfully: {pyd_path}")
|
||
success_count += 1
|
||
else:
|
||
# 检查 build 目录中是否有编译产物
|
||
build_pyd_path = os.path.join('build', 'lib.win-amd64-cpython-314', pyd_path)
|
||
if os.path.exists(build_pyd_path):
|
||
# 如果在 build 目录中，复制到正确位置
|
||
os.makedirs(os.path.dirname(pyd_path), exist_ok=True)
|
||
import shutil
|
||
shutil.copy2(build_pyd_path, pyd_path)
|
||
shutil.copy2(os.path.join('build', 'lib.win-amd64-cpython-314', mypyc_path), mypyc_path)
|
||
print(f" ✓ Compiled successfully (copied from build directory): {pyd_path}")
|
||
success_count += 1
|
||
else:
|
||
print(" ✗ Compiled but cannot find pyd file")
|
||
print(f" Build output:\n{result.stdout[:500]}...")
|
||
except subprocess.CalledProcessError as e:
|
||
print(f" ✗ Compilation failed with exit code {e.returncode}")
|
||
print(f" Error:\n{e.stderr[:500]}...")
|
||
except Exception as e:
|
||
print(f" ✗ Unexpected error: {e}")
|
||
|
||
print("\n--- Compilation Summary ---")
|
||
print(f"Total modules: {len(valid_modules)}")
|
||
print(f"Successfully compiled: {success_count}")
|
||
print(f"Failed: {len(valid_modules) - success_count}")
|
||
|
||
if success_count == 0:
|
||
print("No modules were compiled successfully. Exiting with error.")
|
||
sys.exit(1)
|
||
|