Don't filter out codes when loaded from process

Author: edan-bainglass

src/aiidalab_qe/app/submission/model.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import ipywidgets as ipw
 import traitlets as tl
 from IPython.display import Javascript, display
 
@@ -64,6 +63,7 @@ def confirm(self):
         super().confirm()
         if not self.has_process:
             self._submit()
+        self.lock()
 
     def _update(self, specific=""):
         self.update_process_label()
@@ -150,7 +150,10 @@ def update_process_metadata(self):
             return
         self.process_label = self.process.label
         self.process_description = self.process.description
-        self.locked = True
+
+    def refresh_codes(self, filter_codes_for_user: bool = True):
+        for _, resource_model in self.get_models():
+            resource_model.refresh_codes(filter_codes_for_user=filter_codes_for_user)
 
     def get_model_state(self) -> dict:
         return {
@@ -207,20 +210,6 @@ def _submit(self):
             pk = process_node.pk
             display(Javascript(f"window.history.pushState(null, '', '?pk={pk}');"))
 
-    def _link_model(self, model: ResourceSettingsModel):
-        for dependency in model.dependencies:
-            dependency_parts = dependency.split(".")
-            if len(dependency_parts) == 1:  # from parent, e.g. input_structure
-                target_model = self
-                trait = dependency
-            else:  # from sibling, e.g. workchain.protocol
-                sibling, trait = dependency_parts
-                target_model = self.get_model(sibling)
-            ipw.dlink(
-                (target_model, trait),
-                (model, trait),
-            )
-
     def _get_properties(self) -> list[str]:
         return self.input_parameters.get("workchain", {}).get("properties", [])
 

src/aiidalab_qe/app/wizard/model.py

@@ -45,6 +45,8 @@ def load_from_state(self, state: dict):
                         SubmissionStepModel,
                         self.get_model("submit"),
                     )
+                    if process_uuid is not None:
+                        submission_model.refresh_codes(filter_codes_for_user=False)
                     submission_model.set_model_state(resources_state)
 
                     if step_index >= 3:

src/aiidalab_qe/common/code/model.py

@@ -68,9 +68,15 @@ def activate(self):
     def deactivate(self):
         self.is_active = False
 
-    def update(self, user_email: str, default_code=None, refresh=False):
+    def update(
+        self,
+        user_email: str,
+        default_code: str | None = None,
+        filter_codes_for_user: bool = True,
+        refresh: bool = False,
+    ):
         if not self.options or refresh:
-            self.options = self._get_codes(user_email)
+            self.options = self._get_codes(user_email, filter_codes_for_user)
             if default_code:
                 try:
                     selected = orm.load_code(default_code).uuid
@@ -115,7 +121,11 @@ def _get_uuid(self, identifier):
         # in the app and thus will not be considered as an option!
         return uuid if uuid in [opt[1] for opt in self.options] else None
 
-    def _get_codes(self, user_email: str):
+    def _get_codes(
+        self,
+        user_email: str,
+        filter_codes_for_user: bool = True,
+    ) -> list[tuple[str, str]]:
         user = orm.User.collection.get(email=user_email)
 
         filters = (
@@ -124,25 +134,33 @@ def _get_codes(self, user_email: str):
             else {}
         )
 
-        codes = (
-            orm.QueryBuilder()
-            .append(
-                orm.Code,
-                filters=filters,
-            )
-            .all(flat=True)
+        codes = t.cast(
+            list[orm.Code],
+            (
+                orm.QueryBuilder()
+                .append(
+                    orm.Code,
+                    filters=filters,
+                )
+                .all(flat=True)
+            ),
         )
 
         return [
             (self._full_code_label(code), code.uuid)
             for code in codes
-            if code.computer.is_user_configured(user)
-            and (self.allow_hidden_codes or not code.is_hidden)
-            and (self.allow_disabled_computers or code.computer.is_user_enabled(user))
+            if not filter_codes_for_user
+            or (
+                code.computer.is_user_configured(user)
+                and (self.allow_hidden_codes or not code.is_hidden)
+                and (
+                    self.allow_disabled_computers or code.computer.is_user_enabled(user)
+                )
+            )
         ]
 
     @staticmethod
-    def _full_code_label(code):
+    def _full_code_label(code: orm.Code) -> str:
         return f"{code.label}@{code.computer.label}"
 
 

src/aiidalab_qe/common/panel.py

@@ -172,12 +172,13 @@ def update_active_codes(self):
             else:
                 code_model.deactivate()
 
-    def refresh_codes(self):
+    def refresh_codes(self, filter_codes_for_user: bool = True):
         for _, code_model in self.get_models():
             code_key = code_model.default_calc_job_plugin.split(".")[-1]
             code_model.update(
                 user_email=self.default_user_email,
                 default_code=self.default_codes.get(code_key, {}).get("code"),
+                filter_codes_for_user=filter_codes_for_user,
                 refresh=True,
             )