From aacf2c91e88caaa1387993b6f394151c9240e0ec Mon Sep 17 00:00:00 2001 From: BO ZHANG Date: Tue, 4 Jul 2023 13:51:49 +0800 Subject: [PATCH] assert shutter file exists --- csst_common/data_manager.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/csst_common/data_manager.py b/csst_common/data_manager.py index 74096f1..e54da2d 100644 --- a/csst_common/data_manager.py +++ b/csst_common/data_manager.py @@ -580,16 +580,17 @@ class CsstMsDataManager: def get_shutter(self, detector=6): """ get flat data """ - if self.custom_flat is None: - return os.path.join( - self.path_aux, - "C6.2_ref_mbi", - "CSST_MSC_MS_SHUTTER_{:02d}_{:04d}.fits".format( - detector, 1 if self.ref_version is None else self.ref_version - ) + fp = os.path.join( + self.path_aux, + "C6.2_ref_mbi", + "CSST_MSC_MS_SHUTTER_{:02d}_{:04d}.fits".format( + detector, 1 if self.ref_version is None else self.ref_version ) + ) + if not os.path.exists(fp): + raise FileNotFoundError(fp) else: - return self.custom_flat.format("flat", detector) + return fp def get_axeconf(self): return os.path.join(self.path_aux, "axeconf") # "/home/csstpipeline/L1Pipeline/aux/axeconf" -- GitLab