對(duì)django的User模型和四種擴(kuò)展/重寫方法小結(jié)

更新時(shí)間：2019年08月17日 09:11:40 作者：梁小亦

今天小編就為大家分享一篇對(duì)django的User模型和四種擴(kuò)展/重寫方法小結(jié)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

User模型

User模型是這個(gè)框架的核心部分。他的完整的路徑是在django.contrib.auth.models.User。以下對(duì)這個(gè)User對(duì)象做一個(gè)簡(jiǎn)單了解：

字段：

內(nèi)置的User模型擁有以下的字段：

username：用戶名。150個(gè)字符以內(nèi)。可以包含數(shù)字和英文字符，以及_、@、+、.和-字符。不能為空，且必須唯一！

first_name：歪果仁的first_name，在30個(gè)字符以內(nèi)。可以為空。

last_name：歪果仁的last_name，在150個(gè)字符以內(nèi)?？梢詾榭?。

email：郵箱?？梢詾榭?。

password：密碼。經(jīng)過(guò)哈希過(guò)后的密碼。

groups：分組。一個(gè)用戶可以屬于多個(gè)分組，一個(gè)分組可以擁有多個(gè)用戶。groups這個(gè)字段是跟Group的一個(gè)多對(duì)多的關(guān)系。

user_permissions：權(quán)限。一個(gè)用戶可以擁有多個(gè)權(quán)限，一個(gè)權(quán)限可以被多個(gè)用戶所有用。和Permission屬于一種多對(duì)多的關(guān)系。

is_staff：是否可以進(jìn)入到admin的站點(diǎn)。代表是否是員工。

is_active：是否是可用的。對(duì)于一些想要?jiǎng)h除賬號(hào)的數(shù)據(jù)，我們?cè)O(shè)置這個(gè)值為False就可以了，而不是真正的從數(shù)據(jù)庫(kù)中刪除。

is_superuser：是否是超級(jí)管理員。如果是超級(jí)管理員，那么擁有整個(gè)網(wǎng)站的所有權(quán)限。

last_login：上次登錄的時(shí)間。

date_joined：賬號(hào)創(chuàng)建的時(shí)間。

登錄驗(yàn)證：

Django的驗(yàn)證系統(tǒng)已經(jīng)幫我們實(shí)現(xiàn)了登錄驗(yàn)證的功能。通過(guò)django.contrib.auth.authenticate即可實(shí)現(xiàn)。這個(gè)方法只能通過(guò)username和password來(lái)進(jìn)行驗(yàn)證。示例代碼如下：

from django.contrib.auth import authenticate
user = authenticate(username='Tom', password='111111')
# 如果驗(yàn)證通過(guò)了，那么就會(huì)返回一個(gè)user對(duì)象。
if user is not None:
  # 執(zhí)行驗(yàn)證通過(guò)后的代碼
else:
  # 執(zhí)行驗(yàn)證沒有通過(guò)的代碼。

擴(kuò)展用戶模型：

Django內(nèi)置的User模型雖然已經(jīng)足夠強(qiáng)大了。但是有時(shí)候還是不能滿足我們的需求。比如在驗(yàn)證用戶登錄的時(shí)候，他用的是用戶名作為驗(yàn)證，而我們有時(shí)候需要通過(guò)手機(jī)號(hào)碼或者郵箱來(lái)進(jìn)行驗(yàn)證。還有比如我們想要增加一些新的字段。那么這時(shí)候我們就需要擴(kuò)展用戶模型了。擴(kuò)展用戶模型有多種方式。這里我們來(lái)一一討論下。

1. 設(shè)置Proxy模型：

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

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

好處: 不破壞原來(lái)的User模型的表結(jié)構(gòu)

如果你對(duì)Django提供的字段，以及驗(yàn)證的方法都比較滿意，沒有什么需要改的。但是只是需要在他原有的基礎(chǔ)之上增加一些操作的方法。那么建議使用這種方式。示例代碼如下：

# models.py
class Person(User):
  # 如果模型是一個(gè)代理模型
  # 那么就不能在這個(gè)模型中添加新的Field
  # telephone = models.CharField(max_length=11) # 錯(cuò)誤寫法
  class Meta:
    proxy = True
 
  # proxy正確用法是給模型添加自定義方法
  # 如添加列出黑名單的方法
  def get_blacklist(self):
    return self.objects.filter(is_active=False)

在以上，我們定義了一個(gè)Person類，讓他繼承自User，并且在Meta中設(shè)置proxy=True，說(shuō)明這個(gè)只是User的一個(gè)代理模型。他并不會(huì)影響原來(lái)User模型在數(shù)據(jù)庫(kù)中表的結(jié)構(gòu)。以后如果你想方便的獲取所有黑名單的人，那么你就可以通過(guò)Person.get_blacklist()就可以獲取到。并且User.objects.all()和Person.objects.all()其實(shí)是等價(jià)的。因?yàn)樗麄兌际菑腢ser這個(gè)模型中獲取所有的數(shù)據(jù)。

2. 一對(duì)一外鍵：

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

局限: 只能增加, 不能減少字段, 不能修改戶驗(yàn)證方法: authenticate

好處: 不破壞原來(lái)的User模型的表結(jié)構(gòu)

如果你對(duì)用戶驗(yàn)證方法authenticate沒有其他要求，就是使用username和password即可完成。但是想要在原來(lái)模型的基礎(chǔ)之上添加新的字段，那么可以使用一對(duì)一外鍵的方式。示例代碼如下：

