Flask入门篇（四）自定义异常和响应信息

from flask import Flask, abort, request, Response, make_response


app = Flask(__name__)


"""
abort方法类似于raise，立即终止函数并返回特定的信息给前端
解析1:abort函数可以接收http标准状态码
解析2:abort函数可以接收响应头，即Response对象
"""
@app.route("/login")
def login():
    uid = request.form.get("uid")
    pwd = request.form.get("pwd")
    if uid == "zhangsan" and pwd == "123456":
        return "登陆成功"
    else:
        abort(404)
        resp = Response("登陆失败")
        abort(resp)


"""自定义异常"""
@app.errorhandler(404)
def handle_err(err):
    return "抛出异常，错误信息:%s" % err


"""自定义响应信息"""
@app.route("/index")
def index():
    # 1.以元祖或字典的形式返回，元祖的顺序分别是响应体，状态码，响应头
    return "响应体", 400, [("name", "zhangsn"), ("age", 23)]
    return "响应体", "666 notok", [("name", "zhangsn"), ("age", 23)]
    # 2.用make_response的方法返回
    response = make_response("响应体")
    response.status = "999 notok"
    response.headers["name"] = "zhangsan"  # 不支持中文
    response.headers["age"] = 23
    return response


if __name__ == '__main__':
    app.run(debug=True)