fix

python_core/main.py

@@ -0,0 +1,208 @@
+#!/usr/bin/env python3
+"""
+MixVideo V2 Python Core - Main Entry Point
+用于PyInstaller打包的主入口文件
+"""
+
+import sys
+import os
+import argparse
+import json
+import traceback
+from pathlib import Path
+
+# 添加当前目录到Python路径，确保可以导入python_core模块
+current_dir = Path(__file__).parent
+sys.path.insert(0, str(current_dir))
+
+# 导入核心模块
+try:
+    from utils.logger import get_logger
+    from utils.jsonrpc import create_response_handler, create_progress_reporter
+    from config import get_config
+except ImportError as e:
+    print(f"ERROR: Failed to import core modules: {e}", file=sys.stderr)
+    sys.exit(1)
+
+# 模块映射表
+MODULE_MAP = {
+    # 模板管理
+    'template_manager': 'services.template_manager',
+    'template': 'services.template_manager',
+    
+    # 项目管理
+    'project_manager': 'services.project_manager',
+    'project': 'services.project_manager',
+    
+    # 媒体管理
+    'media_manager': 'services.media_manager',
+    'media': 'services.media_manager',
+    
+    # 音频管理
+    'audio_manager': 'services.audio_manager',
+    'audio': 'services.audio_manager',
+    
+    # 模特管理
+    'model_manager': 'services.model_manager',
+    'model': 'services.model_manager',
+    
+    # 资源分类管理
+    'resource_category_manager': 'services.resource_category_manager',
+    'category': 'services.resource_category_manager',
+    
+    # 文件管理
+    'file_manager': 'services.file_manager',
+    'file': 'services.file_manager',
+    
+    # AI视频生成
+    'ai_video': 'ai_video.video_generator',
+    'video_generator': 'ai_video.video_generator',
+    
+    # 音频处理
+    'audio_processing': 'audio_processing.core',
+    
+    # 视频处理
+    'video_processing': 'video_processing.core',
+}
+
+def setup_logging():
+    """设置日志"""
+    try:
+        logger = get_logger(__name__)
+        return logger
+    except Exception as e:
+        print(f"WARNING: Failed to setup logging: {e}", file=sys.stderr)
+        return None
+
+def load_module_function(module_name, function_name):
+    """动态加载模块和函数"""
+    try:
+        # 解析模块路径
+        if module_name in MODULE_MAP:
+            module_path = MODULE_MAP[module_name]
+        else:
+            module_path = module_name
+        
+        # 导入模块
+        module = __import__(f"python_core.{module_path}", fromlist=[function_name])
+        
+        # 获取函数
+        if hasattr(module, function_name):
+            return getattr(module, function_name)
+        else:
+            raise AttributeError(f"Function '{function_name}' not found in module '{module_path}'")
+            
+    except ImportError as e:
+        raise ImportError(f"Failed to import module '{module_path}': {e}")
+    except Exception as e:
+        raise Exception(f"Failed to load function '{function_name}' from '{module_path}': {e}")
+
+def execute_action(module_name, action, params=None, progress_callback=None):
+    """执行指定的动作"""
+    logger = setup_logging()
+    
+    try:
+        if logger:
+            logger.info(f"Executing action: {module_name}.{action}")
+        
+        # 加载函数
+        func = load_module_function(module_name, action)
+        
+        # 准备参数
+        kwargs = {}
+        if params:
+            kwargs.update(params)
+        
+        # 添加进度回调
+        if progress_callback:
+            kwargs['progress_callback'] = progress_callback
+        
+        # 执行函数
+        result = func(**kwargs)
+        
+        if logger:
+            logger.info(f"Action completed successfully: {module_name}.{action}")
+        
+        return result
+        
+    except Exception as e:
+        error_msg = f"Failed to execute {module_name}.{action}: {str(e)}"
+        if logger:
+            logger.error(error_msg)
+            logger.error(traceback.format_exc())
+        raise Exception(error_msg)
+
+def parse_arguments():
+    """解析命令行参数"""
+    parser = argparse.ArgumentParser(description='MixVideo V2 Python Core')
+    
+    parser.add_argument('--module', '-m', required=True,
+                       help='Module name (e.g., template_manager, project_manager)')
+    
+    parser.add_argument('--action', '-a', required=True,
+                       help='Action to execute (e.g., get_templates, batch_import)')
+    
+    parser.add_argument('--params', '-p', type=str,
+                       help='Parameters as JSON string')
+    
+    parser.add_argument('--config', '-c', type=str,
+                       help='Configuration file path')
+    
+    parser.add_argument('--verbose', '-v', action='store_true',
+                       help='Enable verbose logging')
+    
+    parser.add_argument('--version', action='version', version='MixVideo V2 Python Core 2.0.0')
+    
+    return parser.parse_args()
+
+def main():
+    """主函数"""
+    try:
+        # 解析参数
+        args = parse_arguments()
+        
+        # 设置日志级别
+        if args.verbose:
+            os.environ['LOG_LEVEL'] = 'DEBUG'
+        
+        # 加载配置
+        if args.config:
+            os.environ['CONFIG_FILE'] = args.config
+        
+        # 解析参数
+        params = {}
+        if args.params:
+            try:
+                params = json.loads(args.params)
+            except json.JSONDecodeError as e:
+                print(f"ERROR: Invalid JSON parameters: {e}", file=sys.stderr)
+                sys.exit(1)
+        
+        # 创建响应处理器
+        send_response = create_response_handler()
+        send_progress = create_progress_reporter()
+        
+        # 创建进度回调
+        def progress_callback(step, progress, message, details=None):
+            send_progress(step, progress, message, details or {})
+        
+        # 执行动作
+        result = execute_action(args.module, args.action, params, progress_callback)
+        
+        # 发送成功响应
+        send_response(True, "操作完成", result)
+        
+    except KeyboardInterrupt:
+        print("ERROR: Operation cancelled by user", file=sys.stderr)
+        sys.exit(130)
+    except Exception as e:
+        # 发送错误响应
+        try:
+            send_response = create_response_handler()
+            send_response(False, str(e), None)
+        except:
+            print(f"ERROR: {str(e)}", file=sys.stderr)
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()