Preparation for code cleanup

src/submissions/frontend/functions.py

@@ -1,8 +1,19 @@
 # from ..models import *
 from backend.db.models import *
+from backend.db import lookup_regent_by_type_name
+from tools import check_not_nan
 # from backend.db import lookup_kittype_by_name
 import logging
 import numpy as np
+from backend.excel.parser import SheetParser
+from PyQt6.QtWidgets import (
+    QMainWindow, QLabel, QToolBar, 
+    QTabWidget, QWidget, QVBoxLayout,
+    QPushButton, QFileDialog,
+    QLineEdit, QMessageBox, QComboBox, QDateEdit, QHBoxLayout,
+    QSpinBox, QScrollArea
+)
+
 
 logger = logging.getLogger(f"submissions.{__name__}")
 
@@ -28,6 +39,38 @@ def check_kit_integrity(sub:BasicSubmission|KitType, reagenttypes:list|None=None
     return result
 
 
-
-    
+def insert_reagent_import(ctx:dict, item:str, prsr:SheetParser|None=None) -> QComboBox:
+    add_widget = QComboBox()
+    add_widget.setEditable(True)
+    # Ensure that all reagenttypes have a name that matches the items in the excel parser
+    query_var = item.replace("lot_", "")
+    logger.debug(f"Query for: {query_var}")
+    if prsr != None:
+        if isinstance(prsr.sub[item], np.float64):
+            logger.debug(f"{prsr.sub[item]['lot']} is a numpy float!")
+            try:
+                prsr.sub[item] = int(prsr.sub[item]['lot'])
+            except ValueError:
+                pass
+    # query for reagents using type name from sheet and kit from sheet
+    logger.debug(f"Attempting lookup of reagents by type: {query_var}")
+    # below was lookup_reagent_by_type_name_and_kit_name, but I couldn't get it to work.
+    relevant_reagents = [item.__str__() for item in lookup_regent_by_type_name(ctx=ctx, type_name=query_var)]#, kit_name=prsr.sub['extraction_kit'])]
+    output_reg = []
+    for reagent in relevant_reagents:
+        if isinstance(reagent, set):
+            for thing in reagent:
+                output_reg.append(thing)
+        elif isinstance(reagent, str):
+            output_reg.append(reagent)
+    relevant_reagents = output_reg
+    # if reagent in sheet is not found insert it into items
+    if prsr != None:
+        logger.debug(f"Relevant reagents for {prsr.sub[item]}: {relevant_reagents}")
+        if str(prsr.sub[item]['lot']) not in relevant_reagents and prsr.sub[item]['lot'] != 'nan':
+            if check_not_nan(prsr.sub[item]['lot']):
+                relevant_reagents.insert(0, str(prsr.sub[item]['lot']))
+    logger.debug(f"New relevant reagents: {relevant_reagents}")
+    add_widget.addItems(relevant_reagents)
+    return add_widget