513 lines
20 KiB
Python
513 lines
20 KiB
Python
#!/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()) |