initial commit

src/submissions/frontend/custom_widgets/__init__.py

@@ -0,0 +1,298 @@
+from PyQt6.QtWidgets import (
+    QLabel, QVBoxLayout,
+    QLineEdit, QComboBox, QDialog, 
+    QDialogButtonBox, QDateEdit, QTableView,
+    QTextEdit, QSizePolicy, QWidget,
+    QGridLayout, QPushButton, QSpinBox,
+    QScrollBar
+)
+from PyQt6.QtCore import Qt, QDate, QAbstractTableModel
+from PyQt6.QtGui import QFontMetrics
+
+from backend.db import get_all_reagenttype_names, submissions_to_df, lookup_submission_by_id, lookup_all_sample_types, create_kit_from_yaml
+from jinja2 import Environment, FileSystemLoader
+
+import sys
+from pathlib import Path
+
+if getattr(sys, 'frozen', False):
+    loader_path = Path(sys._MEIPASS).joinpath("files", "templates")
+else:
+    loader_path = Path(__file__).parents[2].joinpath('templates').absolute().__str__()
+loader = FileSystemLoader(loader_path)
+env = Environment(loader=loader)
+
+class AddReagentQuestion(QDialog):
+    def __init__(self, reagent_type:str, reagent_lot:str):
+        super().__init__()
+
+        self.setWindowTitle(f"Add {reagent_lot}?")
+
+        QBtn = QDialogButtonBox.StandardButton.Yes | QDialogButtonBox.StandardButton.No
+
+        self.buttonBox = QDialogButtonBox(QBtn)
+        self.buttonBox.accepted.connect(self.accept)
+        self.buttonBox.rejected.connect(self.reject)
+
+        self.layout = QVBoxLayout()
+        message = QLabel(f"Couldn't find reagent type {reagent_type.replace('_', ' ').title()}: {reagent_lot} in the database.\nWould you like to add it?")
+        self.layout.addWidget(message)
+        self.layout.addWidget(self.buttonBox)
+        self.setLayout(self.layout)
+
+
+class AddReagentForm(QDialog):
+    def __init__(self, ctx:dict, reagent_lot:str|None, reagent_type:str|None):
+        super().__init__()
+
+        if reagent_lot == None:
+            reagent_lot = ""
+
+        self.setWindowTitle("Add Reagent")
+
+        QBtn = QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel
+
+        self.buttonBox = QDialogButtonBox(QBtn)
+        self.buttonBox.accepted.connect(self.accept)
+        self.buttonBox.rejected.connect(self.reject)
+
+        lot_input = QLineEdit()
+        lot_input.setText(reagent_lot)
+        exp_input = QDateEdit(calendarPopup=True)
+        exp_input.setDate(QDate.currentDate())
+        type_input = QComboBox()
+        type_input.addItems([item.replace("_", " ").title() for item in get_all_reagenttype_names(ctx=ctx)])
+        print(f"Trying to find index of {reagent_type}")
+        try:
+            reagent_type = reagent_type.replace("_", " ").title()
+        except AttributeError:
+            reagent_type = None
+        index = type_input.findText(reagent_type, Qt.MatchFlag.MatchEndsWith)
+        
+        if index >= 0:
+            type_input.setCurrentIndex(index)
+        self.layout = QVBoxLayout()
+        self.layout.addWidget(QLabel("Lot"))
+        self.layout.addWidget(lot_input)
+        self.layout.addWidget(QLabel("Expiry"))
+        self.layout.addWidget(exp_input)
+        self.layout.addWidget(QLabel("Type"))
+        self.layout.addWidget(type_input)
+        self.layout.addWidget(self.buttonBox)
+        self.setLayout(self.layout)
+
+
+
+class pandasModel(QAbstractTableModel):
+
+    def __init__(self, data):
+        QAbstractTableModel.__init__(self)
+        self._data = data
+
+    def rowCount(self, parent=None):
+        return self._data.shape[0]
+
+    def columnCount(self, parnet=None):
+        return self._data.shape[1]
+
+    def data(self, index, role=Qt.ItemDataRole.DisplayRole):
+        if index.isValid():
+            if role == Qt.ItemDataRole.DisplayRole:
+                return str(self._data.iloc[index.row(), index.column()])
+        return None
+
+    def headerData(self, col, orientation, role):
+        if orientation == Qt.Orientation.Horizontal and role == Qt.ItemDataRole.DisplayRole:
+            return self._data.columns[col]
+        return None
+        
+
+class SubmissionsSheet(QTableView):
+    def __init__(self, ctx:dict):
+        super().__init__()
+        self.ctx = ctx
+        self.setData()
+        self.resizeColumnsToContents()
+        self.resizeRowsToContents()
+        # self.clicked.connect(self.test)
+        self.doubleClicked.connect(self.show_details)
+ 
+    def setData(self): 
+        # horHeaders = []
+        # for n, key in enumerate(sorted(self.data.keys())):
+        #     horHeaders.append(key)
+        #     for m, item in enumerate(self.data[key]):
+        #         newitem = QTableWidgetItem(item)
+        #         self.setItem(m, n, newitem)
+        # self.setHorizontalHeaderLabels(horHeaders)
+        self.data = submissions_to_df(ctx=self.ctx)
+        self.model = pandasModel(self.data)
+        self.setModel(self.model)
+        # self.resize(800,600)
+
+    def show_details(self, item):
+        index=(self.selectionModel().currentIndex())
+        # print(index)
+        value=index.sibling(index.row(),0).data()
+        dlg = SubmissionDetails(ctx=self.ctx, id=value)
+        if dlg.exec():
+            pass
+
+
+
+    
+class SubmissionDetails(QDialog):
+    def __init__(self, ctx:dict, id:int) -> None:
+        super().__init__()
+
+        self.setWindowTitle("Submission Details")
+        data = lookup_submission_by_id(ctx=ctx, id=id)
+        base_dict = data.to_dict()
+        base_dict['reagents'] = [item.to_sub_dict() for item in data.reagents]
+        template = env.get_template("submission_details.txt")
+        text = template.render(sub=base_dict)
+        txt_field = QTextEdit(self)
+        txt_field.setReadOnly(True)
+        txt_field.document().setPlainText(text)
+
+        font = txt_field.document().defaultFont()
+        fontMetrics = QFontMetrics(font)
+        textSize = fontMetrics.size(0, txt_field.toPlainText())
+
+        w = textSize.width() + 10
+        h = textSize.height() + 10
+        txt_field.setMinimumSize(w, h)
+        txt_field.setMaximumSize(w, h)
+        txt_field.resize(w, h)
+        # txt_field.setSizePolicy(QSizePolicy.Policy.Preferred, QSizePolicy.Policy.MinimumExpanding)
+        QBtn = QDialogButtonBox.StandardButton.Ok
+        # self.buttonBox = QDialogButtonBox(QBtn)
+        # self.buttonBox.accepted.connect(self.accept)
+        txt_field.setText(text)
+        self.layout = QVBoxLayout()
+        self.layout.addWidget(txt_field)
+        # self.layout.addStretch()
+
+
+class ReportDatePicker(QDialog):
+    def __init__(self) -> None:
+        super().__init__()
+
+        self.setWindowTitle("Select Report Date Range")
+
+        QBtn = QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel
+        self.buttonBox = QDialogButtonBox(QBtn)
+        self.buttonBox.accepted.connect(self.accept)
+        self.buttonBox.rejected.connect(self.reject)
+
+        start_date = QDateEdit(calendarPopup=True)
+        start_date.setDate(QDate.currentDate())
+        end_date = QDateEdit(calendarPopup=True)
+        end_date.setDate(QDate.currentDate())
+        self.layout = QVBoxLayout()
+        self.layout.addWidget(QLabel("Start Date"))
+        self.layout.addWidget(start_date)
+        self.layout.addWidget(QLabel("End Date"))
+        self.layout.addWidget(end_date)
+        self.layout.addWidget(self.buttonBox)
+        self.setLayout(self.layout)
+
+
+class KitAdder(QWidget):
+    def __init__(self, parent_ctx:dict):
+        super().__init__()
+        self.ctx = parent_ctx
+        self.grid = QGridLayout()
+        self.setLayout(self.grid)
+        self.submit_btn = QPushButton("Submit")
+        self.grid.addWidget(self.submit_btn,0,0,1,1)
+        self.grid.addWidget(QLabel("Password:"),1,0)
+        self.grid.addWidget(QLineEdit(),1,1)
+        self.grid.addWidget(QLabel("Kit Name:"),2,0)
+        self.grid.addWidget(QLineEdit(),2,1)
+        self.grid.addWidget(QLabel("Used For Sample Type:"),3,0)
+        used_for = QComboBox()
+        used_for.addItems(lookup_all_sample_types(ctx=parent_ctx))
+        used_for.setEditable(True)
+        self.grid.addWidget(used_for,3,1)
+        self.grid.addWidget(QLabel("Cost per run:"),4,0)
+        cost = QSpinBox()
+        cost.setMinimum(0)
+        cost.setMaximum(9999)
+        self.grid.addWidget(cost,4,1)
+        self.add_RT_btn = QPushButton("Add Reagent Type")
+        self.grid.addWidget(self.add_RT_btn)
+        self.add_RT_btn.clicked.connect(self.add_RT)
+        self.submit_btn.clicked.connect(self.submit)
+
+    def add_RT(self):
+        maxrow = self.grid.rowCount()
+        self.grid.addWidget(ReagentTypeForm(parent_ctx=self.ctx), maxrow + 1,0,1,2)
+
+
+    def submit(self):
+        labels, values, reagents = self.extract_form_info(self)
+        info = {item[0]:item[1] for item in zip(labels, values)}
+        print(info)
+        # info['reagenttypes'] = reagents
+        # del info['name']
+        # del info['extension_of_life_(months)']
+        yml_type = {}
+        yml_type['password'] = info['password']
+        used = info['used_for_sample_type'].replace(" ", "_").lower()
+        yml_type[used] = {}
+        yml_type[used]['kits'] = {}
+        yml_type[used]['kits'][info['kit_name']] = {}
+        yml_type[used]['kits'][info['kit_name']]['cost'] = info['cost_per_run']
+        yml_type[used]['kits'][info['kit_name']]['reagenttypes'] = reagents
+        print(yml_type)
+        create_kit_from_yaml(ctx=self.ctx, exp=yml_type)
+
+    def extract_form_info(self, object):
+        labels = []
+        values = []
+        reagents = {}
+        for item in object.findChildren(QWidget):
+            print(item.parentWidget())
+            # if not isinstance(item.parentWidget(), ReagentTypeForm):
+            match item:
+                case QLabel():
+                    labels.append(item.text().replace(" ", "_").strip(":").lower())
+                case QLineEdit():
+                    # ad hoc check to prevent double reporting of qdatedit under lineedit for some reason
+                    if not isinstance(prev_item, QDateEdit) and not isinstance(prev_item, QComboBox) and not isinstance(prev_item, QSpinBox) and not isinstance(prev_item, QScrollBar):
+                        print(f"Previous: {prev_item}")
+                        print(f"Item: {item}, {item.text()}")
+                        values.append(item.text())
+                case QComboBox():
+                    values.append(item.currentText())
+                case QDateEdit():
+                    values.append(item.date().toPyDate())
+                case QSpinBox():
+                    values.append(item.value())
+                case ReagentTypeForm():
+                    
+                    re_labels, re_values, _ = self.extract_form_info(item) 
+                    reagent = {item[0]:item[1] for item in zip(re_labels, re_values)}
+                    print(reagent)
+                    # reagent = {reagent['name:']:{'eol':reagent['extension_of_life_(months):']}}
+                    reagents[reagent['name']] = {'eol_ext':int(reagent['extension_of_life_(months)'])}
+            prev_item = item
+        return labels, values, reagents
+
+
+
+class ReagentTypeForm(QWidget):
+    def __init__(self, parent_ctx:dict) -> None:
+        super().__init__()
+        grid = QGridLayout()
+        self.setLayout(grid)
+        grid.addWidget(QLabel("Name:"),0,0)
+        reagent_getter = QComboBox()
+        reagent_getter.addItems(get_all_reagenttype_names(ctx=parent_ctx))
+        reagent_getter.setEditable(True)
+        grid.addWidget(reagent_getter,0,1)
+        grid.addWidget(QLabel("Extension of Life (months):"),0,2)
+        eol = QSpinBox()
+        eol.setMinimum(0)
+        grid.addWidget(eol, 0,3)