Looking for django Keywords? Try Ask4Keywords

Django 指定自定義用戶模型


Django的內置User模型並不總是適用於某些類型的項目。在某些網站上,使用電子郵件地址而不是用戶名可能更有意義。

您可以覆蓋默認的User模型中添加自定義的User模式向AUTH_USER_MODEL設置,在您的項目設置文件:

AUTH_USER_MODEL = 'myapp.MyUser'

請注意 ,強烈建議在創建任何遷移或首次運行manage.py migrate之前創建AUTH_USER_MODEL 。由於Django的synamic dependecy功能的限制。

例如,在您的博客上,您可能希望其他作者能夠使用電子郵件地址而不是常規用戶名登錄,因此我們創建一個自定義User模型,其電子郵件地址為USERNAME_FIELD

from django.contrib.auth.models import AbstractBaseUser

class CustomUser(AbstractBaseUser):
     email = models.EmailField(unique=True)
     
     USERNAME_FIELD = 'email'

通過繼承AbstractBaseUser我們可以構建一個兼容的User模型。 AbstractBaseUser提供了User模型的核心實現。

為了讓Django manage.py createsuperuser命令知道需要哪些其他字段,我們可以指定REQUIRED_FIELDS 。這個值對Django的其他部分沒有影響!

class CustomUser(AbstractBaseUser):
    ...
    first_name = models.CharField(max_length=254)
    last_name = models.CharField(max_length=254)
    ...
    REQUIRED_FIELDS = ['first_name', 'last_name']

為了符合Django的其他部分,我們仍然需要指定值is_active ,函數get_full_name()get_short_name()

class CustomUser(AbstractBaseUser):
    ...
    is_active = models.BooleanField(default=False)
    ...
    def get_full_name(self):
        full_name = "{0} {1}".format(self.first_name, self.last_name)
        return full_name.strip()

    def get_short_name(self):
        return self.first_name

您還應該為User模型創建自定義UserManager ,這允許Django使用create_user()create_superuser()函數:

from django.contrib.auth.models import BaseUserManager

class CustomUserManager(BaseUserManager):
    def create_user(self, email, first_name, last_name, password=None):
        if not email:
            raise ValueError('Users must have an email address')

        user = self.model(
            email=self.normalize_email(email),
        )

        user.set_password(password)
        user.first_name = first_name
        user.last_name = last_name
        user.save(using=self._db)
        return user

    def create_superuser(self, email, first_name, last_name, password):
        user = self.create_user(
            email=email,
            first_name=first_name,
            last_name=last_name,
            password=password,
        )

        user.is_admin = True
        user.is_active = True
        user.save(using=self.db)
        return user