上一篇我们自定义的 User 并未包含 Django 自带的权限模块，现在我们自定义一个能使用权限模块的自定义 User。
要自定义包含原权限模块的 User 模型，只需要让 User 模型继承 AbstractBaseUser 类的同时还继承 PermissionsMixin 类。
Django 提供的 PermissionsMixin 是一个抽象模型，为您提供支持 Django 权限模型所需的所有方法和数据库字段。
PermissionsMixin 提供了以下方法和属性：
- is_superuser
- get_group_permissions(obj=None)
- get_all_permissions(obj=None)
- has_perm(perm, obj=None)
- has_perms(perm_list, obj=None)
- has_module_perms(package_name)
下面是一个包含 Permission 的自定义 User 模型例子：
from django.db import models
from django.contrib.auth.models import (
BaseUserManager, AbstractBaseUser, PermissionsMixin
)
class MyUserManager(BaseUserManager):
def _create_user(self, email, password, date_of_birth, is_staff, is_superuser, **extra_fields):
if not email:
raise ValueError('The given email must be set')
email = self.normalize_email(email)
user = self.model(
email=email,
date_of_birth=date_of_birth,
is_staff=is_staff,
is_active=True,
is_superuser=is_superuser,
**extra_fields
)
user.set_password(password)
user.save(using=self._db)
return user
def create_user(self, email, date_of_birth, password=None, **extra_fields):
return self._create_user(email, password, date_of_birth, is_staff=False, is_superuser=False, **extra_fields)
def create_superuser(self, email, date_of_birth, password, **extra_fields):
return self._create_user(email, password, date_of_birth, is_staff=True, is_superuser=True, **extra_fields)
class MyUser(AbstractBaseUser, PermissionsMixin):
email = models.EmailField(
verbose_name='email address',
max_length=255,
unique=True,
)
date_of_birth = models.DateField()
# 管理类属性
is_active = models.BooleanField(default=True)
# is_admin = models.BooleanField(default=False)
is_staff = models.BooleanField(default=False)
# 指定 MyUser 模型的管理器
objects = MyUserManager()
# 描述User模型上用作唯一标识符的字段名称
# 字段必须必须是唯一的（即在其定义中设置unique=True）
USERNAME_FIELD = 'email'
# 当通过createsuperuser管理命令创建一个用户时
# 系统会提示用户输入以下字段
REQUIRED_FIELDS = ['date_of_birth']
# 返回 email 地址
def get_full_name(self):
return self.email
# 返回 email 地址
def get_short_name(self):
return self.email
def __str__(self): # __unicode__ on Python 2
return self.email
别忘了在 setting.py 中加上以下内容：
AUTH_USER_MODEL = 'myUserApp.MyUser'
在 shell 中检验我们自定义的 User 模型。
创建一个 superuser：
from myUserApp.models import MyUser
u = MyUser.objects.create_superuser(
email='user_001@django.com',
date_of_birth='1988-8-8',
password='********',
)
查看全部权限：
u.get_all_permissions()
>>> {'auth.add_group', 'myUserApp.delete_myuser', 'auth.delete_group',
'sessions.change_session', 'myUserApp.change_myuser',
'admin.add_logentry', 'contenttypes.change_contenttype',
'myUserApp.add_myuser', 'auth.add_permission',
'contenttypes.add_contenttype', 'admin.delete_logentry',
'contenttypes.delete_contenttype', 'auth.delete_permission',
'auth.change_group', 'auth.change_permission', 'admin.change_logentry',
'sessions.add_session', 'sessions.delete_session'}
