UPDATED: compact argument, half finished

Author: AleixMT

src/arguments.py

@@ -5,7 +5,11 @@
 
 from NAF import NAF_TO_DNI, validate_parse_naf
 from custom_except import *
-from defines import NAF_DATA_PATH
+from defines import NAF_DATA_PATH, DocType
+
+
+def get_compact_init():
+    return {DocType.SALARY: False, DocType.PROOFS: False, DocType.CONTRACT: False, DocType.RNT: False}
 
 
 def parse_date(value, formatting="%Y_%m"):
@@ -32,6 +36,18 @@ def parse_naf(value):
         raise ArgumentNafNotPresent("NAF is not valid" + e.__str__())
 
 
+def parse_compact(value):
+    to_compact = get_compact_init()
+    try:
+        for s in value.split(","):
+            doc_type = DocType.from_string(s)
+            to_compact[doc_type] = True
+        return to_compact
+    except ValueError as e:
+        print(f"Error: {e}")
+        exit(1)
+
+
 def parse_arguments(valid_nafs):
     """Parse and validate command-line arguments"""
     parser = argparse.ArgumentParser(description="Process NAF and date range.")
@@ -42,6 +58,10 @@ def parse_arguments(valid_nafs):
     parser.add_argument("-e", "--end", type=parse_date, required=True, help="End date (YYYY-MM)")
     parser.add_argument("-a", "--author", type=parse_author, required=True, help="author's email doing request")
 
+    parser.add_argument("-c", "--compact", type=parse_compact, required=False, default=get_compact_init() , help="Comma separated list of values "
+                                                                          "that indicate "
+                        "which documents need to be merged in one signle PDF in the output. Possible values are: " + ",".join([dt.value.__str__() for dt in DocType]))
+
     args = parser.parse_args()
 
     return args

src/defines.py

@@ -30,6 +30,28 @@
 NAF_DATA_PATH = os.path.join(INPUT_FOLDER, "NAF_DNI.xlsx")
 
 
+class DocType(Enum):
+    SALARY = "salary"
+    CONTRACT = "contract"
+    RLC = "RLC"
+    RNT = "RNT"
+    PROOFS = "proofs"
+
+    _aliases = {
+        SALARY: {"salary", "SALARY", "Salary", "payslip"},
+        CONTRACT: {"contract", "CONTRACT", "Contract", "agreement"},
+        RLC: {"RLC", "rlc", "R.L.C."},
+        RNT: {"RNT", "rnt", "R.N.T."},
+    }
+
+    @classmethod
+    def from_string(cls, value: str):
+        for doctype, aliases in cls._aliases:
+            if value.strip() in aliases:
+                return doctype
+        raise ValueError(f"Unknown document type: {value}")
+
+
 class RLCType(Enum):
     REGULAR = "regular"
     DELAY = "delay"
@@ -40,3 +62,4 @@ class SalaryType(Enum):
     REGULAR = "Nomines"
     DELAY = "Atrasos"
     EXTRA = "Extres"
+

src/filesystem.py

@@ -26,6 +26,8 @@ def list_dir(input_folder):
     return file_names
 
 
+
+
 def compute_paths(args):
     NOW = datetime.now().strftime("%Y-%m-%d_%H:%M:%S")