"""
创建管理员用户脚本
"""
import sys
import os

# 添加项目根目录到 Python 路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from app.core.database import SessionLocal
from app.models.user import User, UserRole, UserStatus
from app.core.security import get_password_hash


def create_admin_user():
    """创建管理员用户"""
    db = SessionLocal()
    try:
        # 检查是否已存在admin用户
        existing_user = db.query(User).filter(User.username == "admin").first()
        if existing_user:
            print(f"Admin user already exists with ID: {existing_user.id}")
            return

        # 使用与系统相同的密码哈希方法
        password = "admin123"
        hashed_password = get_password_hash(password)

        admin_user = User(
            username="admin",
            email="admin@example.com",
            hashed_password=hashed_password,
            role=UserRole.ADMIN,
            nickname="系统管理员",
            status=UserStatus.ACTIVE,
            balance=1000000,  # 10000元
            is_active=1,
        )

        db.add(admin_user)
        db.commit()
        db.refresh(admin_user)

        print(f"Admin user created successfully with ID: {admin_user.id}")
        print(f"Username: admin")
        print(f"Password: admin123")

    except Exception as e:
        db.rollback()
        print(f"Error creating admin user: {e}")
    finally:
        db.close()


if __name__ == "__main__":
    create_admin_user()