Update of Reagent usage ui

src/submissions/frontend/widgets/equipment_usage_2.py

@@ -0,0 +1,129 @@
+'''
+Creates forms that the user can enter equipment info into.
+'''
+from pprint import pformat
+from PyQt6.QtCore import Qt, QSignalBlocker, pyqtSlot
+from PyQt6.QtWebChannel import QWebChannel
+from PyQt6.QtWebEngineWidgets import QWebEngineView
+from PyQt6.QtWidgets import (
+    QDialog, QComboBox, QCheckBox, QLabel, QWidget, QVBoxLayout, QDialogButtonBox, QGridLayout
+)
+from backend.db.models import Equipment, Run, Process, Procedure, Tips
+from backend.validators.pydant import PydEquipment, PydEquipmentRole, PydTips, PydProcedure
+import logging
+from typing import Generator
+
+from tools import get_application_from_parent, render_details_template, flatten_list
+
+logger = logging.getLogger(f"submissions.{__name__}")
+
+
+class EquipmentUsage(QDialog):
+
+    def __init__(self, parent, procedure: PydProcedure):
+        super().__init__(parent)
+        self.procedure = procedure
+        self.setWindowTitle(f"Equipment Checklist - {procedure.name}")
+        self.used_equipment = self.procedure.equipment
+        self.kit = self.procedure.kittype
+        self.opt_equipment = procedure.proceduretype.get_equipment()
+        self.layout = QVBoxLayout()
+        self.app = get_application_from_parent(parent)
+        self.webview = QWebEngineView(parent=self)
+        self.webview.setContextMenuPolicy(Qt.ContextMenuPolicy.NoContextMenu)
+        self.webview.setMinimumSize(1200, 800)
+        self.webview.setMaximumWidth(1200)
+        # NOTE: Decide if exporting should be allowed.
+        # self.webview.loadFinished.connect(self.activate_export)
+        self.layout = QGridLayout()
+        # NOTE: button to export a pdf version
+        self.layout.addWidget(self.webview, 1, 0, 10, 10)
+        self.setLayout(self.layout)
+        self.setFixedWidth(self.webview.width() + 20)
+        # NOTE: setup channel
+        self.channel = QWebChannel()
+        self.channel.registerObject('backend', self)
+        html = self.construct_html(procedure=procedure)
+        self.webview.setHtml(html)
+        self.webview.page().setWebChannel(self.channel)
+        QBtn = QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel
+        self.buttonBox = QDialogButtonBox(QBtn)
+        self.buttonBox.accepted.connect(self.accept)
+        self.buttonBox.rejected.connect(self.reject)
+        self.layout.addWidget(self.buttonBox, 11, 1, 1, 1)
+
+    @classmethod
+    def construct_html(cls, procedure: PydProcedure, child: bool = False):
+        proceduretype = procedure.proceduretype
+        proceduretype_dict = proceduretype.details_dict()
+        run = procedure.run
+        proceduretype_dict['equipment_json'] = flatten_list([item['equipment_json'] for item in proceduretype_dict['equipment']])
+        # proceduretype_dict['equipment_json'] = [
+        #     {'name': 'Liquid Handler', 'equipment': [
+        #         {'name': 'Other', 'asset_number': 'XXX', 'processes': [
+        #             {'name': 'Trust Me', 'tips': ['Blah']},
+        #             {'name': 'No Me', 'tips': ['Blah', 'Crane']}
+        #         ]
+        #          },
+        #         {'name': 'Biomek', 'asset_number': '5015530', 'processes': [
+        #             {'name': 'Sample Addition', 'tips': ['Axygen 20uL']
+        #              }
+        #         ]
+        #          }
+        #     ]
+        #      }
+        # ]
+        # if procedure.equipment:
+        #     for equipmentrole in proceduretype_dict['equipment']:
+        #         # NOTE: Check if procedure equipment is present and move to head of the list if so.
+        #         try:
+        #             relevant_procedure_item = next((equipment for equipment in procedure.equipment if
+        #                                             equipment.equipmentrole == equipmentrole['name']))
+        #         except StopIteration:
+        #             continue
+        #         item_in_er_list = next((equipment for equipment in equipmentrole['equipment'] if
+        #                                 equipment['name'] == relevant_procedure_item.name))
+        #         equipmentrole['equipment'].insert(0, equipmentrole['equipment'].pop(
+        #             equipmentrole['equipment'].index(item_in_er_list)))
+        html = render_details_template(
+            template_name="support/equipment_usage",
+            css_in=[],
+            js_in=[],
+            proceduretype=proceduretype_dict,
+            run=run.details_dict(),
+            procedure=procedure.__dict__,
+            child=child
+        )
+        return html
+
+    @pyqtSlot(str, str, str, str)
+    def update_equipment(self, equipmentrole: str, equipment: str, process: str, tips: str):
+        try:
+            equipment_of_interest = next(
+                (item for item in self.procedure.equipment if item.equipmentrole == equipmentrole))
+        except StopIteration:
+            equipment_of_interest = None
+        equipment = Equipment.query(name=equipment)
+        if equipment_of_interest:
+            eoi = self.procedure.equipment.pop(self.procedure.equipment.index(equipment_of_interest))
+        else:
+            eoi = equipment.to_pydantic(proceduretype=self.procedure.proceduretype)
+        eoi.name = equipment.name
+        eoi.asset_number = equipment.asset_number
+        eoi.nickname = equipment.nickname
+        process = next((prcss for prcss in equipment.process if prcss.name == process))
+        eoi.process = process.to_pydantic()
+        tips = next((tps for tps in equipment.tips if tps.name == tips))
+        eoi.tips = tips.to_pydantic()
+        self.procedure.equipment.append(eoi)
+        logger.debug(f"Updated equipment: {self.procedure.equipment}")
+
+    def save_procedure(self):
+        sql, _ = self.procedure.to_sql()
+        logger.debug(pformat(sql.__dict__))
+        # import pickle
+        # with open("sql.pickle", "wb") as f:
+        #     pickle.dump(sql, f)
+        # with open("pyd.pickle", "wb") as f:
+        #     pickle.dump(self.procedure, f)
+        sql.save()