flaskrestful响应处理

响应处理

1、序列化数据marchal工具，定义字典数据

装饰器形式marshal_with(fields,envelope='信封')

marshal(data,fields,envelop='')

#Flask-RESTful 响应处理
#1、marshal 序列化数据
from flask import Flask
from flask_restful import Api,Resource,marshal,marshal_with,fields
app = Flask(__name__)
api = Api(app)
#让后端返回的数据可以序列化
class User(object):
    def __init__(self,user_id,mobile,name):
        self.user_id = user_id
        self.mobile = mobile
        self.name = name
#序列化容器
resource_data = {
    'user_id':fields.Integer,
    'mobile':fields.String,
    'name':fields.String
}


class respResource(Resource):
    @marshal_with(resource_data,envelope='data')
    def get(self):
        #return {'data':'hello'}
        user = User('2022','13209880988','xh')
        #return user
        return marshal(user,resource_data)
api.add_resource(respResource,'/index')

if __name__ == '__main__':
    app.run(debug=True)

运行结果 使用postman工具

2、统一数据的json格式

找到flask-restful源码中的output_json函数

重写返回数据的方法data进行嵌套处理

手动调用装饰器

#统一数据交互的格式：返回json格式
from flask import Flask
from flask_restful import Api,Resource
app = Flask(__name__)
api = Api(app)

from flask import make_response,current_app
from flask_restful.utils import PY3
from  json import  dumps

@api.representation('application/json')
def output_json(data,code,headers=None):
    settings = current_app.config.get('RESTFUL_JSON',{})
    if 'message' not in data:
        data = {
            'message':'OK',
            'data':data
        }
    if current_app.debug:
        settings.setdefault('indent',4)
        settings.setdefault('sort_keys',not PY3)
    dumped = dumps(data,**settings)+"\n"

    resp = make_response(dumped,code)
    resp.headers.extend(headers or {})
    return resp
class JsonResource(Resource):
    def get(self):
        return {'data':'hello'}
api.add_resource(JsonResource,'/')
if __name__ == '__main__':
    app.run()

运行结果