在 Flask 中为我的 python 应用程序添加缓存控制

Question

我正在对我的网站进行一些测试。我的应用程序是使用烧瓶开发的，我想在下面的函数中添加 cach-control "max-age=3600" :

from flask import Flask, jsonify, abort   
from waitress import serve
import pandas as pd
from time import time as t

from load_data.connexion_to_blob import connexion_to_blob
from load_data.load_recos import load_recos

# CONFIG FILE
import configparser
config = configparser.ConfigParser()
config.read('config.ini')

# CHARGE  DATA FROM BLOB
mode = 'blob'
blob_service_client = connexion_to_blob("AZURE_STORAGE_CONNECTION_STRING")

load_recos(blob_service_client, langue='fr', mode='blob')


# LOAD DATA IN APP
recos_fr_file = config['PATH']['recos_fr_file']

recos_fr = pd.read_pickle(recos_fr_file)

recos_dict = {'fr': recos_fr}


app = Flask(__name__)

@app.route('/<langue>/<program_id>/<top>')
def get_reco(langue, program_id, top=10):
    t_all = t()
    top = int(top)

    if program_id not in list(recos_dict[langue].keys()):
        abort(404, description="programId not found")

    else:
        recos_list = recos_dict[langue][program_id][:top]
        json_output = {
            "data": [
                {"programId": programId}
                for programId in recos_list
            ]
        }
        print("temps exec total: ", (t() - t_all))
        print('---------------------------------')

        return jsonify(json_output)

if __name__ == '__main__':
    serve(app, host='X.X.X.X', port=XX, threads=8)

我已经检查了一些文档，我的问题是应该在什么级别上完成？谢谢。

Answer 1

我们可以使用以下方式添加缓存控制：

@app.after_request 
def apply_caching(response):
    response.headers['Cache-Control'] = 'public, max-age=3600,stale-while-revalidate=600, stale-if-error=259200'
    return response

在 Flask 中为我的 python 应用程序添加缓存控制

Add cache-control to my python app in Flask

python

flask