fix: 跳过的下载记录不添加到data.yaml里面 (#193)

tangyoha · Sep 4, 2024 · 0d82a96 · 0d82a96
1 parent 506b98c
commit 0d82a96
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/module/app.py b/module/app.py
@@ -796,13 +796,13 @@ def update_config(self, immediate: bool = True):
             unfinished_ids = set(value.ids_to_retry)
 
             for it in value.ids_to_retry:
-                if DownloadStatus.SuccessDownload == value.node.download_status.get(
+                if  value.node.download_status.get(
                     it, DownloadStatus.FailedDownload
-                ):
+                ) in [DownloadStatus.SuccessDownload, DownloadStatus.SkipDownload]:
                     unfinished_ids.remove(it)
 
             for _idx, _value in value.node.download_status.items():
-                if DownloadStatus.SuccessDownload != _value:
+                if DownloadStatus.SuccessDownload != _value and DownloadStatus.SkipDownload != _value:
                     unfinished_ids.add(_idx)
 
             self.chat_download_config[key].ids_to_retry = list(unfinished_ids)