haha

2026-03-05 10:27:28 +08:00
parent 45e21cb7a1
commit 2971bfad8a
18 changed files with 3023 additions and 5 deletions
--- a/scripts/init_recruit_test_data.py
+++ b/scripts/init_recruit_test_data.py
@@ -0,0 +1,140 @@
+# -*- coding: utf-8 -*-
+"""
+数据库初始化脚本：创建测试数据
+用于测试新增的筛选和复聊功能
+"""
+
+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()
+
+from server.models import FilterConfig, ChatScript
+
+
+def create_filter_config():
+    """创建筛选配置示例"""
+    print("创建筛选配置...")
+    
+    # 删除旧的测试配置
+    FilterConfig.objects.filter(name__contains="测试").delete()
+    
+    # 创建新配置
+    config = FilterConfig.objects.create(
+        name="Python开发筛选配置",
+        age_min=22,
+        age_max=35,
+        gender="不限",
+        education="本科",
+        activity="3天内活跃",
+        positions=["Python开发", "后端开发", "全栈开发", "Django开发"],
+        greeting_min=5,
+        greeting_max=20,
+        rest_minutes=30,
+        collection_min=10,
+        collection_max=50,
+        message_interval=30,
+        is_active=True
+    )
+    print(f"✓ 创建筛选配置: {config.name} (ID: {config.id})")
+    
+    return config
+
+
+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:
+        # 创建筛选配置
+        config = create_filter_config()
+        
+        # 创建话术
+        scripts = create_chat_scripts()
+        
+        print("\n" + "=" * 60)
+        print("初始化完成！")
+        print("=" * 60)
+        print(f"\n筛选配置: {config.name}")
+        print(f"  - 年龄: {config.age_min}-{config.age_max}岁")
+        print(f"  - 学历: {config.education}及以上")
+        print(f"  - 活跃度: {config.activity}")
+        print(f"  - 期望职位: {', '.join(config.positions)}")
+        
+        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())