diff --git a/lib_afc_mosaic/universe_map_file.py b/lib_afc_mosaic/universe_map_file.py index 3bbe415..02e04ef 100644 --- a/lib_afc_mosaic/universe_map_file.py +++ b/lib_afc_mosaic/universe_map_file.py @@ -57,6 +57,7 @@ class AFCMUniverseMapFile: "incoming_count": { "total": 0 }, "removed_count": { "total": 0 }, "cleaned_count": { "total": 0 }, + "mosaic_count": { "total": 0 }, "removed_details": {}, "timings": {} } @@ -465,8 +466,8 @@ class AFCMUniverseMapFile: self.metadata["total_rows"] = len(self.final_rows) # Need to write the FINAL file + final_filename = self.metadata.get('mosaic_filename', 'MOSAIC.csv') if len(self.final_rows) > 0: - final_filename = self.metadata.get('mosaic_filename', 'MOSAIC.csv') final_full_pathname = f"/tmp/{final_filename}" write_csv_file(final_full_pathname, self.final_rows[0].keys(), self.final_rows) @@ -479,6 +480,9 @@ class AFCMUniverseMapFile: # remove the temp file now os.remove(final_full_pathname) + self.stats['mosaic_count'][final_filename] = len(self.final_rows) + self.stats['mosaic_count']["total"] = self.stats["mosaic_count"][final_filename] + def write_stats_file(self): """