"""在 Docker 容器中创建超级管理员。

使用方式：
    docker exec -it <api容器名> python /app/scripts/init_admin_docker.py

或者直接复制以下命令到 Docker 容器中执行：
"""
import sys
import os
import bcrypt
import uuid

# 添加 app 路径
sys.path.insert(0, '/app/api')


def create_admin():
    """创建超级管理员账号。"""
    try:
        from insurance.app import create_app
        from insurance.db.compat import db
        from insurance.models.wecom_user import WeComUserMapping

        app = create_app()

        with app.app_context():
            username = "admin"
            password = "Admin@123456"

            # 检查是否已存在
            existing = db.session.query(WeComUserMapping).filter_by(username=username).first()
            if existing:
                print(f"User {username} already exists, role: {existing.role}")
                # 如果不是管理员，更新角色
                if existing.role not in ('super_admin', 'admin'):
                    existing.role = 'super_admin'
                    db.session.commit()
                    print(f"Updated role to super_admin")
                return

            # 创建用户
            password_hash = bcrypt.hashpw(password.encode(), bcrypt.gensalt()).decode()
            mapping = WeComUserMapping(
                wecom_userid=f"local_{uuid.uuid4().hex[:8]}",
                internal_user_id=f"user_{uuid.uuid4().hex[:8]}",
                username=username,
                password_hash=password_hash,
                role="super_admin",
                department="管理组",
                status="active",
            )
            db.session.add(mapping)
            db.session.commit()

            print("=" * 50)
            print("Admin user created successfully!")
            print("=" * 50)
            print(f"  Username: {username}")
            print(f"  Password: {password}")
            print(f"  Role:     super_admin")
            print("=" * 50)

    except Exception as e:
        print(f"Error: {e}")
        import traceback
        traceback.print_exc()


if __name__ == "__main__":
    create_admin()