Addition of new reagent lots updated.

This commit is contained in:
2025-10-03 10:19:17 -05:00
parent 1445d2b93b
commit ccee4b3afe
8 changed files with 112 additions and 114 deletions

View File

@@ -520,6 +520,7 @@ class BaseClass(Base):
if isinstance(field_type, InstrumentedAttribute): if isinstance(field_type, InstrumentedAttribute):
match field_type.property: match field_type.property:
case ColumnProperty(): case ColumnProperty():
return super().__setattr__(key, value) return super().__setattr__(key, value)
case _RelationshipDeclared(): case _RelationshipDeclared():
if field_type.property.uselist: if field_type.property.uselist:

View File

@@ -346,6 +346,7 @@ class Reagent(BaseClass, LogMixin):
return [dict(name=self.name, lot=lot.lot, expiry=lot.expiry + self.eol_ext) for lot in self.reagentlot] return [dict(name=self.name, lot=lot.lot, expiry=lot.expiry + self.eol_ext) for lot in self.reagentlot]
class ReagentLot(BaseClass): class ReagentLot(BaseClass):
pyd_model_name = "Reagent" pyd_model_name = "Reagent"
@@ -445,6 +446,7 @@ class ReagentLot(BaseClass):
output['reagent'] = output['reagent'].name output['reagent'] = output['reagent'].name
return output return output
class Discount(BaseClass): class Discount(BaseClass):
""" """
Relationship table for client labs for certain kits. Relationship table for client labs for certain kits.

View File

