Django如何繼承AbstractUser擴展字段
使用django實現注冊登錄的話,注冊登錄都有現成的代碼,主要是自帶的User字段只有(email,username,password),所以需要擴展User,來增加自己需要的字段
AbstractUser擴展模型User:如果模型User內置的方法符合開發需求,在不改變這些函數方法的情況下,添加模型User的額外字段,可通過AbstractUser方式實現。使用AbstractUser定義的模型會替換原有模型User。
代碼如下:
model.py
#coding:utf8from django.db import modelsfrom django.contrib.auth.models import AbstractUserfrom 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’用戶類型’, default=’0’, 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-8from 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': '密碼', 'required': 'required',}),max_length=20, error_messages={'required': 'password不能為空',})
遷移數據庫
python manage.py makemigrationspython manage.py migrate
views.py
from django.shortcuts import render,render_to_responsefrom .models import MyUserfrom django.http import HttpResponse,HttpResponseRedirectfrom django.template import RequestContextimport timefrom .myclass import formfrom django.template import RequestContextfrom 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 = usernameuser.set_password(password)user.qq = qquser.email = emailuser.mobile = mobileuser.type = typeuser.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_modelUser = get_user_model()
別在其他視圖或者模型里導入你擴展的MyUser model。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持好吧啦網。
相關文章:
