Django auth模块

auth

auth_user

 

我们在开发一个网站的时候，无可避免的需要设计实现网站的用户系统。此时我们需要实现包括用户注册、用户登录、用户认证、注销、修改密码等功能，这还真是个麻烦的事情呢。

Django作为一个完美主义者的终极框架，当然也会想到用户的这些痛点。它内置了强大的用户认证系统--auth，它默认使用 auth_user 表来存储用户数据。

auth模块

from django.contrib import auth

auth提供的方法：

authenticate()

提供了用户认证功能，即验证用户名以及密码是否正确，一般需要username 、password两个关键字参数。

如果认证成功（用户名和密码正确有效），便会返回一个 User 对象。

authenticate()会在该 User 对象上设置一个属性来标识后端已经认证了该用户，且该信息在后续的登录过程中是需要的。

用法：

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

login(HttpRequest,user)

该函数接受一个HttpRequest对象，以及一个经过认证的User对象。

该函数实现一个用户登录的功能。它本质上会在后端为该用户生成相关session数据。

from django.contrib import auth

def login_view(request):
    username = request.POST['username']
    password = request.POST['password']
    user_obj = auth.authenticate(username=username,password=pwd)

    if user_obj:
        auth.login(request,user_obj)
        return redirect(next_url)
    else:
        ...

logout(request)

该函数接受一个HttpRequest对象，无返回值。

当调用该函数时，当前请求的session信息会全部清除。该用户即使没有登录，使用该函数也不会报错。

用法：

from django.contrib import auth

def logout(request):
    auth.logout(request)
    return redirect('/login/')

is_authenticated()

判断当前请求是否通过了认证

用法：

def my_view(request):
    if not request.user.is_authenticated():
        return redirect('%s?next=%s' % (settings.LOGIN_URL, request.path))

login_requierd()

auth给我们提供的一个装饰器工具，用来快捷的给某个视图添加登录校验

用法：

from django.contrib.auth.decorators import login_required

@login_required
def my_view(request):
    ...

注意：

如果用户没有登录，默认会跳转到Django的默认登录url '/accounts/login/'，如自定义登录url，在settings.py中通过LOGIN_URL修改。

LOGIN_URL = '/login/'

create_user()

auth提供一个创建新用户的方法，需要提供必要参数(username,password)等，密码是加密的

用法：

from django.contrib.auth.models import User

user = User.objects.create_user(username='用户名',password='密码')

create_superuser()

auth提供的一个创建新的超级用户的方法，需要提供必要参数(username,password)等

用法：

from django.contrib.auth.models import User

user = User.objects.create_superuser(username='用户名',password='密码')

check_password(password)

auth 提供的一个检查密码是否正确的方法，需要提供当前请求用户的密码

密码正确返回True，否则返回False。

用法：

ok = User.check_password('密码')

set_password(password)

auth 提供的一个修改密码的方法，接收 要设置的新密码 作为参数

注意：设置完一定要调用用户对象的save方法！！！

用法：

User.set_password(password='')
User.save()

一个修改密码功能的简单示例

from django.contrib.auth.models import User


@login_required
def set_password(request):
    user = request.user
    err_msg = ''
    if request.method == 'POST':
        old_password = request.POST.get('old_password', '')
        new_password = request.POST.get('new_password', '')
        repeat_password = request.POST.get('repeat_password', '')
        # 检查旧密码是否正确
        if User.check_password(old_password):
            if not new_password:
                err_msg = '新密码不能为空'
            elif new_password != repeat_password:
                err_msg = '两次密码不一致'
            else:
                User.set_password(new_password)
                User.save()
                return redirect("/login/")
        else:
            err_msg = '原密码输入错误'
    content = {
        'err_msg': err_msg,
    }
    return render(request, 'set_password.html', content)

User对象的属性

User对象属性：username， password

is_staff ： 用户是否拥有网站的管理权限.

is_active ： 是否允许用户登录, 设置为 False，可以在不删除用户的前提下禁止用户登录。

扩展默认的auth_user表

默认的内置认证系统auth_user表字段是固定的。比如，我想要加一个存储用户手机号的字段，怎么办？
我们可以通过继承内置的 AbstractUser 类，来定义一个自己的Model类。
比默认的auth_user表多了一个 phone 字段

from django.contrib.auth.models import AbstractUser

class UserInfo(AbstractUser):
    phone = models.CharField(max_length=11,null=True,unique=True)

    def __str__(self):
        return self.username

注意：

一旦指定了新的认证系统所用的表，就需要重新在数据库中创建该表，而不能继续使用原来默认的auth_user表了

按上面的方式扩展了内置的auth_user表之后，一定要在settings.py中告诉Django，我现在使用新定义的UserInfo表来做用户认证。写法如下：

#引用Django自带的User表，继承使用时需要设置
AUTH_USER_MODEL = "app名.UserInfo"

#执行下面两条命令

python manage.py makemigrations

python manage.py migrate

 生成验证码

pip install pillow

#login.html 中关于验证码

<div class="form-group">
     <label for="id_vcode" style="display: block">验证码</label>
     <input type="text" class="form-control" id="id_vcode" name="vcode" placeholder="验证码" style="width: 250px;display: inline-block">
     <img src="/vcode/" style="width: 250px;height: 35px; display: inline-block; float: right">
</div>

 生成验证码函数

# 专门返回验证码图片的视图
def vcode(request):
    from PIL import Image,ImageDraw,ImageFont
    def vcode_col():
        return random.randint(0,255),random.randint(0,255),random.randint(0,255)
    # 创建一个图片对象
    image_obj = Image.new(
        'RGB',
        (250,35),
        (192,192,192)
    )
    # 在该图片对象上生成一个画笔对象
    draw_obj = ImageDraw.Draw(image_obj)
    # 加载一个字体对象
    font_obj = ImageFont.truetype('static/font/kumo.ttf',34)
    # 生成验证码
    tmp = []
    for i in range(5):
        l = chr(random.randint(97,122))
        u = chr(random.randint(65,90))
        n = str(random.randint(0,9))
        r = random.choice([l,u,n])
        # 将选中的字符写在图片上
        # draw_obj.text((40*i+20,0),r,fill=vcode_col(),font=font_obj)
        draw_obj.text((40*i+20,0),r,fill=(0,0,0),font=font_obj)
        tmp.append(r)
    v_code = "".join(tmp).upper()
    # 将生成的验证码保存
    request.session["v_code"] = v_code

    #直接在内存中保存图片，替代io操作
    from io import BytesIO
    f1 = BytesIO()
    image_obj.save(f1,format="PNG")
    img_data = f1.getvalue()

    return HttpResponse(img_data,content_type="image/png")