From b87fe728012f0c75653cf8af2865dcc87b95f5d4 Mon Sep 17 00:00:00 2001
From: UsualSpec <98665326+UsualSpec@users.noreply.github.com>
Date: Mon, 1 Jul 2024 19:04:13 +0200
Subject: [PATCH] Add linked measurements construct

---
 server/management/app.py                      | 21 ++++++++++++++++---
 .../management/templates/runManagement.html   |  9 ++++++++
 2 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/server/management/app.py b/server/management/app.py
index 73fc311..ade72b0 100644
--- a/server/management/app.py
+++ b/server/management/app.py
@@ -369,6 +369,13 @@ def manageRun(runId):
         selectedClients = []
         for client in clnts:
             selectedClients.append(client["client_uid"])
+        
+        pgcurs.execute("SELECT shared_measurement_id, server_measurement_id FROM measurements WHERE run_id = %(runid)s", {'runid': runId})
+        pgConnection.commit()
+        linkedMmts = pgcurs.fetchall()
+
+        # only show delete if no measurements are linked
+        showDelete = not linkedMmts
 
         # Manage timing
 
@@ -384,7 +391,7 @@ def manageRun(runId):
             stopTime = convertToLocalTime(run["stop_run"]).strftime('%H:%M:%S.%f')[:-3]
             stopDate = convertToLocalTime(run["stop_run"]).strftime('%Y-%m-%d')
 
-        return render_template("runManagement.html", runId=runId, run=run, duts=duts, clientIds=clientIds, selectedClients=selectedClients, serverTz=get_localzone(), startTime=startTime, startDate=startDate, stopTime=stopTime, stopDate=stopDate, showDelete=True)
+        return render_template("runManagement.html", runId=runId, run=run, duts=duts, clientIds=clientIds, selectedClients=selectedClients, serverTz=get_localzone(), startTime=startTime, startDate=startDate, stopTime=stopTime, stopDate=stopDate, showDelete=showDelete, linkedMmts=linkedMmts)
 
 
 @app.route("/runs")
@@ -437,12 +444,20 @@ def deleteRun(runId):
         if not pgcurs.fetchone():
             return "This run doesn't exist", 404
         try:
+            pgcurs.execute("SELECT shared_measurement_id FROM measurements WHERE run_id = %(runid)s", {'runid': runId})
+            pgConnection.commit()
+            linkedMmts = []
+            for mmId in pgcurs:
+                linkedMmts.append(mmId[0])
+            if linkedMmts != []:
+                return "Cannot delete run which is still linked to a measurement. Please delete or unlink the measurement(s) " + str(linkedMmts) + " before deleting the run.", 403
+
             pgcurs.execute("DELETE FROM client_runs WHERE run_id = %(runid)s", {'runid': runId})
             pgcurs.execute("DELETE FROM runs WHERE run_id = %(runid)s", {'runid': runId})
             pgConnection.commit()
-        except ForeignKeyViolation:
+        except ForeignKeyViolation as fke:
             pgConnection.rollback()
-            return "Cannot delete run which is still linked to a measurement. Please delete the measurement before deleting the run.", 403
+            return "Cannot delete run which is still linked to a measurement. Please delete or unlink the measurement(s) before deleting the run. Error: " + str(fke), 403
         return str(runId)
 
 
diff --git a/server/management/templates/runManagement.html b/server/management/templates/runManagement.html
index 52c1d27..a7331ea 100644
--- a/server/management/templates/runManagement.html
+++ b/server/management/templates/runManagement.html
@@ -5,6 +5,15 @@
 <button id="deleteBtn" class="btn btn-danger mb-2" data-bs-toggle="modal" data-bs-target="#confirmDeletionModal">Delete run</button>
 {% endif %}
 {% block runheading %}<h2>{{ runId | e}}</h2>{% endblock %}
+{% if linkedMmts %}
+<div>Linked with
+  <ul>
+    {% for mmt in linkedMmts %}
+    <li><a href="{{ url_for('manageMeasurement', measurementId=mmt.server_measurement_id) }}">{{ mmt.shared_measurement_id | e }}</a></li>
+  {% endfor %}
+  </ul>
+</div>
+{% endif %}
 <div class="alert p-2" id="runStatus">Enter RUN information</div>
 <form id="runForm" method="POST" {% block formaction %}action="{{ url_for('updateRun', runId=runId) }}"{% endblock %} class="bg-light border p-2">
     <label for="dut">Device under test</label>