对django的User模型和四种扩展/重写方法小结


Posted in Python onAugust 17, 2019

User模型

User模型是这个框架的核心部分。他的完整的路径是在django.contrib.auth.models.User。以下对这个User对象做一个简单了解:

字段:

内置的User模型拥有以下的字段:

username: 用户名。150个字符以内。可以包含数字和英文字符,以及_、@、+、.和-字符。不能为空,且必须唯一!

first_name:歪果仁的first_name,在30个字符以内。可以为空。

last_name:歪果仁的last_name,在150个字符以内。可以为空。

email:邮箱。可以为空。

password:密码。经过哈希过后的密码。

groups:分组。一个用户可以属于多个分组,一个分组可以拥有多个用户。groups这个字段是跟Group的一个多对多的关系。

user_permissions:权限。一个用户可以拥有多个权限,一个权限可以被多个用户所有用。和Permission属于一种多对多的关系。

is_staff:是否可以进入到admin的站点。代表是否是员工。

is_active:是否是可用的。对于一些想要删除账号的数据,我们设置这个值为False就可以了,而不是真正的从数据库中删除。

is_superuser:是否是超级管理员。如果是超级管理员,那么拥有整个网站的所有权限。

last_login:上次登录的时间。

date_joined:账号创建的时间。

登录验证:

Django的验证系统已经帮我们实现了登录验证的功能。通过django.contrib.auth.authenticate即可实现。这个方法只能通过username和password来进行验证。示例代码如下:

from django.contrib.auth import authenticate
user = authenticate(username='Tom', password='111111')
# 如果验证通过了,那么就会返回一个user对象。
if user is not None:
  # 执行验证通过后的代码
else:
  # 执行验证没有通过的代码。

扩展用户模型:

Django内置的User模型虽然已经足够强大了。但是有时候还是不能满足我们的需求。比如在验证用户登录的时候,他用的是用户名作为验证,而我们有时候需要通过手机号码或者邮箱来进行验证。还有比如我们想要增加一些新的字段。那么这时候我们就需要扩展用户模型了。扩展用户模型有多种方式。这里我们来一一讨论下。

1. 设置Proxy模型:

作用: 给模型增加操作方法

局限: 不能增加或减少User模型的字段

好处: 不破坏原来的User模型的表结构

如果你对Django提供的字段,以及验证的方法都比较满意,没有什么需要改的。但是只是需要在他原有的基础之上增加一些操作的方法。那么建议使用这种方式。示例代码如下:

# models.py
class Person(User):
  # 如果模型是一个代理模型
  # 那么就不能在这个模型中添加新的Field
  # telephone = models.CharField(max_length=11) # 错误写法
  class Meta:
    proxy = True
 
  # proxy正确用法是给模型添加自定义方法
  # 如添加列出黑名单的方法
  def get_blacklist(self):
    return self.objects.filter(is_active=False)

在以上,我们定义了一个Person类,让他继承自User,并且在Meta中设置proxy=True,说明这个只是User的一个代理模型。他并不会影响原来User模型在数据库中表的结构。以后如果你想方便的获取所有黑名单的人,那么你就可以通过Person.get_blacklist()就可以获取到。并且User.objects.all()和Person.objects.all()其实是等价的。因为他们都是从User这个模型中获取所有的数据。

2. 一对一外键:

作用: 给模型增加新的字段, 新方法

局限: 只能增加, 不能减少字段, 不能修改户验证方法: authenticate

好处: 不破坏原来的User模型的表结构

如果你对用户验证方法authenticate没有其他要求,就是使用username和password即可完成。但是想要在原来模型的基础之上添加新的字段,那么可以使用一对一外键的方式。示例代码如下:

# models.py
from django.contrib.auth.models import User
from django.db import models
from django.dispatch import receiver
from django.db.models.signals import post_save
 
class UserExtension(models.Model):
  user = models.OneToOneField(User,on_delete=models.CASCADE,related_name='extension')
  birthday = models.DateField(null=True,blank=True)
  school = models.CharField(max_length=100)
 
 