# 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()

以上定義一個(gè)UserExtension的模型，并且讓她和User模型進(jìn)行一對(duì)一的綁定，以后我們新增的字段，就添加到UserExtension上。并且還寫了一個(gè)接受保存模型的信號(hào)處理方法，只要是User調(diào)用了save方法，那么就會(huì)創(chuàng)建一個(gè)UserExtension和User進(jìn)行綁定。

# 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')
  # 給擴(kuò)展的字段設(shè)置值
  user.extension.school = 'Harvard'
  user.save()
  return HttpResponse('一對(duì)一擴(kuò)展User模型')

3. 繼承自AbstractUser：

作用: 給模型增加新的字段, 修改戶驗(yàn)證方法: authenticate

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

壞處: 破壞了原來(lái)的User模型的表結(jié)構(gòu)

對(duì)于authenticate不滿意，并且不想要修改原來(lái)User對(duì)象上的一些字段，但是想要增加一些字段，那么這時(shí)候可以直接繼承自django.contrib.auth.models.AbstractUser，其實(shí)這個(gè)類也是django.contrib.auth.models.User的父類。比如我們想要在原來(lái)User模型的基礎(chǔ)之上添加一個(gè)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, 而不是原來(lái)的username字段, 所以u(píng)sername要重寫
  username = models.CharField(max_length=150)
 
  # 指定telephone作為USERNAME_FIELD，以后使用authenticate
  # 函數(shù)驗(yàn)證的時(shí)候，就可以根據(jù)telephone來(lái)驗(yàn)證
  # 而不是原來(lái)的username
  USERNAME_FIELD = 'telephone'
  # USERNAME_FIELD對(duì)應(yīng)的'telephone'字段和密碼字段默認(rèn)是必須的字段
  # 下[]可以添加其它必須的字段, 比如['username', 'email']
  REQUIRED_FIELDS = []
 
  # 重新定義Manager對(duì)象，在創(chuàng)建user的時(shí)候使用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("請(qǐng)?zhí)钊胧謾C(jī)號(hào)碼！")
    if not password:
      raise ValueError("請(qǐng)?zhí)钊朊艽a!")
    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'

這種方式因?yàn)槠茐牧嗽瓉?lái)User模型的表結(jié)構(gòu)，所以必須要在第一次migrate前就先定義好。

4. 繼承自AbstractBaseUser模型：

作用: 給模型增加或減少字段, 修改戶驗(yàn)證方法: authenticate

壞處: 破壞了原來(lái)的User模型的表結(jié)構(gòu)

注意: 繼承自AbstractBaseUser同時(shí)還要繼承PermissionsMixin

如果你想修改默認(rèn)的驗(yàn)證方式，并且對(duì)于原來(lái)User模型上的一些字段不想要，那么可以自定義一個(gè)模型，然后繼承自AbstractBaseUser，再添加你想要的字段。這種方式會(huì)比較麻煩，最好是確定自己對(duì)Django比較了解才推薦使用。步驟如下：

創(chuàng)建模型。示例代碼如下：

# 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中已經(jīng)添加好了的，我們直接繼承就可以了。然后我們?cè)偬砑游覀兿胍淖侄?。比如email、username、telephone等。這樣就可以實(shí)現(xiàn)自己想要的字段了。但是因?yàn)槲覀冎貙懥薝ser，所以應(yīng)該盡可能的模擬User模型：

USERNAME_FIELD：用來(lái)描述User模型名字字段的字符串，作為唯一的標(biāo)識(shí)。如果沒有修改，那么會(huì)使用USERNAME來(lái)作為唯一字段。

REQUIRED_FIELDS：一個(gè)字段名列表，用于當(dāng)通過(guò)createsuperuser管理命令創(chuàng)建一個(gè)用戶時(shí)的提示。

is_active：一個(gè)布爾值，用于標(biāo)識(shí)用戶當(dāng)前是否可用。

get_full_name()：獲取完整的名字。

get_short_name()：一個(gè)比較簡(jiǎn)短的用戶名。

重新定義UserManager：我們還需要定義自己的UserManager，因?yàn)槟J(rèn)的UserManager在創(chuàng)建用戶的時(shí)候使用的是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("請(qǐng)?zhí)钊胧謾C(jī)號(hào)碼！")
    if not password:
      raise ValueError("請(qǐng)?zhí)钊朊艽a!")
    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)

在創(chuàng)建了新的User模型后，還需要在settings中配置好。配置AUTH_USER_MODEL='appname.User'。

# settings.py AUTH_USER_MODEL = 'youappname.User'

如何使用這個(gè)自定義的模型：比如以后我們有一個(gè)Article模型，需要通過(guò)外鍵引用這個(gè)User模型，那么可以通過(guò)以下兩種方式引用。

第一種就是直接將User導(dǎo)入到當(dāng)前文件中。示例代碼如下：

# 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抽象出來(lái)，使用settings.AUTH_USER_MODEL來(lái)表示。示例代碼如下：

# 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)

這種方式因?yàn)槠茐牧嗽瓉?lái)User模型的表結(jié)構(gòu)，所以必須要在第一次migrate前就先定義好。

以上這篇對(duì)django的User模型和四種擴(kuò)展/重寫方法小結(jié)就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: