# -*- 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 ChatScript


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())