我正在为我的应用程序编写Falcon
中间件。当我得到任何错误时,我想提出错误,中断进程并返回我的自定义响应,如下所示:
{
"status": 503,
"message": "No Token found. Token is required."
}
但标准Falcon
错误实现不允许我将自定义字段设置为我的回复。
如何正确解决这个问题?
答案 0 :(得分:4)
花了很多时间后,我以这种有趣的方式解决了这个问题。我将我的代码放在try / catch块中,当发现错误时我决定不提出Falcon
错误,并且在设置响应状态和正文后尝试写return
关键字,因为该方法是void
,所以它不会返回任何内容。现在看起来像:
resp.status = falcon.HTTP_403
resp.body = body
return
答案 1 :(得分:2)
我还在寻找一个例子,这里适合任何仍需要它的人:
from falcon.http_error import HTTPError
class MyHTTPError(HTTPError):
"""Represents a generic HTTP error.
"""
def __init__(self, status, error):
super(MyHTTPError, self).__init__(status)
self.status = status
self.error = error
def to_dict(self, obj_type=dict):
"""Returns a basic dictionary representing the error.
"""
super(MyHTTPError, self).to_dict(obj_type)
obj = self.error
return obj
使用:
error = {"error": [{"message": "Auth token required", "code": "INVALID_HEADER"}]}
raise MyHTTPError(falcon.HTTP_400, error)
答案 2 :(得分:1)
创建falcon文档中解释的自定义异常类,搜索add_error_handler
class RaiseUnauthorizedException(Exception):
def handle(ex, req, resp, params):
resp.status = falcon.HTTP_401
response = json.loads(json.dumps(ast.literal_eval(str(ex))))
resp.body = json.dumps(response)
将自定义异常类添加到falcon API对象
api = falcon.API()
api.add_error_handler(RaiseUnauthorizedException)
答案 3 :(得分:0)
raise falcon.HTTPError(falcon.HTTP_503, 'No Token found. Token is required.')