drf续集之 - 多方式登录、批量操作

最新推荐文章于 2023-06-21 11:01:48 发布

跃上青空

最新推荐文章于 2023-06-21 11:01:48 发布

阅读量682

点赞数

分类专栏：如今的江湖文章标签： drf django

本文链接：https://blog.csdn.net/wuzeipero/article/details/111413451

版权

如今的江湖专栏收录该内容

84 篇文章 5 订阅

订阅专栏

多方式登录

DRF - 多方式登录
DRF - 批量操作
- 代码

DRF - 多方式登录

用户可以以用户名、邮箱、手机号登录

之前写的逻辑处理一般都是在视图类中，现在把逻辑拿到了序列化类的validate全局钩子中，在视图函数中就无需写逻辑了，执行到.is_valid就会执行validate方法

重点：视图类和序列化类之间交互数据的桥梁：context={}（是1个字典）

自定义用户表，登录手动签发token，自定义的认证类

认证通过，正常情况下查询得到当前登录用户

效果

用户名登录
在这里插入图片描述
邮箱登录

手机号登录

代码

`models.py`

from django.db import models


class UserInfo(models.Model):
    username = models.CharField(max_length=16, verbose_name='用户名')
    password = models.CharField(max_length=16, verbose_name='密码')
    phone = models.CharField(max_length=11, verbose_name='手机号')
    email = models.EmailField(verbose_name='邮箱')

    class Meta:
        verbose_name_plural = '用户信息表'

    def __str__(self):
        return self.username

`serializer.py`

import re

from rest_framework_jwt.utils import jwt_encode_handler, jwt_payload_handler
from rest_framework.exceptions import ValidationError
from rest_framework import serializers

from multiple_login.models import UserInfo


class LoginSerializer(serializers.ModelSerializer):
    username = serializers.CharField()

    class Meta:
        model = UserInfo
        fields = ['username', 'password']

    def validate(self, attrs):
        # username可能是邮箱，手机号，用户名
        username = attrs.get('username')
        password = attrs.get('password')
        # 如果是手机号
        if re.match('^1[3-9]\d{9}$', username):
            # 以手机号登录
            user = UserInfo.objects.filter(phone=username).first()
        elif re.match('^.+@.+$', username):
            # 以邮箱登录
            user = UserInfo.objects.filter(email=username).first()
        else:
            # 以用户名登录
            user = UserInfo.objects.filter(username=username).first()
        # 如果user有值并且密码正确
        if user and user.check_password(password):
            # 登录成功，生成token
            # drf-jwt中有通过user对象生成token的方法
            payload = jwt_payload_handler(user)
            token = jwt_encode_handler(payload)
            # token是要在视图类中使用，现在我们在序列化类中
            # self.context.get('request')
            # 视图类和序列化类之间通过context这个字典来传递数据
            self.context['token'] = token
            self.context['username'] = user.username
            return attrs

        else:
            raise ValidationError('用户名或密码错误')

`utils.py`

from rest_framework.views import exception_handler
from rest_framework.response import Response


class APIResponse(Response):
    def __init__(self, code=100, msg='成功', data=None, status=None,
                 headers=None, content_type=None, **kwargs):
        dic = {'code': code, 'msg': msg}
        if data:
            dic['data'] = data
        dic.update(kwargs)
        super().__init__(data=dic, status=status, headers=headers, content_type=content_type)


def common_exception(exc, context):
    # 先调用REST framework默认的异常处理方法获得标准错误响应对象
    response = exception_handler(exc, context)
    # 在此处补充自定义的异常处理
    if response is None:
        response = Response(data={'code': 999, 'msg': str(exc)})

    return response

`auth.py`

import jwt

from rest_framework_jwt.authentication import BaseJSONWebTokenAuthentication
from rest_framework.exceptions import AuthenticationFailed
from rest_framework_jwt.utils import jwt_decode_handler


class JwtAuthentication(BaseJSONWebTokenAuthentication):
    def authenticate(self, request):
        token = request.META.get('HTTP_Authorization'.upper())
        try:
            payload = jwt_decode_handler(token)
        except jwt.ExpiredSignature:
            raise AuthenticationFailed('过期了')
        except jwt.DecodeError:
            raise AuthenticationFailed('解码错误')
        except jwt.InvalidTokenError:
            raise AuthenticationFailed('不合法的token')
        # 得到的user对象，应该是自己user表的user对象
        # user=MyUser.objects.get(id=payload['user_id'])
        user = payload

        return (user, token)

`views.py`

import re

from rest_framework_jwt.settings import api_settings
from rest_framework.views import APIView

from multiple_login.utils import APIResponse
from multiple_login.models import UserInfo

jwt_payload_handler = api_settings.JWT_PAYLOAD_HANDLER
jwt_encode_handler = api_settings.JWT_ENCODE_HANDLER


class MyLoginView(APIView):
    def post(self, request, *args, **kwargs):
        username = request.data.get('username')
        password = request.data.get('password')
        # 如果是手机号
        if re.match('^1[3-9]\d{9}$', username):
            # 以手机号登录
            user = UserInfo.objects.filter(phone=username).first()
        elif re.match('^.+@.+$', username):
            # 以邮箱登录
            user = UserInfo.objects.filter(email=username).first()
        else:
            # 以用户名登录
            user = UserInfo.objects.filter(username=username).first()
        # 如果user有值并且密码正确
        if user and user.password == password:
            # 登录成功，生成token
            # drf-jwt中有通过user对象生成token的方法
            payload = jwt_payload_handler(user)
            token = jwt_encode_handler(payload)
            return APIResponse(token=token, username=user.username)
        else:
            return APIResponse(code=101, msg='用户名或密码错误')

`urls.py`

from django.contrib import admin
from django.urls import path

from multiple_login import views

urlpatterns = [
    path('admin/', admin.site.urls),
    path('login/', views.MyLoginView.as_view()),
]

`settings.py`

REST_FRAMEWORK = {
    # 配置全局异常
    'EXCEPTION_HANDLER': 'multiple_login.utils.common_exception'
}

DRF - 批量操作

功能：群增、群删、群改、群查

代码