在Flask上层实现 RESTful APIs 是非常常见的。开发者碰到的第一件事情就是内置的异常对于 APIs
是表达性不足并且使用的内容格式 text/html
对于 API 使用者用处不大。
比 abort
更好的解决方案就是实现自己的异常类型和安装的错误处理器,产生用户期待错误的格式。
简单的异常类¶
其基本思路是引入一个新的异常,该异常可以采取一个适当的人类可读的消息,一个状态的错误代码和
一些可选的提供更多的错误上下文内容。
这是一个简单的例子:
from flask import jsonify
class InvalidUsage(Exception):
status_code = 400
def __init__(self, message, status_code=None, payload=None):
Exception.__init__(self)
self.message = message
if status_code is not None:
self.status_code = status_code
self.payload = payload
def to_dict(self):
rv = dict(self.payload or ())
rv['message'] = self.message
return rv
视图现在可以引起一个带有错误信息的异常。另外一些附加的东西可以通过 payload 参数以字典形式提供。
注册一个错误处理器¶
在这个时候视图可以引起一个错误,但是它会立即导致内部服务器错误。原因是没有为这个错误类注册处理器。:
@app.errorhandler(InvalidAPIUsage)
def handle_invalid_usage(error):
response = jsonify(error.to_dict())
response.status_code = error.status_code
return response
视图中的用法¶
这里是视图使用这个函数的方法:
@app.route('/foo')
def get_foo():
raise InvalidUsage('This view is gone', status_code=410)
转载请注明:www.ainoob.cn » Flask教程:实现 API 异常