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。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。