[FIXED] Add cache-control to my python app in Flask

Issue

I am doing a few tests about my website. My app was developped using flask, I want to add cach-control "max-age=3600" in my function below :

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) 

I already check some documentation, My question is at what level should this be done? Thank you.

Solution

We can add cach control using :

@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

Answered By - MFatn