From bca9c7879475a9059991ce86d5525a66afd49602 Mon Sep 17 00:00:00 2001 From: Nate Tellis Date: Tue, 28 Jan 2025 16:30:55 -0500 Subject: [PATCH] by default, remove most files from fo dir after od --- src/adam_impact_study/fo_od.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/src/adam_impact_study/fo_od.py b/src/adam_impact_study/fo_od.py index 10ca6bc..e80f6a4 100644 --- a/src/adam_impact_study/fo_od.py +++ b/src/adam_impact_study/fo_od.py @@ -22,6 +22,15 @@ logger.setLevel(os.environ.get("ADAM_LOG_LEVEL", "INFO")) +fo_file_whitelist = [ + "total.json", + "covar.json", + "elem_short.json", + "debug.txt" "observations.ades", + "environ.dat", +] + + def observations_to_ades(observations: Observations) -> Tuple[str, ADESObservations]: """ Convert Observations to ADES format string. @@ -102,6 +111,7 @@ def observations_to_ades(observations: Observations) -> Tuple[str, ADESObservati def run_fo_od( observations: Observations, fo_result_dir: str, + fo_dir_cleanup: bool = True, ) -> Tuple[Orbits, ADESObservations, Optional[str]]: """Run Find_Orb orbit determination with directory-based paths @@ -134,10 +144,14 @@ def run_fo_od( orbit, rejected, error = fo( ades_string, - clean_up=True, out_dir=fo_result_dir, ) + if fo_dir_cleanup: + for file in os.listdir(fo_result_dir): + if file not in fo_file_whitelist: + os.remove(os.path.join(fo_result_dir, file)) + # Re-assign orbit_id to the original value if we found an orbit if len(orbit) > 0: orbit = orbit.set_column("orbit_id", observations.orbit_id[:1])