saving report to timestamped folder instead of outputs

Author: sofiboselli

src/migration_report_package/graph_builder.py

@@ -12,6 +12,7 @@
 class MigrationState(TypedDict):
     """State for the migration graph execution."""
     input_dir: str
+    latest_dir: str
     raw: Dict[str, Any]
     cleaned_raw: Dict[str, Any]
     count: Dict[str, Any]
@@ -33,6 +34,7 @@ def input_node(state: MigrationState) -> MigrationState:
         output_dirs.append((run_dt, res))
 
     _ , latest = max(output_dirs, key=lambda x: x[0])
+    state["latest_dir"] = latest
     ## Get translation results and evaluation notes
     raw = {"translation_results": [], "evaluation": []}
     for name in os.listdir(latest):
@@ -142,6 +144,7 @@ def run(self, input_path: str) -> Dict[str, Any]:
         try:
             initial_state: MigrationState = {
             "input_dir": input_path,
+            "latest_dir": None,
             "raw": [],
             "count": None,
             "json_report": None,
@@ -151,7 +154,8 @@ def run(self, input_path: str) -> Dict[str, Any]:
             final_state = self.compiled_graph.invoke(initial_state)
             report = final_state["md_report"] or {}
             json_report = final_state["json_report"] or {}
-            return report[0], json_report
+            latest_dir = final_state["latest_dir"] or {}
+            return report[0], json_report, latest_dir
 
         except Exception as e:
             raise

src/migration_report_package/main.py

@@ -60,14 +60,14 @@ def main():
     if args.output_dir:
         input_dir = os.path.dirname(args.output_dir)
 
-    md_report, json_report = graph.run(input_dir)
+    md_report, json_report, latest_dir = graph.run(input_dir)
 
     print("Report done!")
 
     if args.md_output:
         output_dir = os.path.dirname(args.md_output)
 
-    save_results(output_dir,md_report)
+    save_results(latest_dir,md_report)
 
     print("JSON Report: ",json_report)