import os
import sys

import django
from faker import Faker
from loguru import logger

# 配置 Django 环境
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "server.settings")  # 修改为你的 settings 路径
django.setup()

from users.models import User

# 初始化 Faker，设置中文（如果需要中文名）
fake = Faker('zh_CN')


def seed_users(count=5):
    logger.info(f"正在生成 {count} 个用户...")

    users = []
    for i in range(count):
        # 生成随机用户名
        username = fake.user_name()

        # 检查用户名是否已存在，避免冲突
        if User.objects.filter(username=username).exists():
            continue

        user = User.objects.create_user(
            username=username,
            email=fake.safe_email(),
            password='password123',  # 统一密码，方便测试登录
            is_active=True,
            is_staff=False,  # 随机是否为管理员
            first_name=fake.first_name(),
            last_name=fake.last_name()
        )
        users.append(user)
        logger.info(f" - 创建用户: {username} (密码: password123)")

    logger.info("用户生成完毕！")


if __name__ == "__main__":
    seed_users(10)