initial commit
This commit is contained in:
Landon Wark
298
src/submissions/frontend/custom_widgets/__init__.py
Normal file
298
src/submissions/frontend/custom_widgets/__init__.py
Normal file
@@ -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)
|
||||
Reference in New Issue
Block a user