add password failure

Jules-WinnfieldX · Apr 8, 2024 · dddfb66 · dddfb66
1 parent 9d474f2
commit dddfb66
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cyberdrop_dl/scraper/crawlers/gofile_crawler.py b/cyberdrop_dl/scraper/crawlers/gofile_crawler.py
@@ -8,7 +8,7 @@
 from aiolimiter import AsyncLimiter
 from yarl import URL
 
-from cyberdrop_dl.clients.errors import ScrapeFailure, DownloadFailure
+from cyberdrop_dl.clients.errors import ScrapeFailure, DownloadFailure, PasswordProtected
 from cyberdrop_dl.scraper.crawler import Crawler
 from cyberdrop_dl.utils.dataclasses.url_objects import ScrapeItem
 from cyberdrop_dl.utils.utilities import get_filename_and_ext, error_handling_wrapper
@@ -61,6 +61,10 @@ async def album(self, scrape_item: ScrapeItem) -> None:
                 raise ScrapeFailure(e.status, e.message)
 
         JSON_Resp = JSON_Resp['data']
+
+        if "password" in JSON_Resp:
+            raise PasswordProtected()
+
         title = await self.create_title(JSON_Resp["name"], content_id, None)
 
         contents = JSON_Resp["children"]