修复了文件夹不存在时无法保存的问题

This commit is contained in:
Yorusora
2025-05-20 21:39:22 +08:00
parent 497e1ced27
commit 6215399cf7
7 changed files with 35 additions and 0 deletions

View File

@ -969,6 +969,8 @@ class Data:
return Result().success(info=Constants.INPUT_FINISHED) return Result().success(info=Constants.INPUT_FINISHED)
def save(self, epoch): def save(self, epoch):
if (not Path(Config["Path"]["Save"]).parent.exists()) or (not Path(Config["Path"]["Save"]).parent.is_dir()):
Path(Config["Path"]["Save"]).parent.mkdir(parents=True, exist_ok=True)
try: try:
pos = Config["pos"] pos = Config["pos"]
ApplyFrequency = Config["ApplyFrequency"] ApplyFrequency = Config["ApplyFrequency"]

View File

@ -322,6 +322,9 @@ class Data:
return Result().success(info=Constants.CUT_PSG_ALIGN_LABEL_FINISHED) return Result().success(info=Constants.CUT_PSG_ALIGN_LABEL_FINISHED)
def save(self): def save(self):
if (not Path(Config["Path"]["SaveFolder"]).exists()) or (not Path(Config["Path"]["SaveFolder"]).is_dir()):
Path(Config["Path"]["SaveFolder"]).mkdir(parents=True, exist_ok=True)
for raw in self.raw.values(): for raw in self.raw.values():
if len(raw) == 0: if len(raw) == 0:
return Result().failure(info=Constants.SAVING_FAILURE + return Result().failure(info=Constants.SAVING_FAILURE +

View File

@ -461,6 +461,9 @@ class Data:
return Result().success(info=Constants.DETECT_JPEAK_PREDICT_FINISHED) return Result().success(info=Constants.DETECT_JPEAK_PREDICT_FINISHED)
def save(self, chunk): def save(self, chunk):
if (not Path(Config["Path"]["Save"]).parent.exists()) or (not Path(Config["Path"]["Save"]).parent.is_dir()):
Path(Config["Path"]["Save"]).parent.mkdir(parents=True, exist_ok=True)
if self.peak is None: if self.peak is None:
return Result().failure(info=Constants.SAVING_FAILURE + Constants.FAILURE_REASON["Data_Not_Exist"]) return Result().failure(info=Constants.SAVING_FAILURE + Constants.FAILURE_REASON["Data_Not_Exist"])

View File

@ -452,6 +452,9 @@ class Data:
return Result().success(info=Constants.DETECT_RPEAK_PREDICT_FINISHED) return Result().success(info=Constants.DETECT_RPEAK_PREDICT_FINISHED)
def save(self, chunk): def save(self, chunk):
if (not Path(Config["Path"]["Save"]).parent.exists()) or (not Path(Config["Path"]["Save"]).parent.is_dir()):
Path(Config["Path"]["Save"]).parent.mkdir(parents=True, exist_ok=True)
if self.peak is None: if self.peak is None:
return Result().failure(info=Constants.SAVING_FAILURE + Constants.FAILURE_REASON["Data_Not_Exist"]) return Result().failure(info=Constants.SAVING_FAILURE + Constants.FAILURE_REASON["Data_Not_Exist"])

View File

@ -1011,6 +1011,9 @@ class Data:
return Result().success(info=Constants.PREPROCESS_FINISHED) return Result().success(info=Constants.PREPROCESS_FINISHED)
def save(self, chunk): def save(self, chunk):
if (not Path(Config["Path"]["Save"]).parent.exists()) or (not Path(Config["Path"]["Save"]).parent.is_dir()):
Path(Config["Path"]["Save"]).parent.mkdir(parents=True, exist_ok=True)
if self.corrected_peak is None: if self.corrected_peak is None:
return Result().failure(info=Constants.SAVING_FAILURE + Constants.FAILURE_REASON["Data_Not_Exist"]) return Result().failure(info=Constants.SAVING_FAILURE + Constants.FAILURE_REASON["Data_Not_Exist"])

View File

@ -1827,6 +1827,9 @@ class Data:
return Result().success(info=info) return Result().success(info=info)
def save_alignInfo(self): def save_alignInfo(self):
if (not Path(Config["Path"]["Save_AlignInfo"]).parent.exists()) or (not Path(Config["Path"]["Save_AlignInfo"]).parent.is_dir()):
Path(Config["Path"]["Save_AlignInfo"]).parent.mkdir(parents=True, exist_ok=True)
try: try:
save_data = { save_data = {
"front": { "front": {
@ -1867,6 +1870,9 @@ class Data:
return Result().success(info=Constants.PRECISELY_ALIGN_SAVING_ALIGNINFO_FINISHED) return Result().success(info=Constants.PRECISELY_ALIGN_SAVING_ALIGNINFO_FINISHED)
def save_res_orgBcg(self, chunk): def save_res_orgBcg(self, chunk):
if (not Path(Config["Path"]["Save_OrgBCG"]).parent.exists()) or (not Path(Config["Path"]["Save_OrgBCG"]).parent.is_dir()):
Path(Config["Path"]["Save_OrgBCG"]).parent.mkdir(parents=True, exist_ok=True)
if self.res_orgBcg is None: if self.res_orgBcg is None:
return Result().failure(info=Constants.PRECISELY_ALIGN_SAVING_RES_ORGBCG_FAILURE + return Result().failure(info=Constants.PRECISELY_ALIGN_SAVING_RES_ORGBCG_FAILURE +
Constants.FAILURE_REASON["res_orgBcg_Not_Exist"]) Constants.FAILURE_REASON["res_orgBcg_Not_Exist"])
@ -1880,6 +1886,9 @@ class Data:
return Result().success(info=Constants.PRECISELY_ALIGN_SAVING_RES_ORGBCG_FINISHED) return Result().success(info=Constants.PRECISELY_ALIGN_SAVING_RES_ORGBCG_FINISHED)
def save_res_BCG(self, chunk): def save_res_BCG(self, chunk):
if (not Path(Config["Path"]["Save_BCG"]).parent.exists()) or (not Path(Config["Path"]["Save_BCG"]).parent.is_dir()):
Path(Config["Path"]["Save_BCG"]).parent.mkdir(parents=True, exist_ok=True)
if self.res_BCG is None: if self.res_BCG is None:
return Result().failure(info=Constants.PRECISELY_ALIGN_SAVING_RES_BCG_FAILURE + return Result().failure(info=Constants.PRECISELY_ALIGN_SAVING_RES_BCG_FAILURE +
Constants.FAILURE_REASON["res_BCG_Not_Exist"]) Constants.FAILURE_REASON["res_BCG_Not_Exist"])
@ -1893,6 +1902,9 @@ class Data:
return Result().success(info=Constants.PRECISELY_ALIGN_SAVING_RES_BCG_FINISHED) return Result().success(info=Constants.PRECISELY_ALIGN_SAVING_RES_BCG_FINISHED)
def save_cut_ECG(self, chunk): def save_cut_ECG(self, chunk):
if (not Path(Config["Path"]["Save_ECG"]).parent.exists()) or (not Path(Config["Path"]["Save_ECG"]).parent.is_dir()):
Path(Config["Path"]["Save_ECG"]).parent.mkdir(parents=True, exist_ok=True)
if self.cut_ECG is None: if self.cut_ECG is None:
return Result().failure(info=Constants.PRECISELY_ALIGN_SAVING_CUT_ECG_FAILURE + return Result().failure(info=Constants.PRECISELY_ALIGN_SAVING_CUT_ECG_FAILURE +
Constants.FAILURE_REASON["cut_ECG_Not_Exist"]) Constants.FAILURE_REASON["cut_ECG_Not_Exist"])
@ -1906,6 +1918,9 @@ class Data:
return Result().success(info=Constants.PRECISELY_ALIGN_SAVING_CUT_ECG_FINISHED) return Result().success(info=Constants.PRECISELY_ALIGN_SAVING_CUT_ECG_FINISHED)
def save_Jpeak(self, chunk): def save_Jpeak(self, chunk):
if (not Path(Config["Path"]["Save_ECG"]).parent.exists()) or (not Path(Config["Path"]["Save_ECG"]).parent.is_dir()):
Path(Config["Path"]["Save_ECG"]).parent.mkdir(parents=True, exist_ok=True)
if self.cut_Jpeak is None: if self.cut_Jpeak is None:
return Result().failure(info=Constants.PRECISELY_ALIGN_SAVING_CUT_JPEAK_FAILURE + return Result().failure(info=Constants.PRECISELY_ALIGN_SAVING_CUT_JPEAK_FAILURE +
Constants.FAILURE_REASON["cut_Jpeak_Not_Exist"]) Constants.FAILURE_REASON["cut_Jpeak_Not_Exist"])
@ -1919,6 +1934,9 @@ class Data:
return Result().success(info=Constants.PRECISELY_ALIGN_SAVING_CUT_JPEAK_FINISHED) return Result().success(info=Constants.PRECISELY_ALIGN_SAVING_CUT_JPEAK_FINISHED)
def save_Rpeak(self, chunk): def save_Rpeak(self, chunk):
if (not Path(Config["Path"]["Save_Rpeak"]).parent.exists()) or (not Path(Config["Path"]["Save_Rpeak"]).parent.is_dir()):
Path(Config["Path"]["Save_Rpeak"]).parent.mkdir(parents=True, exist_ok=True)
if self.cut_Rpeak is None: if self.cut_Rpeak is None:
return Result().failure(info=Constants.PRECISELY_ALIGN_SAVING_CUT_RPEAK_FAILURE + return Result().failure(info=Constants.PRECISELY_ALIGN_SAVING_CUT_RPEAK_FAILURE +
Constants.FAILURE_REASON["cut_Rpeak_Not_Exist"]) Constants.FAILURE_REASON["cut_Rpeak_Not_Exist"])

View File

@ -449,6 +449,9 @@ class Data:
return Result().success(info=Constants.PREPROCESS_FINISHED) return Result().success(info=Constants.PREPROCESS_FINISHED)
def save(self, chunk): def save(self, chunk):
if (not Path(Config["Path"]["Save"]).parent.exists()) or (not Path(Config["Path"]["Save"]).parent.is_dir()):
Path(Config["Path"]["Save"]).parent.mkdir(parents=True, exist_ok=True)
if self.processed_data is None: if self.processed_data is None:
return Result().failure(info=Constants.SAVING_FAILURE + Constants.FAILURE_REASON["Data_Not_Exist"]) return Result().failure(info=Constants.SAVING_FAILURE + Constants.FAILURE_REASON["Data_Not_Exist"])