379 lines
13 KiB
Python
379 lines
13 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
完整修复所有管理器的初始化问题
|
||
|
|
位置: blenderpython/suw_core/test/fix_all_managers.py
|
||
|
|
作者: SUWood Team
|
||
|
|
版本: 1.0.0
|
||
|
|
"""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
|
||
|
|
# 添加项目路径
|
||
|
|
current_dir = os.path.dirname(__file__)
|
||
|
|
suw_core_dir = os.path.dirname(current_dir)
|
||
|
|
blenderpython_dir = os.path.dirname(suw_core_dir)
|
||
|
|
sys.path.insert(0, blenderpython_dir)
|
||
|
|
|
||
|
|
|
||
|
|
def diagnose_manager_issues():
|
||
|
|
"""诊断管理器问题"""
|
||
|
|
print("🔍 诊断管理器初始化问题...")
|
||
|
|
|
||
|
|
try:
|
||
|
|
# 逐个测试管理器导入
|
||
|
|
manager_modules = [
|
||
|
|
'memory_manager',
|
||
|
|
'material_manager',
|
||
|
|
'part_creator',
|
||
|
|
'machining_manager',
|
||
|
|
'selection_manager',
|
||
|
|
'deletion_manager',
|
||
|
|
'hardware_manager',
|
||
|
|
'door_drawer_manager',
|
||
|
|
'dimension_manager',
|
||
|
|
'command_dispatcher'
|
||
|
|
]
|
||
|
|
|
||
|
|
for module_name in manager_modules:
|
||
|
|
try:
|
||
|
|
module = __import__(
|
||
|
|
f'suw_core.{module_name}', fromlist=[module_name])
|
||
|
|
print(f"✅ {module_name} 模块导入成功")
|
||
|
|
|
||
|
|
# 检查是否有对应的管理器类
|
||
|
|
class_names = [attr for attr in dir(module) if attr.endswith(
|
||
|
|
'Manager') or attr.endswith('Creator') or attr.endswith('Dispatcher')]
|
||
|
|
print(f" 发现类: {class_names}")
|
||
|
|
|
||
|
|
# 检查是否有初始化函数
|
||
|
|
init_funcs = [attr for attr in dir(
|
||
|
|
module) if attr.startswith('init_')]
|
||
|
|
print(f" 初始化函数: {init_funcs}")
|
||
|
|
|
||
|
|
# 检查全局实例
|
||
|
|
global_instances = [attr for attr in dir(module) if not attr.startswith(
|
||
|
|
'_') and not callable(getattr(module, attr, None)) and not attr[0].isupper()]
|
||
|
|
print(f" 全局实例: {global_instances}")
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ {module_name} 导入失败: {e}")
|
||
|
|
|
||
|
|
print("\n" + "="*50)
|
||
|
|
|
||
|
|
# 测试 init_all_managers 函数
|
||
|
|
print("🧪 测试 init_all_managers 函数...")
|
||
|
|
from suw_core import init_all_managers
|
||
|
|
|
||
|
|
managers = init_all_managers(None)
|
||
|
|
print(f"📊 init_all_managers 返回: {len(managers)} 个管理器")
|
||
|
|
|
||
|
|
for name, manager in managers.items():
|
||
|
|
if manager is not None:
|
||
|
|
manager_type = getattr(manager, 'manager_type', 'Unknown')
|
||
|
|
if hasattr(manager, 'get_stats') or hasattr(manager, f'get_{name}_stats'):
|
||
|
|
print(f"✅ {name}: 正常 ({type(manager).__name__})")
|
||
|
|
else:
|
||
|
|
print(f"⚠️ {name}: 创建但缺少统计方法 ({type(manager).__name__})")
|
||
|
|
else:
|
||
|
|
print(f"❌ {name}: 未创建")
|
||
|
|
|
||
|
|
return managers
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ 诊断失败: {e}")
|
||
|
|
import traceback
|
||
|
|
traceback.print_exc()
|
||
|
|
return {}
|
||
|
|
|
||
|
|
|
||
|
|
def fix_manager_stats_methods():
|
||
|
|
"""修复管理器统计方法"""
|
||
|
|
print("\n🔧 修复管理器统计方法...")
|
||
|
|
|
||
|
|
# 为每个管理器添加缺失的统计方法
|
||
|
|
missing_stats_fixes = {
|
||
|
|
'memory_manager': '''
|
||
|
|
def get_memory_stats(self) -> Dict[str, Any]:
|
||
|
|
"""获取内存管理器统计信息"""
|
||
|
|
try:
|
||
|
|
stats = {
|
||
|
|
"manager_type": "BlenderMemoryManager",
|
||
|
|
"tracked_objects": len(self.tracked_objects),
|
||
|
|
"tracked_meshes": len(self.tracked_meshes),
|
||
|
|
"tracked_images": len(self.tracked_images),
|
||
|
|
"creation_stats": self.creation_stats.copy(),
|
||
|
|
"blender_available": BLENDER_AVAILABLE
|
||
|
|
}
|
||
|
|
return stats
|
||
|
|
except Exception as e:
|
||
|
|
return {"error": str(e)}
|
||
|
|
''',
|
||
|
|
'material_manager': '''
|
||
|
|
def get_material_stats(self) -> Dict[str, Any]:
|
||
|
|
"""获取材质管理器统计信息"""
|
||
|
|
try:
|
||
|
|
stats = {
|
||
|
|
"manager_type": "MaterialManager",
|
||
|
|
"textures_count": len(self.textures),
|
||
|
|
"material_stats": self.material_stats.copy(),
|
||
|
|
"blender_available": BLENDER_AVAILABLE
|
||
|
|
}
|
||
|
|
return stats
|
||
|
|
except Exception as e:
|
||
|
|
return {"error": str(e)}
|
||
|
|
''',
|
||
|
|
'machining_manager': '''
|
||
|
|
def get_machining_stats(self) -> Dict[str, Any]:
|
||
|
|
"""获取加工管理器统计信息"""
|
||
|
|
try:
|
||
|
|
stats = {
|
||
|
|
"manager_type": "MachiningManager",
|
||
|
|
"machinings_count": len(getattr(self, 'machinings', {})),
|
||
|
|
"creation_stats": getattr(self, 'creation_stats', {}),
|
||
|
|
"blender_available": BLENDER_AVAILABLE
|
||
|
|
}
|
||
|
|
return stats
|
||
|
|
except Exception as e:
|
||
|
|
return {"error": str(e)}
|
||
|
|
''',
|
||
|
|
'selection_manager': '''
|
||
|
|
def get_selection_stats(self) -> Dict[str, Any]:
|
||
|
|
"""获取选择管理器统计信息"""
|
||
|
|
try:
|
||
|
|
stats = {
|
||
|
|
"manager_type": "SelectionManager",
|
||
|
|
"selected_objects": len(getattr(self, 'selected_objects', [])),
|
||
|
|
"blender_available": BLENDER_AVAILABLE
|
||
|
|
}
|
||
|
|
return stats
|
||
|
|
except Exception as e:
|
||
|
|
return {"error": str(e)}
|
||
|
|
''',
|
||
|
|
'deletion_manager': '''
|
||
|
|
def get_deletion_stats(self) -> Dict[str, Any]:
|
||
|
|
"""获取删除管理器统计信息"""
|
||
|
|
try:
|
||
|
|
stats = {
|
||
|
|
"manager_type": "DeletionManager",
|
||
|
|
"deletion_stats": getattr(self, 'deletion_stats', {}),
|
||
|
|
"blender_available": BLENDER_AVAILABLE
|
||
|
|
}
|
||
|
|
return stats
|
||
|
|
except Exception as e:
|
||
|
|
return {"error": str(e)}
|
||
|
|
''',
|
||
|
|
'hardware_manager': '''
|
||
|
|
def get_hardware_stats(self) -> Dict[str, Any]:
|
||
|
|
"""获取五金管理器统计信息"""
|
||
|
|
try:
|
||
|
|
stats = {
|
||
|
|
"manager_type": "HardwareManager",
|
||
|
|
"hardware_count": len(getattr(self, 'hardwares', {})),
|
||
|
|
"blender_available": BLENDER_AVAILABLE
|
||
|
|
}
|
||
|
|
return stats
|
||
|
|
except Exception as e:
|
||
|
|
return {"error": str(e)}
|
||
|
|
''',
|
||
|
|
'door_drawer_manager': '''
|
||
|
|
def get_door_drawer_stats(self) -> Dict[str, Any]:
|
||
|
|
"""获取门抽屉管理器统计信息"""
|
||
|
|
try:
|
||
|
|
stats = {
|
||
|
|
"manager_type": "DoorDrawerManager",
|
||
|
|
"doors_count": len(getattr(self, 'doors', {})),
|
||
|
|
"drawers_count": len(getattr(self, 'drawers', {})),
|
||
|
|
"blender_available": BLENDER_AVAILABLE
|
||
|
|
}
|
||
|
|
return stats
|
||
|
|
except Exception as e:
|
||
|
|
return {"error": str(e)}
|
||
|
|
''',
|
||
|
|
'dimension_manager': '''
|
||
|
|
def get_dimension_stats(self) -> Dict[str, Any]:
|
||
|
|
"""获取尺寸标注管理器统计信息"""
|
||
|
|
try:
|
||
|
|
stats = {
|
||
|
|
"manager_type": "DimensionManager",
|
||
|
|
"dimensions_count": len(getattr(self, 'dimensions', {})),
|
||
|
|
"blender_available": BLENDER_AVAILABLE
|
||
|
|
}
|
||
|
|
return stats
|
||
|
|
except Exception as e:
|
||
|
|
return {"error": str(e)}
|
||
|
|
'''
|
||
|
|
}
|
||
|
|
|
||
|
|
print("📝 需要添加的统计方法:")
|
||
|
|
for manager, code in missing_stats_fixes.items():
|
||
|
|
print(f" {manager}: get_{manager.replace('_manager', '')}_stats")
|
||
|
|
|
||
|
|
print("\n💡 请手动将这些方法添加到对应的管理器类中")
|
||
|
|
return missing_stats_fixes
|
||
|
|
|
||
|
|
|
||
|
|
def create_patched_init_function():
|
||
|
|
"""创建修补的初始化函数"""
|
||
|
|
print("\n🔧 创建修补的初始化函数...")
|
||
|
|
|
||
|
|
patched_init_code = '''
|
||
|
|
def patched_init_all_managers(suw_impl):
|
||
|
|
"""修补版本的管理器初始化函数"""
|
||
|
|
managers = {}
|
||
|
|
|
||
|
|
try:
|
||
|
|
# 尝试初始化每个管理器，失败时使用默认值
|
||
|
|
|
||
|
|
# 材质管理器
|
||
|
|
try:
|
||
|
|
from suw_core.material_manager import MaterialManager
|
||
|
|
managers['material_manager'] = MaterialManager(suw_impl)
|
||
|
|
print("✅ MaterialManager 初始化成功")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ MaterialManager 初始化失败: {e}")
|
||
|
|
managers['material_manager'] = None
|
||
|
|
|
||
|
|
# 部件创建器
|
||
|
|
try:
|
||
|
|
from suw_core.part_creator import PartCreator
|
||
|
|
managers['part_creator'] = PartCreator(suw_impl)
|
||
|
|
print("✅ PartCreator 初始化成功")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ PartCreator 初始化失败: {e}")
|
||
|
|
managers['part_creator'] = None
|
||
|
|
|
||
|
|
# 加工管理器
|
||
|
|
try:
|
||
|
|
from suw_core.machining_manager import MachiningManager
|
||
|
|
managers['machining_manager'] = MachiningManager(suw_impl)
|
||
|
|
print("✅ MachiningManager 初始化成功")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ MachiningManager 初始化失败: {e}")
|
||
|
|
managers['machining_manager'] = None
|
||
|
|
|
||
|
|
# 选择管理器
|
||
|
|
try:
|
||
|
|
from suw_core.selection_manager import SelectionManager
|
||
|
|
managers['selection_manager'] = SelectionManager(suw_impl)
|
||
|
|
print("✅ SelectionManager 初始化成功")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ SelectionManager 初始化失败: {e}")
|
||
|
|
managers['selection_manager'] = None
|
||
|
|
|
||
|
|
# 删除管理器
|
||
|
|
try:
|
||
|
|
from suw_core.deletion_manager import DeletionManager
|
||
|
|
managers['deletion_manager'] = DeletionManager(suw_impl)
|
||
|
|
print("✅ DeletionManager 初始化成功")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ DeletionManager 初始化失败: {e}")
|
||
|
|
managers['deletion_manager'] = None
|
||
|
|
|
||
|
|
# 五金管理器
|
||
|
|
try:
|
||
|
|
from suw_core.hardware_manager import HardwareManager
|
||
|
|
managers['hardware_manager'] = HardwareManager(suw_impl)
|
||
|
|
print("✅ HardwareManager 初始化成功")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ HardwareManager 初始化失败: {e}")
|
||
|
|
managers['hardware_manager'] = None
|
||
|
|
|
||
|
|
# 门抽屉管理器
|
||
|
|
try:
|
||
|
|
from suw_core.door_drawer_manager import DoorDrawerManager
|
||
|
|
managers['door_drawer_manager'] = DoorDrawerManager(suw_impl)
|
||
|
|
print("✅ DoorDrawerManager 初始化成功")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ DoorDrawerManager 初始化失败: {e}")
|
||
|
|
managers['door_drawer_manager'] = None
|
||
|
|
|
||
|
|
# 尺寸标注管理器
|
||
|
|
try:
|
||
|
|
from suw_core.dimension_manager import DimensionManager
|
||
|
|
managers['dimension_manager'] = DimensionManager(suw_impl)
|
||
|
|
print("✅ DimensionManager 初始化成功")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ DimensionManager 初始化失败: {e}")
|
||
|
|
managers['dimension_manager'] = None
|
||
|
|
|
||
|
|
# 命令分发器
|
||
|
|
try:
|
||
|
|
from suw_core.command_dispatcher import CommandDispatcher
|
||
|
|
managers['command_dispatcher'] = CommandDispatcher(suw_impl)
|
||
|
|
print("✅ CommandDispatcher 初始化成功")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ CommandDispatcher 初始化失败: {e}")
|
||
|
|
managers['command_dispatcher'] = None
|
||
|
|
|
||
|
|
success_count = len([m for m in managers.values() if m is not None])
|
||
|
|
print(f"📊 管理器初始化完成: {success_count}/{len(managers)} 成功")
|
||
|
|
|
||
|
|
return managers
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ 管理器初始化总体失败: {e}")
|
||
|
|
return managers
|
||
|
|
|
||
|
|
# 替换全局初始化函数
|
||
|
|
import suw_core
|
||
|
|
suw_core.init_all_managers = patched_init_all_managers
|
||
|
|
'''
|
||
|
|
|
||
|
|
return patched_init_code
|
||
|
|
|
||
|
|
|
||
|
|
def apply_emergency_patch():
|
||
|
|
"""应用紧急修补"""
|
||
|
|
print("\n🚑 应用紧急修补...")
|
||
|
|
|
||
|
|
try:
|
||
|
|
# 执行修补代码
|
||
|
|
patched_code = create_patched_init_function()
|
||
|
|
exec(patched_code)
|
||
|
|
|
||
|
|
print("✅ 紧急修补已应用")
|
||
|
|
|
||
|
|
# 重新测试
|
||
|
|
print("\n🧪 重新测试管理器初始化...")
|
||
|
|
import suw_core
|
||
|
|
managers = suw_core.init_all_managers(None)
|
||
|
|
|
||
|
|
print(
|
||
|
|
f"📊 修补后结果: {len([m for m in managers.values() if m is not None])}/{len(managers)} 成功")
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ 紧急修补失败: {e}")
|
||
|
|
import traceback
|
||
|
|
traceback.print_exc()
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
"""主函数"""
|
||
|
|
print("🔧 开始完整修复所有管理器...")
|
||
|
|
print("="*60)
|
||
|
|
|
||
|
|
# 1. 诊断问题
|
||
|
|
managers = diagnose_manager_issues()
|
||
|
|
|
||
|
|
# 2. 修复统计方法
|
||
|
|
fix_manager_stats_methods()
|
||
|
|
|
||
|
|
# 3. 应用紧急修补
|
||
|
|
apply_emergency_patch()
|
||
|
|
|
||
|
|
print("\n" + "="*60)
|
||
|
|
print("🎯 修复总结:")
|
||
|
|
print("1. 请手动修改各个管理器文件的构造函数")
|
||
|
|
print("2. 请添加缺失的统计方法")
|
||
|
|
print("3. 紧急修补已临时应用")
|
||
|
|
print("4. 重新运行客户端测试效果")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|