接着上节,我们通过jsonify返回json数据非常方便
... return jsonify({"code": 400, "message": message})
返回什么数据则是公司接口编写的规范
返回值的规范(就算值为空,我们也必须返回以下字段)
{ "code": 200, #状态码 "message": "", #信息提示 "data": {}, #返回的数据,比如文章列表等等 }
状态码的规范
200:成功 401:没有授权 400:参数错误 405:方法错误 500:服务器错误
前面,我们每一次返回json数据需要写一串"jsonify({"code": 400, "message": message})"这样很麻烦,项目后期可能还有很多地方需要返回json。因此我们可以把它抽取出来进行封装成一个工具
1、先在项目下面建一个python package命名为utils(工具包,以后项目的工具都写在这里面)
2、在utils.py下新建xjson.py
from flask import jsonify class StatusCode(object): ok = 200 paramserror = 400 unauth = 401 methoderror = 405 servererror = 500 def json_result(code, message, data): return jsonify({"code": code, "message":message, "data":data or {}})
这样,我们就可以在视图返回json的时候这样用:
from utils import xjson ... return xjson.json_result(code=200, message='', data={})
但是这样依然不够方便,我们还是需要加上3个参数。因此我们继续改造xjson.py
from flask import jsonify class StatusCode(object): ok = 200 paramserror = 400 unauth = 401 methoderror = 405 servererror = 500 def json_result(code, message, data): return jsonify({"code": code, "message":message, "data":data or {}}) def json_success(message='', data=None): return json_result(code=StatusCode.ok, message=message, data=data) def json_params_error(message='', data=None): """ 请求参数错误 """ return json_result(code=StatusCode.paramserror, message=message, data=data) def json_unauth_error(message='', data=None): """ 没有权限访问 """ return json_result(code=StatusCode.unauth, message=message, data=data) def json_method_error(message='', data=None): """ 请求方法错误 """ return json_result(code=StatusCode.methoderror, message=message, data=data) def json_server_error(message='', data=None): """ 服务器内部错误 """ return json_result(code=StatusCode.servererror, message=message, data=data)
这样,我们只要根据不同的情况使用定义好的函数就行了。
参数: code(已经不用自己传了),message(成功的状态下可以不传,默认为空),data(没有传此参数,也会自动为空字典))
把上节内容的代码,可以写成如下:
... from utils import xjson from class ResetPwdView(views.MethodView): decorators = [login_required] def get(self): return render_template('cms/cms_resetpwd.html') def post(self): resetpwd_form = ResetPwdForm(request.form) if resetpwd_form.validate(): oldpwd = resetpwd_form.oldpwd.data newpwd = resetpwd_form.newpwd.data user = g.cms_user if user.check_password(oldpwd): user.password = newpwd db.session.commit() return xjson.json_success('修改成功') else: return xjson.json_params_error('原密码错误') else: message = resetpwd_form.get_error() return xjson.json_params_error(message)