[TPC3] - load feito

TPC3/main.py

@@ -0,0 +1,74 @@
+import re
+from dataclasses import dataclass
+
+processes = {}
+
+alinea_a = {}
+
+
+@dataclass
+class Process:
+    name: str
+    grau: str
+    proc_number: int
+
+
+@dataclass
+class Line:
+    folder: int
+    year: int
+    month: int
+    day: int
+    name: str
+    dad: str
+    mom: str
+    associated_processes: list[Process]
+
+
+def main():
+    load()
+    print(processes)
+
+
+def load():
+    regex = r"(\d+)::(\d+)-(\d+)-(\d+)::([A-Za-z ]+)::([A-Za-z ]+)::([A-Za-z ]+)::(([^ \n]* *)*)::"
+    dataset = open("processos.txt", "r")
+    lines = dataset.readlines()
+    for line in lines:
+        if line.strip() == "":
+            continue
+        matched_regex = re.match(regex, line)
+        if matched_regex is not None:
+            folder = int(matched_regex.group(1))
+            year = int(matched_regex.group(2))
+            month = int(matched_regex.group(3))
+            day = int(matched_regex.group(4))
+            name = matched_regex.group(5)
+            dad = matched_regex.group(6)
+            mom = matched_regex.group(7)
+
+            proc_str = matched_regex.group(8)
+            regex2 = r"([A-Za-z ]+),([A-Za-z ]+)\. Proc\.(\d+)\. *"
+            proc_match = re.findall(regex2, proc_str)
+
+            proc_list = []
+            if proc_match is not None:
+                for match in proc_match:
+                    proc_list.append(
+                        Process(match[0].strip(), match[1].strip(), int(match[2]))
+                    )
+
+            line_obj = Line(folder, year, month, day, name, dad, mom, proc_list)
+            if year not in processes:
+                processes[year] = {}
+            if month not in processes[year]:
+                processes[year][month] = {}
+            if day not in processes[year][month]:
+                processes[year][month][day] = []
+            processes[year][month][day].append(line_obj)
+
+    dataset.close()
+
+
+if __name__ == "__main__":
+    main()