(二) 引入错误类和返回类,概念,统一返回格式化内容
创建一个 utils 文件,
1 from flask import make_response, jsonify 2 from enum import Enum 3 # 规定错误类, 采用枚举的方式 4 class ErrCode(Enum): 5 success = {"200": True} 6 is_register={"1002": "this phone is is register"} 7 no_login = {"1001": "this user is no login"} 8 9 def get_code(self): 10 return list(self.value.keys())[0] 11 12 def get_msg(self): 13 return list(self.value.values())[0] 14 15 # 规定返回格式 16 class ReturnValue: 17 def set_value(success, status, data): 18 errcode = ErrCode[status].get_code() 19 errmsg = ErrCode[status].get_msg() 20 return make_response(jsonify({"success": success, "errcode": errcode, "errmsg": errmsg, "data": data}))
使用示例:
1 from app.model.user import Users 2 from app.utils.errcode import ReturnValue 3 4 5 def index(): 6 users = Users.query.filter().first() 7 return ReturnValue.set_value(success=False, status="no_login", data=users.username)
这样的话可以统一返回格式。