PoC with response caching

app.py

@@ -14,6 +14,7 @@
 
 from admin.admin import admin_bp, set_theme_loader
 from api import api_bp
+from cache_config import cache
 from datarequest.datarequest import datarequest_bp
 from deposit.deposit import deposit_bp
 from fileviewer.fileviewer import fileviewer_bp
@@ -128,6 +129,9 @@ def load_admin_setting() -> Dict[str, Any]:
 # Start Flask-Session
 Session(app)
 
+# Initialize the cache.
+cache.init_app(app)
+
 # Start monitoring thread for extracting tech support information
 # Monitor signal file can be set to empty to completely disable monitor thread
 monitor_enabled: bool = app.config.get("MONITOR_SIGNAL_FILE", "/var/www/yoda/show-tech.sig") != ""
@@ -192,6 +196,10 @@ def static_loader() -> Optional[Response]:
         return None
 
 
+def authenticated() -> bool:
+    return g.get('user') is not None and g.get('irods') is not None
+
+
 @app.before_request
 def protect_pages() -> Optional[Response]:
     """Restricted pages access protection."""

cache_config.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+
+__copyright__ = 'Copyright (c) 2024, Utrecht University'
+__license__   = 'GPLv3, see LICENSE'
+
+from flask import g, session
+from flask_caching import Cache
+
+config = {
+    'CACHE_TYPE': 'FileSystemCache',
+    'CACHE_DIR': '/tmp',
+    'CACHE_DEFAULT_TIMEOUT': 300
+}
+
+# Create a Cache instance.
+cache = Cache(config=config)
+
+
+def authenticated() -> bool:
+    return g.get('user') is not None and g.get('irods') is not None
+
+
+def make_key() -> str:
+    if authenticated():
+        return session.sid
+    else:
+        return "unauthenticated"

general/general.py

@@ -6,13 +6,16 @@
 from flask import Blueprint, redirect, render_template, Response, url_for
 from flask_wtf.csrf import CSRFError
 
+from cache_config import cache, make_key
+
 general_bp = Blueprint('general_bp', __name__,
                        template_folder='templates/general',
                        static_folder='static/general',
                        static_url_path='/assets')
 
 
 @general_bp.route('/')
+@cache.cached(timeout=60, make_cache_key=make_key)
 def index() -> Response:
     return render_template('index.html')
 

group_manager/group_manager.py

@@ -6,6 +6,7 @@
 from flask import Blueprint, make_response, render_template, request, Response
 
 import api
+from cache_config import cache, make_key
 
 group_manager_bp = Blueprint('group_manager_bp', __name__,
                              template_folder='templates',
@@ -14,6 +15,7 @@
 
 
 @group_manager_bp.route('/')
+@cache.cached(timeout=60, make_cache_key=make_key)
 def index() -> Response:
     response = api.call('group_data', data={})
     group_hierarchy = response['data']['group_hierarchy']

requirements.txt

@@ -2,6 +2,7 @@ python-irodsclient==2.2.0
 Flask==3.0.2
 flask-session==0.6.0
 Flask-WTF==1.2.1
+Flask-Caching==2.3.0
 mod-wsgi==5.0.0
 pyjwt[crypto]==2.8.0
 requests==2.32.0

setup.cfg

@@ -4,7 +4,7 @@ import-order-style=smarkets
 strictness=short
 docstring_style=sphinx
 max-line-length=127
-application-import-names=admin,api,connman,errors,fileviewer,general,group_manager,monitor,research,search,open_search,stats,user,vault,deposit,intake,datarequest,util
+application-import-names=admin,api,cache_config,connman,errors,fileviewer,general,group_manager,monitor,research,search,open_search,stats,user,vault,deposit,intake,datarequest,util
 exclude=venv
 
 [mypy]