add password failure

Jules-WinnfieldX · Apr 8, 2024 · 9d474f2 · 9d474f2
1 parent a42b015
commit 9d474f2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cyberdrop_dl/utils/utilities.py b/cyberdrop_dl/utils/utilities.py
@@ -13,7 +13,8 @@
 import rich
 from yarl import URL
 
-from cyberdrop_dl.clients.errors import NoExtensionFailure, FailedLoginFailure, InvalidContentTypeFailure
+from cyberdrop_dl.clients.errors import NoExtensionFailure, FailedLoginFailure, InvalidContentTypeFailure, \
+    PasswordProtected
 
 if TYPE_CHECKING:
     from typing import Tuple
@@ -64,6 +65,10 @@ async def wrapper(self, *args, **kwargs):
             await log(f"Scrape Failed: {link} (No File Extension)", 40)
             await self.manager.log_manager.write_scrape_error_log(link, " No File Extension")
             await self.manager.progress_manager.scrape_stats_progress.add_failure("No File Extension")
+        except PasswordProtected:
+            await log(f"Scrape Failed: {link} (Password Protected)", 40)
+            await self.manager.log_manager.write_unsupported_urls_log(link)
+            await self.manager.progress_manager.scrape_stats_progress.add_failure("Password Protected")
         except FailedLoginFailure:
             await log(f"Scrape Failed: {link} (Failed Login)", 40)
             await self.manager.log_manager.write_scrape_error_log(link, " Failed Login")