10 django jsonschema

Posted 2021-05-19 栗子测试开发

django jsonschema

1 定义login schema

login_schema = {
    "$schema": "http://json-schema.org/draft-07/schema#",
    "title": "login",
    "type": "object",
    "properties": {
        "username": {"type": "string"},
        "password": {"type": "string"}
    },
    \'required\': [\'username\', \'password\']
}

 

2 自定义异常

def invalid_json():
    result = {
        \'code\': -1,
        \'msg\': \'Invalid JSON\'
    }
    return result

 

3 封装后端参数校验装饰器

import json
from jsonschema import ValidationError, validate
from django.http import JsonResponse
from common.my_exception import invalid_json

def json_validate(schema):
    def valieated_func(func):
        def _func(self, request, *args, **kwargs):
            try:
                validate(json.loads(request.body), schema)
            except ValidationError as e:
                return JsonResponse(invalid_json(), safe=False)
            else:
                return func(self, request, *args, **kwargs)
        return _func
    return valieated_func

 

4 对登录参数进行校验

class Login(View):
    @json_validate(login_schema)
    def post(self, request):
        json_obj = json.loads(request.body)  # 转字典
        try:
            username = json_obj[\'username\']
            password = json_obj[\'password\']
        except Exception as e:
            result = {\'code\': -1, \'msg\': u\'用户名或密码为空！\'}
            return JsonResponse(result)

        users = User.objects.filter(username=username)  # 返回数组
        if users is None:
            result = {\'code\': -1, \'msg\': u\'用户名或密码错误！\'}
            return JsonResponse(result)
        user = users[0]
        m = hashlib.md5()
        m.update(password.encode())
        # 密码比对
        if m.hexdigest() != user.password:
            result = {\'code\': -1, \'msg\': u\'用户名或密码错误！\'}
            return JsonResponse(result)

        # 生成 token
        token = make_token(username)
        # encode的作用是将unicode编码的字符串编码成二进制数据
        # decode的作用是将二进制数据解码成unicode编码
        result = {\'code\': 1, \'data\': {\'token\': token}, \'msg\': u\'登录成功！\'}
        return JsonResponse(result)