@@ -43,7 +43,7 @@ class ClientSubmission(BaseClass, LogMixin):
submission_category = Column(String(64)) #: i.e. Surveillance submission_category = Column(String(64)) #: i.e. Surveillance
sample_count = Column(INTEGER) #: Number of sample in the procedure sample_count = Column(INTEGER) #: Number of sample in the procedure
full_batch_size = Column(INTEGER) #: Number of wells in provided plate. 0 if no plate. full_batch_size = Column(INTEGER) #: Number of wells in provided plate. 0 if no plate.
comment = Column(JSON) #: comment objects from users. comments = Column(JSON) #: comment objects from users.
run = relationship("Run", back_populates="clientsubmission") #: many-to-one relationship run = relationship("Run", back_populates="clientsubmission") #: many-to-one relationship
contact = relationship("Contact", back_populates="clientsubmission") #: contact representing submitting lab. contact = relationship("Contact", back_populates="clientsubmission") #: contact representing submitting lab.
contact_id = Column(INTEGER, ForeignKey("_contact.id", ondelete="SET NULL", contact_id = Column(INTEGER, ForeignKey("_contact.id", ondelete="SET NULL",
@@ -240,9 +240,9 @@ class ClientSubmission(BaseClass, LogMixin):
custom = None custom = None
runs = None runs = None
try: try:
comments = self.comment comments = self.comments
except Exception as e: except Exception as e:
logger.error(f"Error setting comment: {self.comment}, {e}") logger.error(f"Error setting comment: {self.comments}, {e}")
comments = None comments = None
try: try:
contact = self.contact.name contact = self.contact.name

View File

@@ -3,6 +3,8 @@ Module for clientsubmission parsing
""" """
from __future__ import annotations from __future__ import annotations
import logging import logging
import sys
from datetime import datetime
from pathlib import Path from pathlib import Path
from string import ascii_lowercase from string import ascii_lowercase
from typing import Generator, TYPE_CHECKING from typing import Generator, TYPE_CHECKING
@@ -135,6 +137,9 @@ class ClientSubmissionInfoParser(DefaultKEYVALUEParser, SubmissionTyperMixin):
output['submissiontype']['value'] = self.submissiontype.name.title() output['submissiontype']['value'] = self.submissiontype.name.title()
except KeyError: except KeyError:
pass pass
if isinstance(output['submitted_date']['value'], datetime):
output['submitted_date']['value'] = output['submitted_date']['value'].date()
return output return output

View File

@@ -37,6 +37,8 @@ class PydBaseClass(BaseModel, extra='allow', validate_assignment=True):
def prevalidate(cls, data): def prevalidate(cls, data):
sql_fields = [k for k, v in cls._sql_object.__dict__.items() if isinstance(v, InstrumentedAttribute)] sql_fields = [k for k, v in cls._sql_object.__dict__.items() if isinstance(v, InstrumentedAttribute)]
output = {} output = {}
match data:
case dict():
try: try:
items = data.items() items = data.items()
except AttributeError as e: except AttributeError as e:
@@ -48,6 +50,8 @@ class PydBaseClass(BaseModel, extra='allow', validate_assignment=True):
output[new_key] = value output[new_key] = value
else: else:
output[key] = value output[key] = value
case _:
output = data
return output return output
@model_validator(mode='after') @model_validator(mode='after')
@@ -135,6 +139,7 @@ class PydBaseClass(BaseModel, extra='allow', validate_assignment=True):
continue continue
return list(set(output)) return list(set(output))
class PydResults(PydBaseClass, arbitrary_types_allowed=True): class PydResults(PydBaseClass, arbitrary_types_allowed=True):
result: dict = Field(default={}) result: dict = Field(default={})
result_type: str = Field(default="NA") result_type: str = Field(default="NA")
@@ -186,9 +191,9 @@ class PydReagentLot(PydBaseClass):
class PydReagent(PydBaseClass): class PydReagent(PydBaseClass):
lot: str | None # lot: str | None
reagentrole: str | None reagentrole: str | None
expiry: date | datetime | Literal['NA'] | None = Field(default=None, validate_default=True) # expiry: date | datetime | Literal['NA'] | None = Field(default=None, validate_default=True)
name: str | None = Field(default=None, validate_default=True) name: str | None = Field(default=None, validate_default=True)
missing: bool = Field(default=True) missing: bool = Field(default=True)
comment: str | None = Field(default="", validate_default=True) comment: str | None = Field(default="", validate_default=True)
@@ -219,47 +224,47 @@ class PydReagent(PydBaseClass):
return value return value
return value return value
@field_validator("lot", mode='before') # @field_validator("lot", mode='before')
@classmethod # @classmethod
def rescue_lot_string(cls, value): # def rescue_lot_string(cls, value):
if value is not None: # if value is not None:
return convert_nans_to_nones(str(value).strip()) # return convert_nans_to_nones(str(value).strip())
return value # return value
#
@field_validator("lot") # @field_validator("lot")
@classmethod # @classmethod
def enforce_lot_string(cls, value): # def enforce_lot_string(cls, value):
if value is not None: # if value is not None:
return value.upper().strip() # return value.upper().strip()
return value # return value
#
@field_validator("expiry", mode="before") # @field_validator("expiry", mode="before")
@classmethod # @classmethod
def enforce_date(cls, value): # def enforce_date(cls, value):
if value is not None: # if value is not None:
match value: # match value:
case int(): # case int():
return datetime.fromordinal(datetime(1900, 1, 1).toordinal() + value - 2) # return datetime.fromordinal(datetime(1900, 1, 1).toordinal() + value - 2)
case 'NA': # case 'NA':
return value # return value
case str(): # case str():
return parse(value) # return parse(value)
case date(): # case date():
return datetime.combine(value, datetime.max.time()) # return datetime.combine(value, datetime.max.time())
case datetime(): # case datetime():
return value # return value
case _: # case _:
return convert_nans_to_nones(str(value)) # return convert_nans_to_nones(str(value))
if value is None: # if value is None:
value = datetime.combine(date.today(), datetime.max.time()) # value = datetime.combine(date.today(), datetime.max.time())
return value # return value
#
@field_validator("expiry") # @field_validator("expiry")
@classmethod # @classmethod
def date_na(cls, value): # def date_na(cls, value):
if isinstance(value, date) and value.year == 1970: # if isinstance(value, date) and value.year == 1970:
value = "NA" # value = "NA"
return value # return value
@field_validator("name", mode="before") @field_validator("name", mode="before")
@classmethod @classmethod
@@ -269,7 +274,6 @@ class PydReagent(PydBaseClass):
else: else:
return values.data['reagentrole'].strip() return values.data['reagentrole'].strip()
def improved_dict(self) -> dict: def improved_dict(self) -> dict:
""" """
Constructs a dictionary consisting of model.fields and model.extras Constructs a dictionary consisting of model.fields and model.extras
@@ -292,14 +296,17 @@ class PydReagent(PydBaseClass):
Returns: Returns:
Tuple[Reagent, Report]: Reagent instance and result of function Tuple[Reagent, Report]: Reagent instance and result of function
""" """
from backend.db.models import ReagentLot, Reagent
report = Report() report = Report()
if self.model_extra is not None: if self.model_extra is not None:
self.__dict__.update(self.model_extra) self.__dict__.update(self.model_extra)
reagentlot, new = ReagentLot.query_or_create(lot=self.lot, name=self.name) reagentlot, new = ReagentLot.query_or_create(lot=self.lot, name=self.name)
if new: if new:
reagent = Reagent.query(name=self.name) reagent = Reagent.query(name=self.name, limit=1)
reagentlot.reagent = reagent reagentlot.reagent = reagent
reagentlot.expiry = self.expiry reagentlot.expiry = self.expiry
if isinstance(reagentlot.expiry, str):
reagentlot.expiry = datetime.combine(datetime.strptime(reagentlot.expiry, "%Y-%m-%d"), datetime.max.time())
return reagentlot, report return reagentlot, report
@@ -371,6 +378,7 @@ class PydTips(PydBaseClass):
Returns: Returns:
SubmissionTipsAssociation: Association between queried tips and procedure SubmissionTipsAssociation: Association between queried tips and procedure
""" """
from backend.db.models import TipsLot
report = Report() report = Report()
tips = TipsLot.query(lot=self.lot, limit=1) tips = TipsLot.query(lot=self.lot, limit=1)
return tips, report return tips, report
@@ -388,6 +396,7 @@ class PydEquipment(PydBaseClass):
@field_validator('equipmentrole', mode='before') @field_validator('equipmentrole', mode='before')
@classmethod @classmethod
def get_role_name(cls, value): def get_role_name(cls, value):
from backend.db.models import EquipmentRole
match value: match value:
case list(): case list():
value = value[0] value = value[0]
@@ -402,6 +411,7 @@ class PydEquipment(PydBaseClass):
@field_validator('processes', mode='before') @field_validator('processes', mode='before')
@classmethod @classmethod
def process_to_pydantic(cls, value, values): def process_to_pydantic(cls, value, values):
from backend.db.models import ProcessVersion, Process
if isinstance(value, GeneratorType): if isinstance(value, GeneratorType):
value = [item for item in value] value = [item for item in value]
value = convert_nans_to_nones(value) value = convert_nans_to_nones(value)
@@ -431,6 +441,7 @@ class PydEquipment(PydBaseClass):
@field_validator('tips', mode='before') @field_validator('tips', mode='before')
@classmethod @classmethod
def tips_to_pydantic(cls, value, values): def tips_to_pydantic(cls, value, values):
from backend.db.models import TipsLot
if isinstance(value, GeneratorType): if isinstance(value, GeneratorType):
value = [item for item in value] value = [item for item in value]
value = convert_nans_to_nones(value) value = convert_nans_to_nones(value)
@@ -467,6 +478,7 @@ class PydEquipment(PydBaseClass):
Returns: Returns:
Tuple[Equipment, RunEquipmentAssociation]: SQL objects Tuple[Equipment, RunEquipmentAssociation]: SQL objects
""" """
from backend.db.models import Equipment, ProcedureEquipmentAssociation, Process
report = Report() report = Report()
if isinstance(procedure, str): if isinstance(procedure, str):
procedure = Procedure.query(name=procedure) procedure = Procedure.query(name=procedure)
@@ -680,6 +692,7 @@ class PydProcess(PydBaseClass, extra="allow"):
@report_result @report_result
def to_sql(self): def to_sql(self):
from backend.db.models import ProcessVersion
report = Report() report = Report()
name = self.name.split("-")[0] name = self.name.split("-")[0]
# NOTE: can't use query_or_create due to name not being part of ProcessVersion # NOTE: can't use query_or_create due to name not being part of ProcessVersion
@@ -725,12 +738,12 @@ class PydElastic(BaseModel, extra="allow", arbitrary_types_allowed=True):
# NOTE: Generified objects below: # NOTE: Generified objects below:
class PydProcedure(PydBaseClass, arbitrary_types_allowed=True): class PydProcedure(PydBaseClass, arbitrary_types_allowed=True):
proceduretype: ProcedureType | None = Field(default=None) proceduretype: Any | None = Field(default=None)
run: Run | str | None = Field(default=None) run: Any | str | None = Field(default=None)
name: dict = Field(default=dict(value="NA", missing=True), validate_default=True) name: dict = Field(default=dict(value="NA", missing=True), validate_default=True)
technician: dict = Field(default=dict(value="NA", missing=True)) technician: dict = Field(default=dict(value="NA", missing=True))
repeat: bool = Field(default=False) repeat: bool = Field(default=False)
repeat_of: Procedure | None = Field(default=None) repeat_of: Any | None = Field(default=None)
plate_map: str | None = Field(default=None) plate_map: str | None = Field(default=None)
reagent: list | None = Field(default=[]) reagent: list | None = Field(default=[])
reagentrole: dict | None = Field(default={}, validate_default=True) reagentrole: dict | None = Field(default={}, validate_default=True)
@@ -919,7 +932,10 @@ class PydProcedure(PydBaseClass, arbitrary_types_allowed=True):
reg.save() reg.save()
def to_sql(self, new: bool = False): def to_sql(self, new: bool = False):
from backend.db.models import RunSampleAssociation, ProcedureSampleAssociation from backend.db.models import (
RunSampleAssociation, ProcedureSampleAssociation, Procedure, ProcedureReagentLotAssociation,
ProcedureEquipmentAssociation
)
logger.debug(f"incoming pyd: {pformat([item.__dict__ for item in self.equipment])}") logger.debug(f"incoming pyd: {pformat([item.__dict__ for item in self.equipment])}")
if new: if new:
sql = Procedure() sql = Procedure()
@@ -1042,9 +1058,11 @@ class PydClientSubmission(PydBaseClass):
def enforce_submitted_date(cls, value): def enforce_submitted_date(cls, value):
match value: match value:
case str(): case str():
value = dict(value=datetime.strptime(value, "%Y-%m-%d %H:%M:%S"), missing=False) value = dict(value=datetime.strptime(value, "%Y-%m-%d %H:%M:%S").date(), missing=False)
case date() | datetime(): case date():
value = dict(value=value, missing=False) value = dict(value=value, missing=False)
case datetime():
value = dict(value=value.date(), missing=False)
case _: case _:
pass pass
return value return value
@@ -1162,6 +1180,7 @@ class PydClientSubmission(PydBaseClass):
def to_sql(self): def to_sql(self):
sql = super().to_sql() sql = super().to_sql()
from backend.db.models import SubmissionType
assert not any([isinstance(item, PydSample) for item in sql.sample]) assert not any([isinstance(item, PydSample) for item in sql.sample])
sql.sample = [] sql.sample = []
if not sql.submissiontype: if not sql.submissiontype:
@@ -1662,44 +1681,3 @@ class PydRun(PydBaseClass): #, extra='allow'):
samples.append(sample) samples.append(sample)
samples = sorted(samples, key=itemgetter("submission_rank")) samples = sorted(samples, key=itemgetter("submission_rank"))
return samples return samples
# class PydResults(PydBaseClass, arbitrary_types_allowed=True):
# result: dict = Field(default={})
# result_type: str = Field(default="NA")
# img: None | bytes = Field(default=None)
# parent: Procedure | ProcedureSampleAssociation | None = Field(default=None)
# date_analyzed: datetime | None = Field(default=None)
#
# @field_validator("date_analyzed")
# @classmethod
# def set_today(cls, value):
# match value:
# case str():
# value = datetime.strptime(value, "%Y-%m-%d %H:%M:%S")
# case datetime():
# pass
# case date():
# value = datetime.combine(value, datetime.max.time())
# case _:
# value = datetime.now()
# return value
#
# def to_sql(self):
# sql, _ = Results.query_or_create(result_type=self.result_type, result=self.results)
# try:
# check = sql.image
# except FileNotFoundError:
# check = False
# if not check:
# sql.image = self.img
# if not sql.date_analyzed:
# sql.date_analyzed = self.date_analyzed
# match self.parent:
# case ProcedureSampleAssociation():
# sql.sampleprocedureassociation = self.parent
# case Procedure():
# sql.procedure = self.parent
# case _:
# logger.error("Improper association found.")
# return sql

View File

@@ -80,7 +80,6 @@ class ProcedureCreation(QDialog):
proceduretype_dict['equipment'] = [sanitize_object_for_json(object) for object in proceduretype_dict['equipment']] proceduretype_dict['equipment'] = [sanitize_object_for_json(object) for object in proceduretype_dict['equipment']]
regex = re.compile(r".*R\d$") regex = re.compile(r".*R\d$")
proceduretype_dict['previous'] = [""] + [item.name for item in self.run.procedure if item.proceduretype == self.proceduretype and not bool(regex.match(item.name))] proceduretype_dict['previous'] = [""] + [item.name for item in self.run.procedure if item.proceduretype == self.proceduretype and not bool(regex.match(item.name))]
# sys.exit(f"ProcedureDict:\n{pformat(proceduretype_dict)}")
html = render_details_template( html = render_details_template(
template_name="procedure_creation", template_name="procedure_creation",
js_in=["procedure_form", "grid_drag", "context_menu"], js_in=["procedure_form", "grid_drag", "context_menu"],
@@ -90,12 +89,13 @@ class ProcedureCreation(QDialog):
plate_map=self.plate_map, plate_map=self.plate_map,
edit=self.edit edit=self.edit
) )
# with open("procedure_creation.html", "w") as f:
# f.write(html)
self.webview.setHtml(html) self.webview.setHtml(html)
@pyqtSlot(str, str, str, str) @pyqtSlot(str, str, str, str)
def update_equipment(self, equipmentrole: str, equipment: str, processversion: str, tips: str): def update_equipment(self, equipmentrole: str, equipment: str, processversion: str, tips: str):
from backend.db.models import Equipment, ProcessVersion, TipsLot from backend.db.models import Equipment, ProcessVersion, TipsLot
logger.debug(f"\n\nEquipmentRole: {equipmentrole}, Equipment: {equipment}, Process: {processversion}, Tips: {tips}\n\n")
try: try:
equipment_of_interest = next( equipment_of_interest = next(
(item for item in self.procedure.equipment if item.equipmentrole == equipmentrole)) (item for item in self.procedure.equipment if item.equipmentrole == equipmentrole))
@@ -156,10 +156,10 @@ class ProcedureCreation(QDialog):
@pyqtSlot(str, str, str, str) @pyqtSlot(str, str, str, str)
def add_new_reagent(self, reagentrole: str, name: str, lot: str, expiry: str): def add_new_reagent(self, reagentrole: str, name: str, lot: str, expiry: str):
from backend.validators.pydant import PydReagent from backend.validators.pydant import PydReagentLot
expiry = datetime.datetime.strptime(expiry, "%Y-%m-%d") expiry = datetime.datetime.strptime(expiry, "%Y-%m-%d")
logger.debug(f"{reagentrole}, {name}, {lot}, {expiry}") logger.debug(f"{reagentrole}, {name}, {lot}, {expiry}")
pyd = PydReagent(reagentrole=reagentrole, name=name, lot=lot, expiry=expiry) pyd = PydReagentLot(reagentrole=reagentrole, name=name, lot=lot, expiry=expiry)
self.procedure.reagentrole[reagentrole].insert(0, pyd) self.procedure.reagentrole[reagentrole].insert(0, pyd)
self.set_html() self.set_html()
@@ -171,6 +171,12 @@ class ProcedureCreation(QDialog):
return return
self.procedure.update_reagents(reagentrole=reagentrole, name=name, lot=lot, expiry=expiry) self.procedure.update_reagents(reagentrole=reagentrole, name=name, lot=lot, expiry=expiry)
@pyqtSlot(str, result=list)
def get_reagent_names(self, reagentrole_name: str):
from backend.db.models import ReagentRole
reagentrole = ReagentRole.query(name=reagentrole_name)
return [item.name for item in reagentrole.get_reagents(proceduretype=self.procedure.proceduretype)]
def return_sql(self, new: bool = False): def return_sql(self, new: bool = False):
output = self.procedure.to_sql(new=new) output = self.procedure.to_sql(new=new)
return output return output

View File

@@ -194,7 +194,7 @@ function contextListener() {
function clickListener() { function clickListener() {
document.addEventListener( "click", function(e) { document.addEventListener( "click", function(e) {
var clickeElIsLink = clickInsideElement( e, contextMenuLinkClassName ); var clickeElIsLink = clickInsideElement( e, contextMenuLinkClassName );
backend.log(e.target.id)
if ( clickeElIsLink ) { if ( clickeElIsLink ) {
e.preventDefault(); e.preventDefault();
menuItemListener( clickeElIsLink ); menuItemListener( clickeElIsLink );

View File

@@ -42,7 +42,7 @@ var changed_it = new Event('change');
var reagentRoles = document.getElementsByClassName("reagentrole"); var reagentRoles = document.getElementsByClassName("reagentrole");
for(let i = 0; i < reagentRoles.length; i++) { for(let i = 0; i < reagentRoles.length; i++) {
reagentRoles[i].addEventListener("change", function() { reagentRoles[i].addEventListener("change", async function() {
if (reagentRoles[i].value.includes("--New--")) { if (reagentRoles[i].value.includes("--New--")) {
// alert("Create new reagent.") // alert("Create new reagent.")
var br = document.createElement("br"); var br = document.createElement("br");
@@ -50,9 +50,15 @@ for(let i = 0; i < reagentRoles.length; i++) {
var new_form = document.createElement("form"); var new_form = document.createElement("form");
new_form.setAttribute("class", "new_reagent_form") new_form.setAttribute("class", "new_reagent_form")
new_form.setAttribute("id", reagentRoles[i].id + "_addition") new_form.setAttribute("id", reagentRoles[i].id + "_addition")
var rr_name = document.createElement("input"); var rr_name = document.createElement("select");
rr_name.setAttribute("type", "text");
rr_name.setAttribute("id", "new_" + reagentRoles[i].id + "_name"); rr_name.setAttribute("id", "new_" + reagentRoles[i].id + "_name");
var rr_options = await backend.get_reagent_names(reagentRoles[i].id).then(
function(result) {
result.forEach( function(item) {
rr_name.options.add( new Option(item));
});
}
);
var rr_name_label = document.createElement("label"); var rr_name_label = document.createElement("label");
rr_name_label.setAttribute("for", "new_" + reagentRoles[i].id + "_name"); rr_name_label.setAttribute("for", "new_" + reagentRoles[i].id + "_name");
rr_name_label.innerHTML = "Name:"; rr_name_label.innerHTML = "Name:";