mirror of
https://github.com/codl/forget
synced 2024-12-14 01:14:34 +01:00
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from flask import url_for, abort
|
|
import os
|
|
|
|
|
|
def cachebust(app):
|
|
# pylint: disable=unused-variable
|
|
|
|
@app.route('/static-cb/<int:timestamp>/<path:filename>')
|
|
def static_cachebust(timestamp, filename):
|
|
path = os.path.join(app.static_folder, filename)
|
|
try:
|
|
mtime = os.stat(path).st_mtime
|
|
except Exception:
|
|
return abort(404)
|
|
if abs(mtime - timestamp) > 1:
|
|
abort(404)
|
|
else:
|
|
resp = app.view_functions['static'](filename=filename)
|
|
resp.headers.set(
|
|
'cache-control',
|
|
'public, immutable, max-age={}'.format(60*60*24*365))
|
|
if 'expires' in resp.headers:
|
|
resp.headers.remove('expires')
|
|
return resp
|
|
|
|
@app.context_processor
|
|
def replace_url_for():
|
|
return dict(url_for=cachebust_url_for)
|
|
|
|
def cachebust_url_for(endpoint, **kwargs):
|
|
if endpoint == 'static':
|
|
endpoint = 'static_cachebust'
|
|
path = os.path.join(app.static_folder, kwargs.get('filename'))
|
|
kwargs['timestamp'] = int(os.stat(path).st_mtime)
|
|
return url_for(endpoint, **kwargs)
|
|
|
|
return cachebust_url_for
|