@receiver(post_save,sender=User)
def create_user_extension(sender,instance,created,**kwargs):
  if created:
    UserExtension.objects.create(user=instance)
  else:
    instance.extension.save()

以上定义一个UserExtension的模型,并且让她和User模型进行一对一的绑定,以后我们新增的字段,就添加到UserExtension上。并且还写了一个接受保存模型的信号处理方法,只要是User调用了save方法,那么就会创建一个UserExtension和User进行绑定。

# views.py
from django.contrib.auth.models import User
from django.http import HttpResponse
 
 
def one_to_one_view(request):
  user = User.objects.create_user(username='Tom',email='tom@qq.com',password='111111')
  # 给扩展的字段设置值
  user.extension.school = 'Harvard'
  user.save()
  return HttpResponse('一对一扩展User模型')

3. 继承自AbstractUser:

作用: 给模型增加新的字段, 修改户验证方法: authenticate

局限: 只能增加, 不能减少字段

坏处: 破坏了原来的User模型的表结构

对于authenticate不满意,并且不想要修改原来User对象上的一些字段,但是想要增加一些字段,那么这时候可以直接继承自django.contrib.auth.models.AbstractUser,其实这个类也是django.contrib.auth.models.User的父类。比如我们想要在原来User模型的基础之上添加一个telephone和school字段。示例代码如下:

# models.py
from django.contrib.auth.models import AbstractUser
class User(AbstractUser):
  telephone = models.CharField(max_length=11,unique=True)
  school = models.CharField(max_length=100)
  # 指定telephone作为USERNAME_FIELD, 而不是原来的username字段, 所以username要重写
  username = models.CharField(max_length=150)
 
  # 指定telephone作为USERNAME_FIELD,以后使用authenticate
  # 函数验证的时候,就可以根据telephone来验证
  # 而不是原来的username
  USERNAME_FIELD = 'telephone'
  # USERNAME_FIELD对应的'telephone'字段和密码字段默认是必须的字段
  # 下[]可以添加其它必须的字段, 比如['username', 'email']
  REQUIRED_FIELDS = []
 
  # 重新定义Manager对象,在创建user的时候使用telephone和
  # password,而不是使用username和password
  objects = UserManager()
 
 
# 重写UserManager
class UserManager(BaseUserManager):
  use_in_migrations = True
 
  def _create_user(self, telephone, password, **extra_fields):
    if not telephone:
      raise ValueError("请填入手机号码!")
    if not password:
      raise ValueError("请填入密码!")
    user = self.model(telephone=telephone, **extra_fields)
    user.set_password(password)
    user.save(using=self._db)
    return user
 
  def create_user(self, telephone, password, **extra_fields):
    extra_fields.setdefault('is_staff', False)
    extra_fields.setdefault('is_superuser', False)
    return self._create_user(telephone, password, **extra_fields)
 
  def create_superuser(self, telephone, password, **extra_fields):
    extra_fields.setdefault('is_staff', True)
    extra_fields.setdefault('is_superuser', True)
 
    if extra_fields.get('is_staff') is not True:
      raise ValueError('Superuser must have is_staff=True.')
    if extra_fields.get('is_superuser') is not True:
      raise ValueError('Superuser must have is_superuser=True.')
 
    return self._create_user(telephone, password, **extra_fields)

然后再在settings中配置好

# settings.py
AUTH_USER_MODEL = 'youappname.User'

这种方式因为破坏了原来User模型的表结构,所以必须要在第一次migrate前就先定义好。

4. 继承自AbstractBaseUser模型:

作用: 给模型增加或减少字段, 修改户验证方法: authenticate

坏处: 破坏了原来的User模型的表结构

注意: 继承自AbstractBaseUser同时还要继承PermissionsMixin

如果你想修改默认的验证方式,并且对于原来User模型上的一些字段不想要,那么可以自定义一个模型,然后继承自AbstractBaseUser,再添加你想要的字段。这种方式会比较麻烦,最好是确定自己对Django比较了解才推荐使用。步骤如下:

