flask设置统一的响应返回

第一种方法:

status_msg = {
    200: '成功!',
    10000: '参数数据不完整',
    10011: '用户名不合法',
    10012: '密码不合法',
    10013: '二次密码不一致',
    10014: '手机号不合法',
    10015: '邮箱不合法',
    10016: 'token已过期,无法使用',
    10017: '请先登录后在使用',
    10018: '该用户不存在!',
    10019: '该用户已存在!',
    10020: '文件不存在!',
    20000: '异常错误'}


def to_dict_msg(status=200, data=None, msg=None):
    return {
        'status': status,
        'data': data,
        'msg': msg if msg else status_msg.get(status)
    }

第二种:

from flask import make_response, jsonify


class JsonResponse(object):
    """
    统一的json返回格式
    """

    def __init__(self, code=200, data=None, msg=''):
        self.data = data
        self.code = code
        self.msg = msg

    @classmethod
    def success(cls, code=200, data=None, msg='success'):
        return cls(code, data, msg)

    @classmethod
    def error(cls, code=20000, data=None, msg='error'):
        return cls(code, data, msg)

    def to_dict(self):
        return {
            "code": self.code,
            "msg": self.msg,
            "data": self.data
        }

    def to_response(self):
        response_data = self.to_dict()
        response = make_response(jsonify(response_data))
        response.headers['Content-Type'] = 'application/json'
        return response

你可能感兴趣的:(Flask,flask,python,后端)