Django中通过AbstractUser扩展用户模型及认证配置详解
本文详解如何在Django中利用AbstractUser类扩展内置User模型,增加QQ、微信等自定义字段。重点讲解settings.py中AUTH_USER_MODEL的配置、认证后端AUTHENTICATION_BACKENDS的逗号规范,以及解决密码明文存储和认证报错的实际代码实现,帮助开发者避免常见配置陷阱。
Django用户模型扩展概述
在Django项目中实现注册与登录功能时,系统默认提供的User模型仅包含email、username和password字段。若业务需求涉及更多用户信息,则需基于AbstractUser对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'用户类型', 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-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": "密码", "required": "required",}),
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、使用Django内置认证模块进行登录时,若处理不当常返回None。
user = authenticate(username=username, password=password) print(user)
查阅源码可知,check_password方法依赖哈希值进行校验。若注册时直接赋值密码,会导致明文入库,因此必须修改密码存储方式。
user.password=password
上述写法为明文存储,不符合安全规范,需更改为以下调用方式:
user.set_password(password)
补充说明:
在扩展User表后,获取User模型实例的推荐方法是通过get_user_model(),而非直接导入自定义模型类。
from django.contrib.auth import get_user_model User = get_user_model()
请避免在其他视图或模型中直接导入自定义的MyUser模型,应统一使用get_user_model()获取。
以上内容涵盖了Django扩展用户模型的核心配置与注意事项,希望能为相关开发工作提供参考。
免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。









