Large scale refactor to improve db efficiency

src/submissions/backend/db/models/__init__.py

@@ -1,12 +1,46 @@
 '''
 Contains all models for sqlalchemy
 '''
-from sqlalchemy.ext.declarative import declarative_base
+from typing import Any
+from sqlalchemy.orm import declarative_base
+import logging
+from pprint import pformat
 
 Base = declarative_base()
 metadata = Base.metadata
 
+logger = logging.getLogger(f"submissions.{__name__}")
+
+def find_subclasses(parent:Any, attrs:dict) -> Any:
+    """
+    Finds subclasses of a parent that does contain all 
+    attributes if the parent does not.
+
+    Args:
+        parent (_type_): Parent class.
+        attrs (dict): Key:Value dictionary of attributes
+
+    Raises:
+        AttributeError: Raised if no subclass is found.
+
+    Returns:
+        _type_: Parent or subclass.
+    """    
+    if len(attrs) == 0:
+        return parent
+    if any([not hasattr(parent, attr) for attr in attrs]):
+        # looks for first model that has all included kwargs
+        try:
+            model = [subclass for subclass in parent.__subclasses__() if all([hasattr(subclass, attr) for attr in attrs])][0]
+        except IndexError as e:
+            raise AttributeError(f"Couldn't find existing class/subclass of {parent} with all attributes:\n{pformat(attrs)}")
+    else:
+        model = parent
+    logger.debug(f"Using model: {model}")
+    return model
+
 from .controls import Control, ControlType
 from .kits import KitType, ReagentType, Reagent, Discount, KitTypeReagentTypeAssociation, SubmissionType, SubmissionTypeKitTypeAssociation
 from .organizations import Organization, Contact
 from .submissions import BasicSubmission, BacterialCulture, Wastewater, WastewaterArtic, WastewaterSample, BacterialCultureSample, BasicSample, SubmissionSampleAssociation, WastewaterAssociation
+