ensure runtime template

Author: JinZhou5042

app.py

@@ -35,6 +35,8 @@ def setup_request_logging(app, runtime_state):
     @app.before_request
     def log_request_info():
         runtime_state.template_lock.renew()
+        request_folder = request.args.get('folder')
+        runtime_state.ensure_runtime_template(request_folder)
         runtime_state.log_request(request)
         request._start_time = time.time()
 

routes/runtime_state.py

@@ -143,12 +143,10 @@ def log_warning(self, message):
         self.logger.warning(f"{self.log_prefix} {message}")
 
     def log_request(self, request):
-        self.logger.info(
-            f"{self.log_prefix} {build_request_info_string(request)}")
+        self.logger.info(f"{self.log_prefix} {build_request_info_string(request)}")
 
     def log_response(self, response, request, duration=None):
-        self.logger.info(
-            f"{self.log_prefix} {build_response_info_string(response, request, duration)}")
+        self.logger.info(f"{self.log_prefix} {build_response_info_string(response, request, duration)}")
 
     def reload_data_if_needed(self):
         if not self.data_parser:
@@ -170,6 +168,17 @@ def _get_current_pkl_files_fingerprint(self):
 
         return get_files_fingerprint(self.data_parser.pkl_files)
 
+    def ensure_runtime_template(self, runtime_template):
+        if not runtime_template:
+            return
+
+        if self.template_lock.is_locked():
+            return
+
+        if runtime_template == os.path.basename(self.runtime_template):
+            return
+
+        self.reload_template(runtime_template)
 
     def reload_template(self, runtime_template):
         # init template and data parser

routes/utils.py

@@ -229,4 +229,4 @@ def get_files_fingerprint(files):
             continue
         parts.append(f"{file}:{stat['mtime']}:{stat['size']}")
 
-    return hashlib.md5(";".join(parts).encode()).hexdigest()
+    return hashlib.md5(";".join(parts).encode()).hexdigest()

static/js/modules/base.js

@@ -142,10 +142,13 @@ export class BaseModule {
         this.svg = null;
     }
 
-    async fetchData() {
+    async fetchData(folder) {
         this.clearSVG();
 
-        const response = await fetch(this.api_url);
+        const response = await fetch(
+            `${this.api_url}?` +
+            `folder=${folder}`
+        );
         const data = await response.json();
 
         if (!data) {

static/js/modules/task_subgraphs.js

@@ -10,7 +10,9 @@ export class TaskSubgraphsModule extends BaseModule {
         if (!subgraph_id) return;
 
         const response = await fetch(
-            `${this.api_url}?subgraph_id=${subgraph_id}` +
+            `${this.api_url}?` +
+            `folder=${folder}` +
+            `&subgraph_id=${subgraph_id}` +
             `&plot_failed_task=${plot_failed_task}` +
             `&plot_recovery_task=${plot_recovery_task}`
         );