创建模型。示例代码如下:

# models.py
from django.contrib.auth.base_user import AbstractBaseUser
from django.contrib.auth.models import PermissionsMixin
from django.db import models
 
 
class User(AbstractBaseUser,PermissionsMixin):
   email = models.EmailField(unique=True)
   username = models.CharField(max_length=150)
   telephone = models.CharField(max_length=11,unique=True)
   is_staff = models.BooleanField(default=False)
   is_active = models.BooleanField(default=True)
 
   USERNAME_FIELD = 'telephone'
   REQUIRED_FIELDS = []
 
  # 这里的UserManager同方法3, 需要重写
   objects = UserManager()
 
   def get_full_name(self):
     return self.username
 
   def get_short_name(self):
     return self.username

其中password和last_login是在AbstractBaseUser中已经添加好了的,我们直接继承就可以了。然后我们再添加我们想要的字段。比如email、username、telephone等。这样就可以实现自己想要的字段了。但是因为我们重写了User,所以应该尽可能的模拟User模型:

USERNAME_FIELD:用来描述User模型名字字段的字符串,作为唯一的标识。如果没有修改,那么会使用USERNAME来作为唯一字段。

REQUIRED_FIELDS:一个字段名列表,用于当通过createsuperuser管理命令创建一个用户时的提示。

is_active:一个布尔值,用于标识用户当前是否可用。

get_full_name():获取完整的名字。

get_short_name():一个比较简短的用户名。

重新定义UserManager:我们还需要定义自己的UserManager,因为默认的UserManager在创建用户的时候使用的是username和password,那么我们要替换成telephone。示例代码如下:

# models.py
from django.contrib.auth.base_user import BaseUserManager
 
 
# 重写UserManager
class UserManager(BaseUserManager):
  use_in_migrations = True
 
  def _create_user(self, telephone, password, **extra_fields):
    if not telephone:
      raise ValueError("请填入手机号码!")
    if not password:
      raise ValueError("请填入密码!")
    user = self.model(telephone=telephone, **extra_fields)
    user.set_password(password)
    user.save(using=self._db)
    return user
 
  def create_user(self, telephone, password, **extra_fields):
    extra_fields.setdefault('is_staff', False)
    extra_fields.setdefault('is_superuser', False)
    return self._create_user(telephone, password, **extra_fields)
 
  def create_superuser(self, telephone, password, **extra_fields):
    extra_fields.setdefault('is_staff', True)
    extra_fields.setdefault('is_superuser', True)
 
    if extra_fields.get('is_staff') is not True:
      raise ValueError('Superuser must have is_staff=True.')
    if extra_fields.get('is_superuser') is not True:
      raise ValueError('Superuser must have is_superuser=True.')
 
    return self._create_user(telephone, password, **extra_fields)

在创建了新的User模型后,还需要在settings中配置好。配置AUTH_USER_MODEL='appname.User'。

# settings.py AUTH_USER_MODEL = 'youappname.User'

如何使用这个自定义的模型:比如以后我们有一个Article模型,需要通过外键引用这个User模型,那么可以通过以下两种方式引用。

第一种就是直接将User导入到当前文件中。示例代码如下:

# models.py
from django.db import models
from myauth.models import User
class Article(models.Model):
  title = models.CharField(max_length=100)
  content = models.TextField()
  author = models.ForeignKey(User, on_delete=models.CASCADE)

这种方式是可以行得通的。但是为了更好的使用性,建议还是将User抽象出来,使用settings.AUTH_USER_MODEL来表示。示例代码如下:

# models.py
from django.db import models
from django.conf import settings
class Article(models.Model):
  title = models.CharField(max_length=100)
  content = models.TextField()
  author = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)

这种方式因为破坏了原来User模型的表结构,所以必须要在第一次migrate前就先定义好。

