diff --git a/csst_common/data_manager.py b/csst_common/data_manager.py index 5f67bf7719657eb074b957d471e5b92a7cc2ee5e..69ebb8b7cf56d1cf42f3432ee636d918a43ec654 100644 --- a/csst_common/data_manager.py +++ b/csst_common/data_manager.py @@ -157,14 +157,14 @@ class CsstMsDataManager: _survey: str = "MS", obs_type: str = "SCI", l0_post: str = "L0_1", - log_ppl="", - log_mod="", - clear_dir=False, - verbose=True, - n_jobs=18, - backend="multiprocessing", - device="CPU", - stamps="", + log_ppl: str = "", + log_mod: str = "", + clear_dir: bool = False, + verbose: bool = True, + n_jobs: int = 18, + backend: str = "multiprocessing", + device: str = "CPU", + stamps: str = "", ): # set DFS log dir @@ -256,15 +256,13 @@ class CsstMsDataManager: self.custom_dark = None self.custom_flat = None - if stamps == "" or stamps is None: - self.stamps = None - else: - self.stamps = open(stamps, "w+") - self.write_stamp() + self.stamps = stamps + self.write_stamp() def write_stamp(self): - if self.stamps is not None: - self.stamps.write(f"{time.Time.now().isot}\n") + if self.stamps is not None and not self.stamps == "": + with open(self.stamps, "a+") as f: + f.write(f"{time.Time.now().isot}\n") # DFS APIs @property