From 733863fadca690f631d66ad93b16a174360ab40e Mon Sep 17 00:00:00 2001
From: Matthias Saimpert <matthias.saimpert@cern.ch>
Date: Thu, 16 Oct 2025 09:01:24 +0000
Subject: [PATCH] Automate the ticking of the allow transition stage option
 during sign-off of unmasking stage

(cherry picked from commit b4f7ea0700ff29f0779a17e77c10f614df80d05d)

94ee9e1f attempt to automate the ticking of the allow transition stage option in sign...
29369bd7 fix precommit
5742b4ec logic fix
801bb18a some fixes
470ca597 fix and improve logic

Co-authored-by: Matthias Saimpert <matthias.saimpert@cern.ch>
---
 viewer/pages/qc.py                    | 16 ++++++++++++++--
 viewer/templates/select_QC_table.html |  1 +
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/viewer/pages/qc.py b/viewer/pages/qc.py
index d3ce8baa0..bfa88e55f 100644
--- a/viewer/pages/qc.py
+++ b/viewer/pages/qc.py
@@ -2556,6 +2556,8 @@ def select_test():
     page_docs["page"] = "select_test"
     page_docs["title"] = "Sign off QC test in this stage"
 
+    page_docs["enable_transition"] = False
+
     if not session.get("logged_in", False):
         return render_template("401.html")
 
@@ -2598,6 +2600,18 @@ def select_test():
         )
     )
 
+    # Allow transition to OBWBP by default if OBWBP stage is enabled in QC customize menu
+    stageInfo = userdb.QC.stages.find_one({"code": componentType})
+
+    if (
+        page_docs["stage"] == "MODULE/PARYLENE_UNMASKING"
+        and stageInfo.get("disabled_tests", {})
+        .get("MODULE/WIREBOND_PROTECTION", {})
+        .get("disabled", 1)
+        == 0
+    ):
+        page_docs["enable_transition"] = True
+
     # Augment test results
     for test in page_docs["tests"]:
         test["resultHTML"] = json2html.convert(test["results"])
@@ -2700,8 +2714,6 @@ def select_test():
             # endfor fieldCode
         # endfor testType
 
-        stageInfo = userdb.QC.stages.find_one({"code": componentType})
-
         # Check if current stage is a special stage -  if so, don't increment stage
         if current_stage in special_stages:
             next_stage = current_stage
diff --git a/viewer/templates/select_QC_table.html b/viewer/templates/select_QC_table.html
index 928859bce..e512e5a48 100644
--- a/viewer/templates/select_QC_table.html
+++ b/viewer/templates/select_QC_table.html
@@ -119,6 +119,7 @@
                 value="enable"
                 id="enableTransitionToAlternative"
                 style="width: 1rem; height: 1rem;"
+                {% if enable_transition %}checked{% endif %}
               />
               <label class="form-check-label h5" for="enableTransitionToAlternative">Allow Stage Transition to Alternative Stages</label>
             </div>
-- 
GitLab