#!/usr/bin/env python3 """ 人物档案生成脚本 根据模板生成结构化的人物档案markdown文件 """ import os import sys import json from datetime import datetime from pathlib import Path import logging from typing import Optional, Dict, List, Tuple # 配置日志 logging.basicConfig(level=logging.INFO, format='%(levelname)s: %(message)s') logger = logging.getLogger(__name__) class CharacterProfileGenerator: """人物档案生成器""" def __init__(self, template_type="standard", workspace=None): """初始化生成器 Args: template_type: 模板类型，可选值: 'protagonist', 'antagonist', 'supporting', 'standard' workspace: 工作目录路径，启用LoreBible管理功能 """ self.template_type = template_type self.templates = self._load_templates() self.workspace = workspace self.enable_enhanced_features = workspace is not None if self.enable_enhanced_features: logger.info(f"启用增强功能，工作目录: {workspace}") # 延迟导入，避免循环依赖 self.lore_bible_manager = None self.conflict_detector = None self.profile_session = None def _init_enhanced_components(self): """初始化增强组件""" if not self.enable_enhanced_features: return try: # 延迟导入 from lore_bible_manager import LoreBibleManager from conflict_detector import ConflictDetector from profile_session import ProfileSession, SessionConfig # 初始化管理器 self.lore_bible_manager = LoreBibleManager(self.workspace) # 验证并创建目录结构 is_valid, missing_dirs = self.lore_bible_manager.validate_directory_structure() if not is_valid: logger.warning(f"目录结构不完整，缺失: {missing_dirs}") logger.info("正在创建缺失目录...") if self.lore_bible_manager.create_directory_structure(): logger.info("目录结构创建成功") else: logger.error("创建目录结构失败") # 初始化冲突检测器 self.conflict_detector = ConflictDetector() # 配置会话（稍后在generate_markdown中创建） self.session_config = SessionConfig( workspace=self.workspace, character_name="", # 稍后设置 template_type=self.template_type, enable_validation=True, enable_conflict_check=True, require_confirmation=True ) logger.info("增强组件初始化完成") except ImportError as e: logger.error(f"导入增强组件失败: {e}") logger.error("请确保lore_bible_manager.py、conflict_detector.py、profile_session.py在脚本目录中") self.enable_enhanced_features = False except Exception as e: logger.error(f"初始化增强组件失败: {e}") self.enable_enhanced_features = False def _load_templates(self): """加载模板配置""" templates = { "protagonist": { "name": "主角模板", "sections": [ {"id": "basic", "title": "基本信息", "required": True}, {"id": "appearance", "title": "外貌特征", "required": True}, {"id": "personality", "title": "性格特点", "required": True}, {"id": "background", "title": "背景故事", "required": True}, {"id": "motivation", "title": "动机层次", "required": True}, {"id": "relationships", "title": "人物关系", "required": True}, {"id": "development", "title": "故事发展", "required": True}, {"id": "core_identity", "title": "核心身份", "required": False}, {"id": "notes", "title": "创作笔记", "required": False} ] }, "antagonist": { "name": "反派模板", "sections": [ {"id": "basic", "title": "基本信息", "required": True}, {"id": "appearance", "title": "外貌特征", "required": True}, {"id": "personality", "title": "性格特点", "required": True}, {"id": "core_belief", "title": "核心理念", "required": True}, {"id": "motivation", "title": "动机发展", "required": True}, {"id": "mirror", "title": "镜像对比", "required": True}, {"id": "resources", "title": "资源能力", "required": True}, {"id": "development", "title": "故事发展", "required": True} ] }, "supporting": { "name": "配角模板", "sections": [ {"id": "basic", "title": "基本定位", "required": True}, {"id": "identity", "title": "独立身份", "required": True}, {"id": "function", "title": "功能性设计", "required": True}, {"id": "relationships", "title": "关系发展", "required": True}, {"id": "development", "title": "发展可能性", "required": True} ] }, "standard": { "name": "标准模板", "sections": [ {"id": "basic", "title": "基本信息", "required": True}, {"id": "appearance", "title": "外貌特征", "required": True}, {"id": "personality", "title": "性格特点", "required": True}, {"id": "background", "title": "背景故事", "required": True}, {"id": "relationships", "title": "人物关系", "required": True}, {"id": "development", "title": "故事发展", "required": True}, {"id": "notes", "title": "创作笔记", "required": False} ] } } return templates def _get_section_content(self, section_id, character_data): """获取章节内容 Args: section_id: 章节ID character_data: 角色数据字典 Returns: 章节内容字符串 """ section_templates = { "basic": """- **姓名**：{name} - **年龄**：{age} - **性别**：{gender} - **职业/身份**：{occupation} - **故事中的角色**：{role}""", "appearance": """- **整体印象**：{overall_impression} - **面部特征**：{facial_features} - **身材体型**：{body_type} - **着装风格**：{clothing_style} - **标志性特征**：{distinctive_features}""", "personality": """- **核心性格**：{core_personality} - **优点**：{strengths} - **缺点**：{weaknesses} - **价值观**：{values} - **恐惧**：{fears} - **渴望**：{desires}""", "background": """- **出身背景**：{origin} - **关键经历**：{key_experiences} - **转折点**：{turning_points} - **未解之谜**：{unsolved_mysteries}""", "relationships": """- **与主角关系**：{relationship_with_protagonist} - **重要关系人**：{important_relationships} - **敌对关系**：{enemy_relationships} - **情感羁绊**：{emotional_bonds}""", "development": """- **角色目标**：{goals} - **内在冲突**：{internal_conflicts} - **外在冲突**：{external_conflicts} - **发展弧线**：{development_arc} - **可能的结局**：{possible_endings}""", "notes": """- **灵感来源**：{inspiration_sources} - **象征意义**：{symbolic_meanings} - **潜在发展**：{potential_developments}""", "core_identity": """- **本质自我**：{true_self} - **社会面具**：{social_mask} - **理想自我**：{ideal_self} - **恐惧自我**：{feared_self}""", "motivation": """- **表层目标**：{surface_goals} - **情感需求**：{emotional_needs} - **存在需求**：{existential_needs} - **未意识需求**：{unconscious_needs}""", "core_belief": """- **世界观**：{worldview} - **核心信念**：{core_beliefs} - **正义观**：{justice_view} - **变革愿景**：{change_vision}""", "mirror": """- **相似点**：{similarities} - **分歧点**：{divergences} - **对立逻辑**：{opposition_logic} - **潜在转化**：{potential_transformation}""", "resources": """- **核心团队**：{core_team} - **盟友势力**：{ally_forces} - **影响范围**：{influence_scope} - **弱点环节**：{weakness_points}""", "function": """- **推动剧情的方式**：{plot_push_methods} - **服务主角的方式**：{protagonist_service_methods} - **独特价值**：{unique_value} - **退出时机**：{exit_timing}""" } template = section_templates.get(section_id, "") if not template: return "" # 从character_data中获取数据，如果不存在则使用占位符 content = template for key in character_data: placeholder = "{" + key + "}" if placeholder in content: content = content.replace(placeholder, character_data.get(key, f"[待填写{key}]")) # 清理未替换的占位符 import re content = re.sub(r'\{[^}]*\}', '[待填写]', content) return content def generate_markdown(self, character_data, output_path=None): """生成markdown档案 Args: character_data: 角色数据字典，必须包含'name'字段 output_path: 输出文件路径，如果为None则返回字符串 Returns: 如果output_path为None，返回markdown字符串；否则写入文件 """ if 'name' not in character_data: raise ValueError("角色数据必须包含'name'字段") template = self.templates.get(self.template_type, self.templates["standard"]) # 生成markdown内容 lines = [] lines.append(f"# {character_data['name']} - 角色档案") lines.append("") lines.append(f"> 生成时间：{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") lines.append(f"> 模板类型：{template['name']}") lines.append("") for section in template['sections']: section_id = section['id'] section_title = section['title'] lines.append(f"## {section_title}") lines.append("") content = self._get_section_content(section_id, character_data) if content: lines.append(content) else: lines.append(f"*{section_title}内容待填写*") lines.append("") # 添加元数据部分 lines.append("---") lines.append("") lines.append("## 档案元数据") lines.append("") lines.append(f"- **角色类型**：{template['name']}") lines.append(f"- **创建时间**：{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") lines.append(f"- **状态**：{'草稿' if character_data.get('status') != 'final' else '完成'}") lines.append(f"- **版本**：{character_data.get('version', '1.0')}") lines.append("") markdown_content = "\n".join(lines) if output_path: # 确保目录存在 output_dir = os.path.dirname(output_path) if output_dir: os.makedirs(output_dir, exist_ok=True) with open(output_path, 'w', encoding='utf-8') as f: f.write(markdown_content) print(f"档案已生成：{output_path}") return output_path else: return markdown_content def generate_enhanced_profile(self, character_data, require_confirmation=True): """使用增强功能生成角色档案 Args: character_data: 角色数据字典，必须包含'name'字段 require_confirmation: 是否需要用户确认 Returns: (是否成功, 最终文件路径, 冲突列表) """ if not self.enable_enhanced_features: logger.error("增强功能未启用，请指定workspace参数初始化生成器") return False, None, [] if 'name' not in character_data: raise ValueError("角色数据必须包含'name'字段") # 初始化增强组件 if self.lore_bible_manager is None: self._init_enhanced_components() if not self.enable_enhanced_features: return False, None, [] try: from profile_session import ProfileSession, SessionConfig # 更新会话配置 self.session_config.character_name = character_data['name'] self.session_config.require_confirmation = require_confirmation # 创建会话 session = ProfileSession(self.session_config) # 生成基础markdown内容 markdown_content = self.generate_markdown(character_data, output_path=None) # 保存临时档案 temp_path = session.save_temp_profile(markdown_content) if not temp_path: logger.error("保存临时档案失败") return False, None, [] # 验证和冲突检测 is_valid, conflicts = session.validate_and_check_conflicts(character_data) # 展示给用户 if require_confirmation: user_confirmed = session.present_to_user(conflicts) if not user_confirmed: logger.info("用户取消创建") session.cancel() return False, None, conflicts else: logger.info("跳过用户确认") # 确认并移动 final_path = session.confirm_and_move() if not final_path: logger.error("移动档案失败") return False, None, conflicts # 清理会话 session.cleanup() logger.info(f"角色档案创建成功: {final_path}") return True, final_path, conflicts except Exception as e: logger.error(f"增强生成失败: {e}") return False, None, [] def create_from_cli(self): """从命令行交互创建""" print("=== 人物档案生成器 ===") print("") # 选择模板 print("请选择角色类型：") print("1. 主角") print("2. 反派") print("3. 配角") print("4. 标准") choice = input("请输入选择 (1-4，默认4): ").strip() type_map = {"1": "protagonist", "2": "antagonist", "3": "supporting", "4": "standard"} template_type = type_map.get(choice, "standard") self.template_type = template_type print(f"使用模板：{self.templates[template_type]['name']}") print("") # 收集基本信息 character_data = {} print("请输入角色基本信息：") character_data['name'] = input("角色姓名: ").strip() or "未命名角色" character_data['age'] = input("年龄: ").strip() or "未知" character_data['gender'] = input("性别: ").strip() or "未指定" character_data['occupation'] = input("职业/身份: ").strip() or "未指定" character_data['role'] = input("故事中的角色: ").strip() or "未指定" print("") print("其他信息将在生成的档案中以占位符形式出现，请后续编辑完善。") # 生成文件名 safe_name = "".join(c for c in character_data['name'] if c.isalnum() or c in (' ', '-', '_')).rstrip() filename = f"character_profile_{safe_name}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md" # 选择输出路径 default_output = os.path.join(os.getcwd(), filename) output_path = input(f"输出文件路径 (默认: {default_output}): ").strip() or default_output # 生成档案 try: result = self.generate_markdown(character_data, output_path) print("") print("✓ 人物档案生成成功！") print(f"文件位置: {result if isinstance(result, str) else output_path}") print("") print("下一步：") print("1. 使用文本编辑器打开生成的markdown文件") print("2. 根据模板提示填写详细信息") print("3. 保存并用于创作参考") except Exception as e: print(f"生成失败: {e}") return 1 return 0 def main(): """主函数""" if len(sys.argv) > 1: # 命令行参数模式 import argparse parser = argparse.ArgumentParser(description='生成人物档案markdown文件') parser.add_argument('--type', '-t', choices=['protagonist', 'antagonist', 'supporting', 'standard'], default='standard', help='角色类型') parser.add_argument('--name', '-n', required=True, help='角色姓名') parser.add_argument('--age', help='年龄') parser.add_argument('--gender', help='性别') parser.add_argument('--occupation', help='职业/身份') parser.add_argument('--role', help='故事中的角色') parser.add_argument('--output', '-o', help='输出文件路径') parser.add_argument('--workspace', '-w', help='工作目录路径，启用LoreBible管理功能') parser.add_argument('--no-confirm', action='store_true', help='跳过用户确认（仅增强模式）') parser.add_argument('--interactive', '-i', action='store_true', help='交互模式') args = parser.parse_args() if args.interactive: # 交互模式，暂时不支持增强功能 generator = CharacterProfileGenerator(args.type) return generator.create_from_cli() else: character_data = { 'name': args.name, 'age': args.age or '未知', 'gender': args.gender or '未指定', 'occupation': args.occupation or '未指定', 'role': args.role or '未指定' } # 检查是否启用增强模式 if args.workspace: # 增强模式 generator = CharacterProfileGenerator(args.type, workspace=args.workspace) # 使用增强生成 success, final_path, conflicts = generator.generate_enhanced_profile( character_data, require_confirmation=not args.no_confirm ) if success: print(f"✓ 角色档案创建成功: {final_path}") if conflicts: print(f" 检测到 {len(conflicts)} 个问题，已处理") return 0 else: print(f"✗ 角色档案创建失败") if conflicts: print(f" 存在 {len(conflicts)} 个问题需要解决") return 1 else: # 传统模式 generator = CharacterProfileGenerator(args.type) if args.output: output_path = args.output else: safe_name = "".join(c for c in args.name if c.isalnum() or c in (' ', '-', '_')).rstrip() filename = f"character_profile_{safe_name}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md" output_path = os.path.join(os.getcwd(), filename) result = generator.generate_markdown(character_data, output_path) print(f"档案已生成: {output_path}") return 0 else: # 交互模式 generator = CharacterProfileGenerator() return generator.create_from_cli() if __name__ == "__main__": sys.exit(main())