1. 程式人生 > >Python進階-Flask鉤子應用場景及使用介紹

Python進階-Flask鉤子應用場景及使用介紹

在正常執行的程式碼前中後,強行插入執行一段你想要實現的功能的程式碼,這種函式就叫做鉤子函式。鉤子函式就是等同於高速公路上的收費站,進高速之前給你一個卡,並檢查你是否超重。離開之前收你,也可以攔住你安檢一下。

一、基礎概念:

request: Flask的請求上下文,包含請求變數如:method、args、form、values、endpoint、headers、remote_addr都是比較常用的。

session:Flask的請求上下文,用於存放使用者的會話資訊。

current_app:Flask的應用上下文,返回當前app的方法和屬性,可以勉強理解為類全域性變數。

二、七種鉤子

第一個鉤子:@app.before_first_request

只在第一次請求之前執行,也就是啟動專案,不會執行,只會在第一次有人發起請求時,才會觸發這個鉤子中的程式碼。

全域性場景:可以帶動一個非同步執行的函式,進行一些健康指標的檢查,如果發現有異常,則截斷後續的請求,將整個Flask應用停止。

@app.before_first_request
def first_request():
    print('只有在處理第一次請求之前執行')

第二個鉤子:@app.before_request

這是最重要的一個鉤子,在每次請求之前可以注入你要的邏輯的鉤子。在app下的before_request,過濾的是全部請求。結合Blueprint的before_request,則是過濾該藍圖下的請求。所以我們就可以進行分層過濾,定製化過濾。

全域性的場景包含:共享session的鑑權函式、請求黑白名單過濾、根據endpoint進行請求j等。

藍圖場景包含api的請求必填欄位校驗,是否json請求校驗,請求的token校驗等。

api = Blueprint('api', __name__)
requied = {
    'api.register':['email','username','password']
}

# 鉤子 在請求執行之前
@api.before_request
def before_request():

    # 請求格式校驗攔截
    if not request.is_json:
        return '帶引數請求請使用json格式'
    # 缺少必填引數攔截
    try:
        if request.endpoint in requied:
            if request.method == "POST":
                missparam_list = [x for x in requied[request.endpoint] if x.encode('utf8') not in list(parse.parse_qs(request.data).keys())]
            else:
                missparam_list = [x for x in requied[request.endpoint] if x not in request.json.keys()]
    
            if len(missparam_list) > 0:
                return "缺少以下引數:{0}"
    except Exception as e:
        app.logger.error(e)
        return "{0}".format(e)

第三個鉤子:@app.errorhandler

當訪問應用出錯時,根據錯誤響應碼,進行一些定製化的操作,如返回一個可愛的404頁面。也可以進行一些報錯登記。

場景:可以用redis進行錯誤請求計數,超過一定量則進行告警。可以重定向到一個定製的錯誤程式碼頁面等。

@app.errorhandler(404)
def page_not_found(error):
    return render_template('otherpage/404.html'),404

第四個鉤子:@app.context_processor

這個鉤子也很實用,是將一些常量按字典的格式返回,則可以在jinja2的模版中引用。這樣就不用在每個檢視函式中都render_template中重複去寫一次。程式碼更簡潔。

場景:在html中,直接用{{jidan}}就會在頁面顯示yao。等同於app.add_template_global('yao',''jidan)

@app.context_processor
def context_rocessor():
    return {'jidan':'yao'}

第五個鉤子:@app.after_request

和上個鉤子類似,差別在於是請求完成時執行,它和之前鉤子有點不同,必須傳入一個引數來接收響應物件,並在最後return 這個引數,也就是返回響應內容。

場景:一般用於格式化響應結果,包括響應請求頭,響應的格式等。

@app.after_request
def after_request(response):
    response.headers['jidan'] = 'yaoyao'
    return response

第六個鉤子@app.teardown_request

和第五個鉤子功能類似,在響應銷燬時,執行一個繫結的函式。做一些操作。

區別點在於:

after_request: 每一個請求之後繫結一個函式,如果請求沒有異常。

teardown_request: 每一個請求之後繫結一個函式,即使遇到了異常。

場景:銷燬DB連線等。

@app.teardown_request
def teardown_db(exception):
    db = getattr(g, 'database', None)
    if db is not None:
        db.close()

第七個鉤子@app.teardown_appcontext

之前介紹的大部分是請求上下文的鉤子,這個屬於應用上下文的鉤子。不管是否有異常,當APP上下文被移除之後執行的函式, 可以進行資料庫的提交或者回滾。

場景:DB事務操作。

@app.teardown_appcontext
def teardown(cmd=None):
    if cmd is None:
        db.session.commit()
    else:
        db.session.rollback()
    db.session.remove()