Files
boss_dp/scripts/init_recruit_test_data.py

103 lines
3.1 KiB
Python
Raw Normal View History

2026-03-05 10:27:28 +08:00
# -*- coding: utf-8 -*-
"""
数据库初始化脚本创建测试数据
2026-03-06 10:47:46 +08:00
用于测试招聘回复与复聊功能
2026-03-05 10:27:28 +08:00
"""
import os
import sys
import django
# 设置 Django 环境
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'server.settings')
django.setup()
2026-03-06 10:47:46 +08:00
from server.models import ChatScript
2026-03-05 10:27:28 +08:00
def create_chat_scripts():
"""创建话术示例"""
print("\n创建话术配置...")
# 删除旧的测试话术
ChatScript.objects.filter(position__in=["Python开发", "通用", "测试"]).delete()
scripts = [
{
"position": "Python开发",
"script_type": "first",
"content": "您好看到您的简历Python技术栈很符合我们的要求我们这边是做XXX项目的期待与您进一步沟通。",
"keywords": "",
},
{
"position": "Python开发",
"script_type": "followup",
"content": "您好,不知道您是否方便留个联系方式?后续沟通会更及时一些。",
"keywords": "",
},
{
"position": "Python开发",
"script_type": "wechat",
"content": "后续沟通会更及时,您方便留一下您的微信号吗?我这边加您。",
"keywords": "微信,联系方式",
},
{
"position": "通用",
"script_type": "first",
"content": "您好!看到您的简历很符合我们的要求,期待与您进一步沟通。",
"keywords": "",
},
{
"position": "通用",
"script_type": "followup",
"content": "您好,期待与您进一步沟通,方便留个联系方式吗?",
"keywords": "",
},
{
"position": "通用",
"script_type": "wechat",
"content": "后续沟通会更及时,您方便留一下您的微信号吗?我这边加您。",
"keywords": "微信,联系方式",
},
]
created_scripts = []
for script_data in scripts:
script = ChatScript.objects.create(**script_data, is_active=True)
created_scripts.append(script)
print(f"✓ 创建话术: {script.position} - {script.get_script_type_display()}")
return created_scripts
def main():
print("=" * 60)
print("BOSS招聘自动化 - 初始化测试数据")
print("=" * 60)
try:
# 创建话术
scripts = create_chat_scripts()
print("\n" + "=" * 60)
print("初始化完成!")
print("=" * 60)
print(f"\n话术配置: 共 {len(scripts)}")
for script in scripts:
print(f" - {script.position} / {script.get_script_type_display()}")
print("\n现在可以运行招聘任务测试新功能了!")
except Exception as e:
print(f"\n✗ 错误: {e}")
import traceback
traceback.print_exc()
return 1
return 0
if __name__ == "__main__":
sys.exit(main())