From 22dcd6472a17cee2ec8cc50e22d8f6e07e7cb3ae Mon Sep 17 00:00:00 2001
From: BO ZHANG <bozhang@nao.cas.cn>
Date: Fri, 17 Feb 2023 00:02:26 +0800
Subject: [PATCH] use str for dm.stamps

---
 csst_common/data_manager.py | 28 +++++++++++++---------------
 1 file changed, 13 insertions(+), 15 deletions(-)

diff --git a/csst_common/data_manager.py b/csst_common/data_manager.py
index 5f67bf7..69ebb8b 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
-- 
GitLab