1. 程式人生 > 程式設計 >Django如何繼承AbstractUser擴充套件欄位

Django如何繼承AbstractUser擴充套件欄位

使用django實現註冊登入的話,註冊登入都有現成的程式碼,主要是自帶的User欄位只有(email,username,password),所以需要擴充套件User,來增加自己需要的欄位

AbstractUser擴充套件模型User:如果模型User內建的方法符合開發需求,在不改變這些函式方法的情況下,新增模型User的額外欄位,可通過AbstractUser方式實現。使用AbstractUser定義的模型會替換原有模型User。

程式碼如下:

model.py

#coding:utf8
from django.db import models
from django.contrib.auth.models import AbstractUser
from django.utils.encoding import python_2_unicode_compatible
 
# Create your models here.
@python_2_unicode_compatible    
"""是django內建的相容python2和python3的unicode語法的一個裝飾器
只是針對 __str__ 方法而用的,__str__方法是為了後臺管理(admin)和django shell的顯示,Meta類也是為後臺顯示服務的
"""
class MyUser(AbstractUser):
  qq = models.CharField(u'qq號',max_length=16)
  weChat =models.CharField(u'微信賬號',max_length=100)
  mobile =models.CharField(u'手機號',primary_key=True,max_length=11)
  identicard =models.BooleanField(u'×××認證',default=False)               #預設是0,未認證, 1:×××認證, 2:視訊認證
  refuserid = models.CharField(u'推薦人ID',max_length=20)
  Level = models.CharField(u'使用者等級',default='0',max_length=2)            #預設是0,使用者等級0-9
  vevideo = models.BooleanField(u'視訊認證',default=False)           #預設是0,未認證。 1:已認證
  Type =models.CharField(u'使用者型別',max_length=1)             #預設是0,未認證, 1:刷手 2:商家
 
  def __str__(self):
    return self.username

settings.py

AUTH_USER_MODEL = 'appname.MyUser'
AUTHENTICATION_BACKENDS = ('django.contrib.auth.backends.ModelBackend',)

注意:

1、擴充套件user表後,要在settings.py 新增

AUTH_USER_MODEL = 'appname.擴充套件user的class name'

2、認證後臺要在settings新增,尤其記得加逗號,否則報錯

認證後臺不加的報錯

Django-AttributeError 'User' object has no attribute 'backend'

沒加逗號的報錯

ImportError: a doesn't look like a module path

form.py

#coding:utf-8
from django import forms
 
#登錄檔單
class RegisterForm(forms.Form):
  username = forms.CharField(label='使用者名稱',max_length=100)
  password = forms.CharField(label='密碼',widget=forms.PasswordInput())
  password2 = forms.CharField(label='確認密碼',widget=forms.PasswordInput())
  mobile = forms.CharField(label='手機號',max_length=11)
  email = forms.EmailField()
  qq = forms.CharField(label='QQ號',max_length=16)
  type = forms.ChoiceField(label='註冊型別',choices=(('buyer','買家'),('saler','商家')))
 
  def clean(self):
    if not self.is_valid():
      raise forms.ValidationError('所有項都為必填項')
    elif self.cleaned_data['password2'] != self.cleaned_data['password']:
      raise forms.ValidationError('兩次輸入密碼不一致')
    else:
      cleaned_data = super(RegisterForm,self).clean()
    return cleaned_data
 
#登陸表單
class LoginForm(forms.Form):
  username = forms.CharField(label='使用者名稱',widget=forms.TextInput(attrs={"placeholder": "使用者名稱","required": "required",}),max_length=50,error_messages={"required": "username不能為空",})
  password = forms.CharField(label='密碼',widget=forms.PasswordInput(attrs={"placeholder": "密碼",max_length=20,error_messages={"required": "password不能為空",})

遷移資料庫

python manage.py makemigrations
python manage.py migrate

views.py

from django.shortcuts import render,render_to_response
from .models import MyUser
from django.http import HttpResponse,HttpResponseRedirect
from django.template import RequestContext
import time
from .myclass import form
from django.template import RequestContext
from django.contrib.auth import authenticate,login,logout
 
#註冊
def register(request):
  error = []
  # if request.method == 'GET':
  #   return render_to_response('register.html',{'uf':uf})
  if request.method == 'POST':
    uf = form.RegisterForm(request.POST)
    if uf.is_valid():
      username = uf.cleaned_data['username']
      password = uf.cleaned_data['password']
      password2 = uf.cleaned_data['password2']
      qq = uf.cleaned_data['qq']
      email = uf.cleaned_data['email']
      mobile = uf.cleaned_data['mobile']
      type = uf.cleaned_data['type']
      if not MyUser.objects.all().filter(username=username):
        user = MyUser()
        user.username = username
        user.set_password(password)
        user.qq = qq
        user.email = email
        user.mobile = mobile
        user.type = type
        user.save()
        return render_to_response('member.html',{'username': username})
  else:
    uf = form.RegisterForm()
  return render_to_response('register.html',{'uf':uf,'error':error})
 
#登陸  
def do_login(request):
  if request.method =='POST':
    lf = form.LoginForm(request.POST)
    if lf.is_valid():
      username = lf.cleaned_data['username']
      password = lf.cleaned_data['password']
      user = authenticate(username=username,password=password)        #django自帶auth驗證使用者名稱密碼
      if user is not None:                         #判斷使用者是否存在
        if user.is_active:                         #判斷使用者是否啟用
          login(request,user)                         #使用者資訊驗證成功後把登陸資訊寫入session
          return render_to_response("member.html",{'username':username})
        else:
          return render_to_response('disable.html',{'username':username})
      else:
        return HttpResponse("無效的使用者名稱或者密碼!!!")
  else:
    lf = form.LoginForm()
  return render_to_response('index.html',{'lf':lf})
   
#退出
def do_logout(request):
  logout(request)
  return HttpResponseRedirect('/')

注意:

1、登陸的時候用自帶的認證模組總是報none

user = authenticate(username=username,password=password)
print(user)

檢視原始碼發現是check_password的方法是用hash進行校驗,之前註冊的password寫法是

user.password=password

這種寫法是明文入庫,需要更改密碼的入庫寫法

user.set_password(password)

補充

一個快速拿到User表的方法,特別在擴充套件User表時,你在settings.py配置的User。

from django.contrib.auth import get_user_model
User = get_user_model()

別在其他檢視或者模型裡匯入你擴充套件的MyUser model。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。