以上这篇对django的User模型和四种扩展/重写方法小结就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python正则抓取网易新闻的方法示例
Apr 21 Python
python中print()函数的“,”与java中System.out.print()函数中的“+”功能详解
Nov 24 Python
微信跳一跳小游戏python脚本
Jan 05 Python
python try except 捕获所有异常的实例
Oct 18 Python
Python os.rename() 重命名目录和文件的示例
Oct 25 Python
Python安装whl文件过程图解
Feb 18 Python
Django使用Celery加redis执行异步任务的实例内容
Feb 20 Python
python怎么提高计算速度
Jun 11 Python
django 装饰器 检测登录状态操作
Jul 02 Python
python产生模拟数据faker库的使用详解
Nov 04 Python
Django执行源生mysql语句实现过程解析
Nov 12 Python
详解Python牛顿插值法
May 11 Python
python3.6编写的单元测试示例
Aug 17 #Python
python3 实现的对象与json相互转换操作示例
Aug 17 #Python
python3实现的zip格式压缩文件夹操作示例
Aug 17 #Python
django 通过url实现简单的权限控制的例子
Aug 16 #Python
对Django中内置的User模型实例详解
Aug 16 #Python
对Django中的权限和分组管理实例讲解
Aug 16 #Python
django创建最简单HTML页面跳转方法
Aug 16 #Python
You might like
PHILIPS D1835/D1875的电路分析与打理
2021/03/02 无线电
php绘图之在图片上写中文和英文的方法
2015/01/24 PHP
CI框架文件上传类及图像处理类用法分析
2016/05/18 PHP
PHP实现的注册,登录及查询用户资料功能API接口示例
2017/06/06 PHP
Javascript 倒计时源代码.(时.分.秒) 详细注释版
2011/05/09 Javascript
js禁止document element对象选中文本实现代码
2013/03/21 Javascript
Node.js安装教程和NPM包管理器使用详解
2014/08/16 Javascript
一个jquery实现的不错的多行文字图片滚动效果
2014/09/28 Javascript
js实现双击图片放大单击缩小的方法
2015/02/17 Javascript
JavaScript中window.open用法实例详解
2015/04/15 Javascript
javascript实现下拉提示选择框
2015/12/29 Javascript
JS实现类似51job上的地区选择效果示例
2016/11/17 Javascript
原生JS实现移动端web轮播图详解(结合Tween算法造轮子)
2017/09/10 Javascript
利用Javascript开发一个二维周视图日历
2017/12/14 Javascript
浅谈如何使用webpack构建多页面应用
2018/05/30 Javascript
详解小程序原生使用ES7 async/await语法
2018/08/06 Javascript
Vue2.0 v-for filter列表过滤功能的实现
2018/09/07 Javascript
详解微信小程序的不同函数调用的几种方法
2019/05/08 Javascript
JQuery实现简单的复选框树形结构图示例【附源码下载】
2019/07/16 jQuery
Python实现对PPT文件进行截图操作的方法
2015/04/28 Python
Python 私有函数的实例详解
2017/09/11 Python
python numpy 一维数组转变为多维数组的实例
2018/07/02 Python
解决python3 pika之连接断开的问题
2018/12/18 Python
django 连接数据库 sqlite的例子
2019/08/14 Python
python实现while循环打印星星的四种形状
2019/11/23 Python
浅谈Python程序的错误:变量未定义
2020/06/02 Python
pytorch 把图片数据转化成tensor的操作
2021/03/04 Python
CSS3系列之3D制作方法案例
2017/08/14 HTML / CSS
美国购车网站:TrueCar
2016/10/19 全球购物
蹦床仓库:Trampoline Warehouse
2018/12/06 全球购物
Andrew Marc官网:设计师外套的领先制造商
2019/10/30 全球购物
CSS代码检查工具stylelint的使用方法详解
2021/03/27 HTML / CSS
写求职信有哪些注意事项
2014/05/08 职场文书
组工干部对照检查材料
2014/08/25 职场文书
公司离职证明范本(5篇)
2014/09/17 职场文书
Django利用AJAX技术实现博文实时搜索
2021/05/06 Python