django2.0继承并重载框架自带的用户模型

最新推荐文章于 2023-10-19 19:39:43 发布

原创最新推荐文章于 2023-10-19 19:39:43 发布 · 897 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#django #鉴权 #用户模型 #自定义 #用户属性

Python 专栏收录该内容

57 篇文章

订阅专栏

本文详细介绍了在Django框架中如何自定义用户模型，包括继承AbstractUser和AbstractBaseUser，以及如何重载默认的主键和用户字段，实现自定义用户管理。

使用django建站，需要使用django自带的用户授权和管理功能，又需要自定义用户模型，于是继承并重载了默认用户模型的属性，记录如下。

在settings.py中：

# 指定默认用户模型为自定义的模型
AUTH_USER_MODEL = 'mysite.User'

在models.py中：

from django.contrib.auth.models import AbstractUser, UnicodeUsernameValidator

class User(AbstractUser):
    username_validator = UnicodeUsernameValidator()
    username = models.CharField('用户名', max_length=20, unique=True, default='', validators=[username_validator])
    password = models.CharField('密码', max_length=128, default='')
    phone = models.CharField('手机号', primary_key=True, max_length=11, default='')

    # 这里重载了默认的主键，默认的主键是email
    EMAIL_FIELD = 'phone'
    REQUIRED_FIELDS = ['phone']

    def __str__(self):
        return '%s, %s' % (self.username, self.phone)

    class Meta(AbstractUser.Meta):
        swappable = 'AUTH_USER_MODEL'

或者更彻底的自定义：

from django.db import models
from django.contrib.auth.base_user import AbstractBaseUser


class UserManager(models.Manager):
    use_in_migrations = False

    def create_user(self, **extra_fields):
        user = self.model(**extra_fields)
        user.set_password(extra_fields['password'])
        user.save(using=self._db)
        # Do something ...
        return user


class User(AbstractBaseUser):
    username = models.CharField('用户名', max_length=100, primary_key=True)

    USERNAME_FIELD = 'username'
    objects = UserManager()

    def __str__(self):
        return self.username

    class Meta:
        verbose_name = 'user'
        verbose_name_plural = 'users'