125 lines
4.5 KiB
Python
125 lines
4.5 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import datetime
|
|||
|
|
import shutil
|
|||
|
|
|
|||
|
|
def read_modified_resumes():
|
|||
|
|
"""读取所有修改版简历文件"""
|
|||
|
|
modified_resumes = {}
|
|||
|
|
folder_path = "网页未导入数据/能源产业/能源修改版简历"
|
|||
|
|
|
|||
|
|
# 获取所有.md文件
|
|||
|
|
for filename in os.listdir(folder_path):
|
|||
|
|
if filename.endswith('.md'):
|
|||
|
|
position_name = filename.replace('.md', '')
|
|||
|
|
file_path = os.path.join(folder_path, filename)
|
|||
|
|
|
|||
|
|
with open(file_path, 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
modified_resumes[position_name] = content
|
|||
|
|
|
|||
|
|
return modified_resumes
|
|||
|
|
|
|||
|
|
def update_mock_file_with_modified():
|
|||
|
|
"""更新Mock文件，添加修改版简历内容"""
|
|||
|
|
|
|||
|
|
mock_file = "src/mocks/resumeInterviewMock.js"
|
|||
|
|
|
|||
|
|
# 备份文件
|
|||
|
|
backup_path = f"{mock_file}.backup_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}"
|
|||
|
|
shutil.copy(mock_file, backup_path)
|
|||
|
|
print(f"✅ 已备份文件到：{backup_path}")
|
|||
|
|
|
|||
|
|
# 读取修改版简历
|
|||
|
|
modified_resumes = read_modified_resumes()
|
|||
|
|
print(f"📚 找到 {len(modified_resumes)} 个修改版简历文件")
|
|||
|
|
|
|||
|
|
# 读取当前mock文件
|
|||
|
|
with open(mock_file, 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
|
|||
|
|
# 更新resumeTemplates中对应岗位的modified字段
|
|||
|
|
import re
|
|||
|
|
|
|||
|
|
updated_count = 0
|
|||
|
|
for position_name, resume_content in modified_resumes.items():
|
|||
|
|
print(f" 处理岗位: {position_name}")
|
|||
|
|
|
|||
|
|
# 转义特殊字符用于正则表达式
|
|||
|
|
escaped_position = re.escape(position_name)
|
|||
|
|
|
|||
|
|
# 查找该岗位在resumeTemplates中的位置
|
|||
|
|
# 模式：查找 "position": "岗位名称" 后的 content 对象
|
|||
|
|
pattern = rf'("position":\s*"{escaped_position}"[^{{]*?"content":\s*{{[^}}]*?"original":[^,]*?,)(\s*"modified":\s*)"[^"]*"'
|
|||
|
|
|
|||
|
|
# 准备替换的内容（转义JSON字符串）
|
|||
|
|
escaped_content = json.dumps(resume_content, ensure_ascii=False)
|
|||
|
|
|
|||
|
|
# 执行替换
|
|||
|
|
new_content, count = re.subn(
|
|||
|
|
pattern,
|
|||
|
|
rf'\1\2{escaped_content}',
|
|||
|
|
content
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if count > 0:
|
|||
|
|
content = new_content
|
|||
|
|
updated_count += 1
|
|||
|
|
print(f" ✅ 已更新")
|
|||
|
|
else:
|
|||
|
|
# 如果第一种模式失败，尝试另一种模式（没有modified字段的情况）
|
|||
|
|
pattern2 = rf'("position":\s*"{escaped_position}"[^{{]*?"content":\s*{{[^}}]*?"original":[^}}]*?)(}})'
|
|||
|
|
|
|||
|
|
# 添加modified字段
|
|||
|
|
replacement = rf'\1,\n "modified": {escaped_content}\2'
|
|||
|
|
new_content, count = re.subn(pattern2, replacement, content)
|
|||
|
|
|
|||
|
|
if count > 0:
|
|||
|
|
content = new_content
|
|||
|
|
updated_count += 1
|
|||
|
|
print(f" ✅ 已添加modified字段")
|
|||
|
|
else:
|
|||
|
|
print(f" ⚠️ 未找到匹配位置")
|
|||
|
|
|
|||
|
|
# 写回文件
|
|||
|
|
with open(mock_file, 'w', encoding='utf-8') as f:
|
|||
|
|
f.write(content)
|
|||
|
|
|
|||
|
|
print(f"\n✅ 成功更新 {updated_count}/{len(modified_resumes)} 个岗位的修改版简历")
|
|||
|
|
|
|||
|
|
# 验证更新
|
|||
|
|
verify_updates(mock_file, modified_resumes.keys())
|
|||
|
|
|
|||
|
|
def verify_updates(mock_file, position_names):
|
|||
|
|
"""验证更新是否成功"""
|
|||
|
|
with open(mock_file, 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
|
|||
|
|
print("\n📊 验证结果：")
|
|||
|
|
for position_name in position_names:
|
|||
|
|
# 检查是否包含该岗位的modified内容
|
|||
|
|
pattern = f'"position": "{position_name}"'
|
|||
|
|
if pattern in content:
|
|||
|
|
# 进一步检查是否有modified字段
|
|||
|
|
# 查找该岗位后面的modified字段
|
|||
|
|
start_idx = content.find(pattern)
|
|||
|
|
if start_idx > 0:
|
|||
|
|
# 查找该岗位content对象的范围（简化检查）
|
|||
|
|
end_idx = content.find('"position":', start_idx + len(pattern))
|
|||
|
|
if end_idx < 0:
|
|||
|
|
end_idx = start_idx + 2000 # 如果是最后一个，取后面2000字符
|
|||
|
|
|
|||
|
|
section = content[start_idx:end_idx]
|
|||
|
|
if '"modified":' in section and position_name in section:
|
|||
|
|
print(f" ✅ {position_name}: 已包含修改版")
|
|||
|
|
else:
|
|||
|
|
print(f" ❌ {position_name}: 未包含修改版")
|
|||
|
|
else:
|
|||
|
|
print(f" ⚠️ {position_name}: 未找到该岗位")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
update_mock_file_with_modified()
|