(二) 引入错误类和返回类，概念，统一返回格式化内容

创建一个 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}))

View Code

使用示例：

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)

View Code

这样的话可以统一返回格式。