tongquU/tongqu_backend_v2/accounts/models.py

from django.db import models
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin
from django.utils import timezone
from django.utils.translation import gettext_lazy as _

class CustomUserManager(BaseUserManager):
    """
    自定义用户模型的管理器。
    """
    def create_user(self, email, phone_number, password=None, **extra_fields):
        """
        创建并保存一个具有给定邮箱、手机号和密码的用户。
        """
        if not email:
            raise ValueError(_('用户必须有一个邮箱地址'))
        if not phone_number:
            raise ValueError(_('用户必须有一个手机号'))

        email = self.normalize_email(email)
        # 确保普通用户创建时这些字段为False或默认值
        extra_fields.setdefault('is_staff', False)
        extra_fields.setdefault('is_superuser', False)
        extra_fields.setdefault('is_active', True) # 普通用户默认激活

        user = self.model(email=email, phone_number=phone_number, **extra_fields)
        user.set_password(password) # 处理密码哈希
        user.save(using=self._db)
        return user

    def create_superuser(self, email, phone_number, password=None, **extra_fields):
        """
        创建并保存一个具有给定邮箱、手机号和密码的超级用户。
        """
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)
        extra_fields.setdefault('is_active', True)

        if extra_fields.get('is_staff') is not True:
            raise ValueError(_('超级用户必须将 is_staff 设置为 True.'))
        if extra_fields.get('is_superuser') is not True:
            raise ValueError(_('超级用户必须将 is_superuser 设置为 True.'))
        
        # 对于 create_superuser，我们直接调用模型的构造函数并设置密码，
        # 因为我们已经明确设置了 is_staff 和 is_superuser。
        # 调用 self.create_user 可能会覆盖这些值，除非 create_user 逻辑特别处理。
        if not email: # 再次检查，虽然 create_user 也会检查
            raise ValueError(_('超级用户必须有一个邮箱地址'))
        if not phone_number:
            raise ValueError(_('超级用户必须有一个手机号'))

        email = self.normalize_email(email)
        user = self.model(email=email, phone_number=phone_number, **extra_fields)
        user.set_password(password)
        user.save(using=self._db)
        return user

class CustomUser(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(_('邮箱地址'), unique=True)
    phone_number = models.CharField(_('手机号'), max_length=20, unique=True)
    nickname = models.CharField(_('昵称'), max_length=100, blank=True)
    avatar = models.ImageField(_('头像'), upload_to='avatars/', null=True, blank=True) # 需要 Pillow 包
    bio = models.TextField(_('简介'), blank=True)
    school = models.CharField(_('学校'), max_length=100, blank=True)

    is_staff = models.BooleanField(
        _('职员状态'),
        default=False,
        help_text=_('指明用户是否可以登录到管理站点。'),
    )
    is_active = models.BooleanField(
        _('激活状态'),
        default=True,
        help_text=_(
            '指明用户是否被认为是活跃的。'
            '取消选择此项而不是删除账户。'
        ),
    )
    date_joined = models.DateTimeField(_('注册日期'), default=timezone.now)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['phone_number', 'nickname'] # 创建超级用户时需要的额外字段

    objects = CustomUserManager() # 关联自定义的 Manager

    def __str__(self):
        return self.email

    class Meta:
        verbose_name = _('用户')
        verbose_name_plural = _('用户们')