#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ AI驱动软件著作权申请材料生成系统 - 统一入口脚本 版本: 1.0 这是系统的主入口脚本，提供统一的命令行界面来访问所有功能 """ import sys import argparse import subprocess from pathlib import Path class Colors: """终端颜色定义""" RED = '\033[0;31m' GREEN = '\033[0;32m' YELLOW = '\033[1;33m' BLUE = '\033[0;34m' PURPLE = '\033[0;35m' CYAN = '\033[0;36m' NC = '\033[0m' def print_colored(color, message): """打印带颜色的消息""" print(f"{color}{message}{Colors.NC}") def print_header(): """打印系统标题""" print_colored(Colors.CYAN, "="*70) print_colored(Colors.PURPLE, "🤖 AI驱动软件著作权申请材料生成系统") print_colored(Colors.BLUE, " 统一管理工具 v1.0") print_colored(Colors.CYAN, "="*70) def get_script_path(): """获取脚本根目录""" return Path(__file__).parent.absolute() def run_command(command, description="执行命令"): """运行命令并返回结果""" print_colored(Colors.BLUE, f"🔄 {description}...") try: result = subprocess.run(command, shell=True, check=True, capture_output=False, text=True) return result.returncode == 0 except subprocess.CalledProcessError as e: print_colored(Colors.RED, f"❌ 命令执行失败: {e}") return False def init_project(args): """初始化新项目""" script_path = get_script_path() / "scripts" / "init" / "init_project.py" if not args.name: print_colored(Colors.RED, "❌ 请提供项目名称") print("用法: ai-copyright.py init <项目名称>") return False cmd = f"python3 {script_path} {args.name}" if args.force: cmd += " --force" return run_command(cmd, f"初始化项目 '{args.name}'") def generate_code(args): """生成源代码""" script_path = get_script_path() / "scripts" / "generators" if args.type == "all": script = script_path / "generate_all_sourcecode.py" desc = "生成所有源代码" elif args.type == "frontend": script = script_path / "generate_frontend_sourcecode.py" desc = "生成前端源代码" elif args.type == "backend": script = script_path / "generate_backend_sourcecode.py" desc = "生成后端源代码" else: print_colored(Colors.RED, "❌ 无效的生成类型") return False return run_command(f"python3 {script}", desc) def check_project(args): """检查项目""" script_path = get_script_path() / "scripts" / "validators" / "check_project.py" cmd = f"python3 {script_path}" if args.quick: cmd += " --quick" if args.path: cmd += f" {args.path}" return run_command(cmd, "检查项目完整性") def run_tests(args): """运行测试""" script_path = get_script_path() / "scripts" / "validators" / "run_tests.py" cmd = f"python3 {script_path}" if args.path: cmd += f" {args.path}" return run_command(cmd, "运行自动化测试") def validate_frontend(_args): """验证前端页面""" script_path = get_script_path() / "scripts" / "validators" / "validate_frontend_pages.py" return run_command(f"python3 {script_path}", "验证前端页面完整性") def show_status(_args): """显示项目状态""" print_colored(Colors.CYAN, "\n📊 项目状态概览") print("-" * 50) # 检查配置文件 config_file = get_script_path() / "ai-copyright-config.json" template_file = get_script_path() / "config" / "ai-copyright-config.example.json" if config_file.exists(): print_colored(Colors.GREEN, "✅ 项目配置文件存在") elif template_file.exists(): print_colored(Colors.YELLOW, "⚠️ 使用模板配置文件，请复制并自定义") print(f" cp {template_file} {config_file}") else: print_colored(Colors.RED, "❌ 配置文件缺失") # 检查关键目录 dirs_to_check = [ ("requires_docs", "输入文档目录"), ("output_docs", "输出文档目录"), ("output_sourcecode", "生成代码目录"), ("specs_docs", "规范文档目录"), ("system_prompts", "AI提示词目录") ] for dir_name, desc in dirs_to_check: dir_path = get_script_path() / dir_name if dir_path.exists(): print_colored(Colors.GREEN, f"✅ {desc} 存在") else: print_colored(Colors.RED, f"❌ {desc} 缺失") return True def main(): """主函数""" parser = argparse.ArgumentParser( description='AI驱动软件著作权申请材料生成系统 - 统一管理工具', formatter_class=argparse.RawDescriptionHelpFormatter, epilog=""" 示例用法: %(prog)s init "我的项目" # 初始化新项目 %(prog)s generate all # 生成所有源代码 %(prog)s generate frontend # 生成前端代码 %(prog)s check --quick # 快速检查项目 %(prog)s test # 运行自动化测试 %(prog)s validate-frontend # 验证前端页面 %(prog)s status # 显示项目状态 """ ) subparsers = parser.add_subparsers(dest='command', help='可用命令') # init 命令 init_parser = subparsers.add_parser('init', help='初始化新项目') init_parser.add_argument('name', help='项目名称') init_parser.add_argument('--force', '-f', action='store_true', help='强制覆盖现有目录') # generate 命令 gen_parser = subparsers.add_parser('generate', help='生成源代码') gen_parser.add_argument('type', choices=['all', 'frontend', 'backend'], help='生成类型') # check 命令 check_parser = subparsers.add_parser('check', help='检查项目完整性') check_parser.add_argument('--quick', '-q', action='store_true', help='快速检查') check_parser.add_argument('path', nargs='?', help='项目路径') # test 命令 test_parser = subparsers.add_parser('test', help='运行自动化测试') test_parser.add_argument('path', nargs='?', help='项目路径') # validate-frontend 命令 subparsers.add_parser('validate-frontend', help='验证前端页面完整性') # status 命令 subparsers.add_parser('status', help='显示项目状态') args = parser.parse_args() print_header() if not args.command: print_colored(Colors.YELLOW, "\n⚠️ 未指定命令，显示帮助信息:") parser.print_help() return 0 # 执行对应命令 command_map = { 'init': init_project, 'generate': generate_code, 'check': check_project, 'test': run_tests, 'validate-frontend': validate_frontend, 'status': show_status } func = command_map.get(args.command) if func: success = func(args) if success: print_colored(Colors.GREEN, f"\n✅ 命令 '{args.command}' 执行成功") else: print_colored(Colors.RED, f"\n❌ 命令 '{args.command}' 执行失败") return 0 if success else 1 else: print_colored(Colors.RED, f"❌ 未知命令: {args.command}") return 1 if __name__ == "__main__": sys.exit(main())