3

我有这样的代码

def find_user(user_id):
    if not users.lookup(user_id):
        return jsonify(success=False, reason="not found")
    else:
        return users.lookup(user_id)

@app.route(...)
def view1(...):
    user = find_user(...)
    if user:
       do_something
    return jsonify(success=True, ....)

关键是我有一个辅助函数，如果condition满足我想立即返回响应。在这种情况下，如果找不到用户，我想返回一个请求，说明请求失败并给出原因。

问题是因为该函数还可以返回一些其他值（比如用户对象）。我不会从请求中返回。相反，因为我们得到了一个分配，我们将继续使用其余的代码路径。

我可能可以进行类型检查，但这很难看！

理想情况下，这是更长的代码

def view(...)
    user = users.lookup(user_id)
    if not user:
       return jsonify(success=False....)
    # if user is true, do something else

辅助函数的重点find_user是提取通用代码，然后在其他一些类似的 api 中重用。

有没有办法直接在我当前的代码中返回响应？看起来这是一个死胡同......除非我改变代码的编写方式（如何返回值......）

4

2 回答 2

10

您应该在 中引发异常find_user，然后添加一个处理程序，该处理程序为该异常返回正确的响应。来自http://flask.pocoo.org/docs/patterns/apierrors/的示例：

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

@app.errorhandler(InvalidAPIUsage)
def handle_invalid_usage(error):
    response = jsonify(error.to_dict())
    response.status_code = error.status_code
    return response

使用此示例，在您的find_user函数中，您可以执行以下操作：

raise InvalidUsage('User not found', status_code=404)
于 2013-09-13T05:53:29.743 回答
2

根据这个答案，您也可以简单地 

return jsonify(success=False....), 400

返回状态码 400。

于 2016-01-06T11:11:48.487 回答