Better flexibility with parsers pulling methods from database objects.

src/submissions/backend/validators/__init__.py

@@ -0,0 +1,92 @@
+import logging, re
+from pathlib import Path
+from openpyxl import load_workbook
+from backend.db.models import BasicSubmission
+from tools import Settings
+
+
+logger = logging.getLogger(f"submissions.{__name__}")
+
+class RSLNamer(object):
+    """
+    Object that will enforce proper formatting on RSL plate names.
+    NOTE: Depreciated in favour of object based methods in 'submissions.py'
+    """
+    def __init__(self, ctx, instr:str, sub_type:str|None=None):
+        self.ctx = ctx
+        self.submission_type = sub_type
+        
+        if self.submission_type == None:
+            self.submission_type = self.retrieve_submission_type(ctx=self.ctx, instr=instr)
+        print(self.submission_type)
+        if self.submission_type != None:
+            enforcer = BasicSubmission.find_polymorphic_subclass(polymorphic_identity=self.submission_type)
+            self.parsed_name = self.retrieve_rsl_number(instr=instr, regex=enforcer.get_regex())
+            self.parsed_name = enforcer.enforce_name(ctx=ctx, instr=self.parsed_name)
+
+    @classmethod
+    def retrieve_submission_type(cls, ctx:Settings, instr:str|Path) -> str:
+        match instr:
+            case Path():
+                logger.debug(f"Using path method.")
+                if instr.exists():
+                    wb = load_workbook(instr)
+                    try:
+                        submission_type = [item.strip().title() for item in wb.properties.category.split(";")][0]
+                    except AttributeError:
+                        try:
+                            for type in ctx.submission_types:
+                                # This gets the *first* submission type that matches the sheet names in the workbook 
+                                if wb.sheetnames == ctx.submission_types[type]['excel_map']:
+                                    submission_type = type.title()
+                        except:
+                            submission_type = cls.retrieve_submission_type(ctx=ctx, instr=instr.stem.__str__())
+            case str():
+                regex = BasicSubmission.construct_regex()
+                logger.debug(f"Using string method.")
+                m = regex.search(instr)
+                try:
+                    submission_type = m.lastgroup
+                except AttributeError as e:
+                    logger.critical("No RSL plate number found or submission type found!")
+            case _:
+                submission_type = None
+        if submission_type == None:
+            from frontend.custom_widgets import SubmissionTypeSelector
+            dlg = SubmissionTypeSelector(ctx, title="Couldn't parse submission type.", message="Please select submission type from list below.")
+            if dlg.exec():
+                submission_type = dlg.parse_form()
+        submission_type = submission_type.replace("_", " ")
+        return submission_type
+
+    @classmethod
+    def retrieve_rsl_number(cls, instr:str|Path, regex:str|None=None):
+        """
+        Uses regex to retrieve the plate number and submission type from an input string
+
+        Args:
+            in_str (str): string to be parsed
+        """    
+        if regex == None:
+            regex = BasicSubmission.construct_regex()
+        else:
+            regex = re.compile(rf'{regex}', re.IGNORECASE | re.VERBOSE)
+        match instr:
+            case Path():
+                m = regex.search(instr.stem)
+            case str():
+                logger.debug(f"Using string method.")
+                m = regex.search(instr)
+            case _:
+                pass
+        if m != None:
+            try:
+                parsed_name = m.group().upper().strip(".")
+            except:
+                parsed_name = None
+        else: 
+            parsed_name = None
+        logger.debug(f"Got parsed submission name: {parsed_name}")
+        return parsed_name
+        
+from .pydant import *