#!/usr/bin/env python
"""
首次部署初始化脚本

该脚本用于首次部署系统时执行所有必要的初始化操作，包括：
1. 检查数据库连接
2. 执行数据库迁移
3. 创建管理员用户
4. 创建默认用户组
5. 收集静态文件

使用方法:
    python add_status_field.py --init

或者使用 Django 标准命令:
    python manage.py makemigrations
    python manage.py migrate
    python manage.py createsuperuser
"""

import os
import sys
import django

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'backend.settings')
django.setup()

from django.core.management import call_command
from django.contrib.auth import get_user_model
from django.contrib.auth.models import Group


def check_database_connection():
    """
    检查数据库连接
    """
    print("=" * 60)
    print("检查数据库连接...")
    print("=" * 60)

    try:
        from django.db import connection
        with connection.cursor() as cursor:
            cursor.execute("SELECT 1")
            print("✓ 数据库连接正常")
            return True
    except Exception as e:
        print(f"✗ 数据库连接失败: {e}")
        print("\n请检查 conf/env.py 中的数据库配置")
        return False


def run_migrations():
    """
    执行数据库迁移
    """
    print("\n" + "=" * 60)
    print("执行数据库迁移...")
    print("=" * 60)

    try:
        print("\n[1/2] 生成迁移文件...")
        call_command('makemigrations', verbosity=1)
        print("✓ 迁移文件生成成功")

        print("\n[2/2] 应用迁移...")
        call_command('migrate', verbosity=1)
        print("✓ 数据库迁移成功")
        return True

    except Exception as e:
        print(f"✗ 数据库迁移失败: {e}")
        import traceback
        traceback.print_exc()
        return False


def create_admin_user(username='admin', password='admin123', email='admin@example.com', name='系统管理员'):
    """
    创建管理员用户
    """
    print("\n" + "=" * 60)
    print("创建管理员用户...")
    print("=" * 60)

    User = get_user_model()

    try:
        user = User.objects.get(username=username)
        print(f"管理员用户 '{username}' 已存在，跳过创建")
        print(f"  用户名: {user.username}")
        print(f"  姓名: {user.name}")
        print(f"  邮箱: {user.email}")
        return True

    except User.DoesNotExist:
        try:
            user = User.objects.create_superuser(
                username=username,
                email=email,
                password=password
            )
            user.name = name
            user.phone = None
            user.show_hosts = False
            user.status = 'approved'
            user.save()

            print("✓ 管理员用户创建成功")
            print(f"  用户名: {username}")
            print(f"  密码: {password}")
            print(f"  邮箱: {email}")
            print(f"  姓名: {name}")
            print("\n⚠️  重要提示: 请在首次登录后修改默认密码!")
            return True

        except Exception as e:
            print(f"✗ 创建管理员用户失败: {e}")
            import traceback
            traceback.print_exc()
            return False


def create_default_groups():
    """
    创建默认用户组
    """
    print("\n" + "=" * 60)
    print("创建默认用户组...")
    print("=" * 60)

    groups = [
        {'name': '管理员组', 'description': '拥有所有权限的管理员组'},
        {'name': '开发人员组', 'description': '开发人员组'},
        {'name': '测试人员组', 'description': '测试人员组'},
        {'name': '只读用户组', 'description': '只读用户组'},
    ]

    created_count = 0
    for group_info in groups:
        try:
            Group.objects.get(name=group_info['name'])
            print(f"  用户组 '{group_info['name']}' 已存在")
        except Group.DoesNotExist:
            try:
                Group.objects.create(name=group_info['name'])
                print(f"  ✓ 创建用户组: {group_info['name']}")
                created_count += 1
            except Exception as e:
                print(f"  ✗ 创建用户组 '{group_info['name']}' 失败: {e}")

    print(f"\n✓ 成功创建 {created_count} 个用户组")
    return True


def collect_static_files():
    """
    收集静态文件
    """
    print("\n" + "=" * 60)
    print("收集静态文件...")
    print("=" * 60)

    try:
        call_command('collectstatic', '--noinput', verbosity=1)
        print("✓ 静态文件收集成功")
        return True

    except Exception as e:
        print(f"✗ 静态文件收集失败: {e}")
        import traceback
        traceback.print_exc()
        return False


def display_summary():
    """
    显示部署摘要
    """
    print("\n" + "=" * 60)
    print("部署摘要")
    print("=" * 60)

    User = get_user_model()

    print(f"\n用户数量: {User.objects.count()}")
    print(f"用户组数量: {Group.objects.count()}")

    print("\n默认管理员账号:")
    print("  用户名: admin")
    print("  密码: admin123")
    print("\n⚠️  请在首次登录后立即修改默认密码!")


def main():
    """
    主函数
    """
    import argparse

    parser = argparse.ArgumentParser(description='首次部署初始化脚本')
    parser.add_argument('--init', action='store_true', help='执行完整的初始化流程')
    parser.add_argument('--migrate', action='store_true', help='仅执行数据库迁移')
    parser.add_argument('--create-admin', action='store_true', help='仅创建管理员用户')
    parser.add_argument('--create-groups', action='store_true', help='仅创建用户组')
    parser.add_argument('--collect-static', action='store_true', help='仅收集静态文件')

    args = parser.parse_args()

    if args.init:
        print("\n")
        print("╔" + "=" * 58 + "╗")
        print("║" + " " * 10 + "接口自动化平台 - 首次部署初始化" + " " * 13 + "║")
        print("╚" + "=" * 58 + "╝")
        print()

        steps = [
            ("检查数据库连接", check_database_connection),
            ("执行数据库迁移", run_migrations),
            ("创建管理员用户", create_admin_user),
            ("创建默认用户组", create_default_groups),
            ("收集静态文件", collect_static_files),
        ]

        failed_steps = []

        for step_name, step_func in steps:
            if not step_func():
                failed_steps.append(step_name)
                print(f"\n✗ '{step_name}' 失败，终止部署")
                break

        if not failed_steps:
            display_summary()
            print("\n" + "=" * 60)
            print("✓ 部署初始化完成!")
            print("=" * 60)
            print("\n系统已准备就绪，可以启动服务:")
            print("  python manage.py runserver")
            print()
            return 0
        else:
            print("\n" + "=" * 60)
            print("✗ 部署初始化失败!")
            print("=" * 60)
            print(f"\n失败的步骤: {', '.join(failed_steps)}")
            return 1

    elif args.migrate:
        return 0 if run_migrations() else 1

    elif args.create_admin:
        return 0 if create_admin_user() else 1

    elif args.create_groups:
        return 0 if create_default_groups() else 1

    elif args.collect_static:
        return 0 if collect_static_files() else 1

    else:
        parser.print_help()
        print("\n使用示例:")
        print("  python add_status_field.py --init          # 完整初始化")
        print("  python add_status_field.py --migrate       # 仅迁移数据库")
        print("  python add_status_field.py --create-admin   # 仅创建管理员")
        print("\n或使用 Django 标准命令:")
        print("  python manage.py makemigrations")
        print("  python manage.py migrate")
        print("  python manage.py createsuperuser")
        return 0


if __name__ == '__main__':
    sys.exit(main())