From 146626dd9a515ec458fc0cff32e13f1cd356107f Mon Sep 17 00:00:00 2001
From: root <root@sker>
Date: Sun, 13 Jul 2025 11:48:32 +0800
Subject: [PATCH] fix

---
 python_core/cli/cli.py             |   2 +
 python_core/cli/commands/system.py | 274 +++++++++++++++++++++++++++++
 python_core/config.py              |   2 +-
 session.json                       |   8 +
 4 files changed, 285 insertions(+), 1 deletion(-)
 create mode 100644 python_core/cli/commands/system.py
 create mode 100644 session.json

diff --git a/python_core/cli/cli.py b/python_core/cli/cli.py
index 9eb1703..db7a7fe 100644
--- a/python_core/cli/cli.py
+++ b/python_core/cli/cli.py
@@ -13,6 +13,7 @@ from python_core.cli.commands.template import template_app
 from python_core.cli.commands.resource_category import resource_category_app
 from python_core.cli.commands.auth import auth_app
 from python_core.cli.commands.model import model_app
+from python_core.cli.commands.system import system_app
 
 app = typer.Typer(
     name="mixvideo",
@@ -37,6 +38,7 @@ app.add_typer(template_app, name="template")
 app.add_typer(resource_category_app, name="resource-category")
 app.add_typer(auth_app, name="auth")
 app.add_typer(model_app, name="model")
+app.add_typer(system_app, name="system")
 
 @app.command()
 def init():
diff --git a/python_core/cli/commands/system.py b/python_core/cli/commands/system.py
new file mode 100644
index 0000000..d315f2c
--- /dev/null
+++ b/python_core/cli/commands/system.py
@@ -0,0 +1,274 @@
+"""
+系统管理 CLI 命令
+"""
+
+import json
+import uuid
+from pathlib import Path
+from typing import Optional
+import typer
+from python_core.utils.jsonrpc_enhanced import create_response_handler
+from python_core.database.user_postgres import user_table
+from python_core.database.resource_category_postgres import resource_category_table
+from python_core.config import settings
+from python_core.utils.logger import setup_logger
+
+logger = setup_logger(__name__)
+
+# 创建系统应用
+system_app = typer.Typer(help="系统管理命令")
+
+
+@system_app.command("init")
+def init_system(
+    force: bool = typer.Option(False, "--force", help="强制重新初始化"),
+    verbose: bool = typer.Option(False, "--verbose", "-v", help="详细输出"),
+    json_output: bool = typer.Option(True, "--json", help="JSON格式输出")
+):
+    """初始化系统，创建默认数据"""
+    response = create_response_handler()
+    
+    try:
+        # 检查是否已经初始化
+        session_file = Path("session.json")
+        if session_file.exists() and not force:
+            with open(session_file, 'r', encoding='utf-8') as f:
+                session_data = json.load(f)
+            
+            response.success({
+                'already_initialized': True,
+                'session_data': session_data,
+                'message': '系统已经初始化，使用 --force 强制重新初始化'
+            })
+            return
+        
+        # 创建默认用户
+        default_user_id = str(uuid.uuid4())
+        default_access_token = str(uuid.uuid4())
+        
+        # 创建 session.json
+        session_data = {
+            'user_id': default_user_id,
+            'access_token': default_access_token,
+            'username': 'default',
+            'email': 'default@mixvideo.com',
+            'initialized_at': '2025-07-13T11:43:00.000000',
+            'app_version': settings.app_version
+        }
+        
+        with open(session_file, 'w', encoding='utf-8') as f:
+            json.dump(session_data, f, indent=2, ensure_ascii=False)
+        
+        # 创建默认资源分类
+        default_categories = [
+            {
+                'title': '开场介绍',
+                'ai_prompt': '视频开头的介绍部分，包括标题、logo等',
+                'color': '#FF6B6B'
+            },
+            {
+                'title': '主要内容',
+                'ai_prompt': '视频的核心内容部分',
+                'color': '#4ECDC4'
+            },
+            {
+                'title': '转场过渡',
+                'ai_prompt': '视频片段之间的过渡效果',
+                'color': '#45B7D1'
+            },
+            {
+                'title': '结尾总结',
+                'ai_prompt': '视频结尾的总结和结束语',
+                'color': '#96CEB4'
+            },
+            {
+                'title': '背景音乐',
+                'ai_prompt': '视频的背景音乐和音效',
+                'color': '#FFEAA7'
+            },
+            {
+                'title': '字幕文本',
+                'ai_prompt': '视频中的字幕和文字说明',
+                'color': '#DDA0DD'
+            },
+            {
+                'title': '特效动画',
+                'ai_prompt': '视频中的特效和动画效果',
+                'color': '#98D8C8'
+            },
+            {
+                'title': '产品展示',
+                'ai_prompt': '产品或服务的展示片段',
+                'color': '#F7DC6F'
+            }
+        ]
+        
+        created_categories = []
+        for category_data in default_categories:
+            try:
+                from python_core.database.types import ResourceCategory
+                category = ResourceCategory(
+                    id="",  # 将由数据库生成
+                    title=category_data['title'],
+                    ai_prompt=category_data['ai_prompt'],
+                    color=category_data['color'],
+                    is_active=True,
+                    is_cloud=False,
+                    user_id='default',
+                    created_at="",
+                    updated_at=""
+                )
+                category_id = resource_category_table.create_category(category)
+                if category_id:
+                    created_categories.append(category_data['title'])
+                    logger.info(f"Created default category: {category_data['title']}")
+            except Exception as e:
+                logger.warning(f"Failed to create category {category_data['title']}: {e}")
+        
+        # 创建必要的目录
+        directories = [
+            'templates',
+            'projects',
+            'uploads',
+            'temp',
+            'logs'
+        ]
+        
+        created_dirs = []
+        for dir_name in directories:
+            dir_path = Path(dir_name)
+            if not dir_path.exists():
+                dir_path.mkdir(parents=True, exist_ok=True)
+                created_dirs.append(str(dir_path))
+                logger.info(f"Created directory: {dir_path}")
+        
+        response.success({
+            'initialized': True,
+            'session_file': str(session_file),
+            'session_data': session_data,
+            'created_categories': created_categories,
+            'created_directories': created_dirs,
+            'message': f'系统初始化成功，创建了 {len(created_categories)} 个默认分类和 {len(created_dirs)} 个目录'
+        })
+        
+    except Exception as e:
+        logger.error(f"系统初始化失败: {e}")
+        response.error(-32603, f"系统初始化失败: {str(e)}")
+
+
+@system_app.command("status")
+def system_status(
+    verbose: bool = typer.Option(False, "--verbose", "-v", help="详细输出"),
+    json_output: bool = typer.Option(True, "--json", help="JSON格式输出")
+):
+    """检查系统状态"""
+    response = create_response_handler()
+    
+    try:
+        # 检查数据库连接
+        db_status = "connected"
+        try:
+            # 测试各个表的连接
+            user_count = user_table.get_user_count()
+            category_count = resource_category_table.get_category_count()
+        except Exception as e:
+            db_status = f"error: {str(e)}"
+            user_count = 0
+            category_count = 0
+        
+        # 检查 session.json
+        session_file = Path("session.json")
+        session_status = "exists" if session_file.exists() else "missing"
+        session_data = None
+        if session_file.exists():
+            try:
+                with open(session_file, 'r', encoding='utf-8') as f:
+                    session_data = json.load(f)
+            except Exception as e:
+                session_status = f"error: {str(e)}"
+        
+        # 检查目录
+        directories = ['templates', 'projects', 'uploads', 'temp', 'logs']
+        dir_status = {}
+        for dir_name in directories:
+            dir_path = Path(dir_name)
+            dir_status[dir_name] = {
+                'exists': dir_path.exists(),
+                'is_dir': dir_path.is_dir() if dir_path.exists() else False,
+                'path': str(dir_path.absolute())
+            }
+        
+        response.success({
+            'database': {
+                'status': db_status,
+                'user_count': user_count,
+                'category_count': category_count
+            },
+            'session': {
+                'status': session_status,
+                'data': session_data
+            },
+            'directories': dir_status,
+            'app_version': settings.app_version,
+            'message': f'系统状态检查完成 - 数据库: {db_status}, 会话: {session_status}'
+        })
+        
+    except Exception as e:
+        logger.error(f"系统状态检查失败: {e}")
+        response.error(-32603, f"系统状态检查失败: {str(e)}")
+
+
+@system_app.command("reset")
+def reset_system(
+    confirm: bool = typer.Option(False, "--confirm", help="确认重置系统"),
+    keep_users: bool = typer.Option(True, "--keep-users", help="保留用户数据"),
+    verbose: bool = typer.Option(False, "--verbose", "-v", help="详细输出"),
+    json_output: bool = typer.Option(True, "--json", help="JSON格式输出")
+):
+    """重置系统（危险操作）"""
+    response = create_response_handler()
+    
+    if not confirm:
+        response.error(-32602, "请使用 --confirm 参数确认重置操作")
+        return
+    
+    try:
+        reset_actions = []
+        
+        # 删除 session.json
+        session_file = Path("session.json")
+        if session_file.exists():
+            session_file.unlink()
+            reset_actions.append("删除 session.json")
+        
+        # 清理分类数据
+        try:
+            categories = resource_category_table.get_all_resource_categories()
+            for category in categories:
+                resource_category_table.delete_resource_category(category.id, hard_delete=True)
+            reset_actions.append(f"删除 {len(categories)} 个分类")
+        except Exception as e:
+            logger.warning(f"清理分类数据失败: {e}")
+        
+        # 可选：清理用户数据
+        if not keep_users:
+            try:
+                # 这里可以添加清理用户数据的逻辑
+                reset_actions.append("清理用户数据")
+            except Exception as e:
+                logger.warning(f"清理用户数据失败: {e}")
+        
+        response.success({
+            'reset': True,
+            'actions': reset_actions,
+            'keep_users': keep_users,
+            'message': f'系统重置完成，执行了 {len(reset_actions)} 个操作'
+        })
+        
+    except Exception as e:
+        logger.error(f"系统重置失败: {e}")
+        response.error(-32603, f"系统重置失败: {str(e)}")
+
+
+if __name__ == "__main__":
+    system_app()
diff --git a/python_core/config.py b/python_core/config.py
index 434395d..dee5a7e 100644
--- a/python_core/config.py
+++ b/python_core/config.py
@@ -18,7 +18,7 @@ except ImportError:
 project_root = Path(__file__).parent.parent
 class Settings(BaseSettings):
     """Application settings with environment variable support."""
-    db: str = "postgres://mixvideo_user:mixvideo_password@192.168.0.126:5433/mixvideo"
+    db: str = "postgres://mixvideo_user:mixvideo_password@localhost:5432/mixvideo"
     # Application Info
     app_name: str = "MixVideo V2"
     app_version: str = "2.0.0"
diff --git a/session.json b/session.json
new file mode 100644
index 0000000..31bdf7c
--- /dev/null
+++ b/session.json
@@ -0,0 +1,8 @@
+{
+  "user_id": "dd07e880-084c-45d4-a4e4-0b0d985061e3",
+  "access_token": "53253960-dc11-42b7-b7aa-3d01b1e496b3",
+  "username": "default",
+  "email": "default@mixvideo.com",
+  "initialized_at": "2025-07-13T11:43:00.000000",
+  "app_version": "2.0.0"
+}
\ No newline at end of file