Fix downloader _should_retry criteria

ieasybooks · Jun 29, 2024 · bf2d47d · bf2d47d
1 parent 51dbb91
commit bf2d47d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tafrigh/downloader.py b/tafrigh/downloader.py
@@ -56,10 +56,10 @@ def _initialize_youtube_dl_without_archive(self) -> None:
   def _should_retry(self, url_data: dict[str, Any]) -> bool:
     if '_type' in url_data and url_data['_type'] == 'playlist':
       for entry in url_data['entries']:
-        if entry and not os.path.exists(os.path.join(self.output_dir, f"{entry['id']}.{entry['ext']}")):
+        if entry and not os.path.exists(os.path.join(self.output_dir, f"{entry['id']}.mp3")):
           return True
     else:
-      if not os.path.exists(os.path.join(self.output_dir, f"{url_data['id']}.{url_data['ext']}")):
+      if not os.path.exists(os.path.join(self.output_dir, f"{url_data['id']}.mp3")):
         return True
 
     return False