add error wrapper for redgifs token

Jules-WinnfieldX · Jan 1, 2024 · 91ffad8 · 91ffad8
1 parent 65756f2
commit 91ffad8
Show file tree

Hide file tree

Showing 4 changed files with 83 additions and 80 deletions.
diff --git a/cyberdrop_dl/__init__.py b/cyberdrop_dl/__init__.py
@@ -1 +1 @@
-__version__ = "5.1.28"
+__version__ = "5.1.29"
diff --git a/cyberdrop_dl/scraper/crawlers/redgifs_crawler.py b/cyberdrop_dl/scraper/crawlers/redgifs_crawler.py
@@ -28,12 +28,13 @@ async def fetch(self, scrape_item: ScrapeItem) -> None:
         task_id = await self.scraping_progress.add_task(scrape_item.url)
 
         if not self.token:
-            await self.manage_token()
+            await self.manage_token(self.redgifs_api / "v2/auth/temporary")
 
-        if "users" in scrape_item.url.parts:
-            await self.user(scrape_item)
-        else:
-            await self.post(scrape_item)
+        if self.token:
+            if "users" in scrape_item.url.parts:
+                await self.user(scrape_item)
+            else:
+                await self.post(scrape_item)
 
         await self.scraping_progress.remove_task(task_id)
 
@@ -89,9 +90,10 @@ async def post(self, scrape_item: ScrapeItem) -> None:
 
     """~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~"""
 
-    async def manage_token(self) -> None:
+    @error_handling_wrapper
+    async def manage_token(self, token_url: URL) -> None:
         """Gets/Sets the redgifs token and header"""
         async with self.request_limiter:
-            json_obj = await self.client.get_json(self.domain, self.redgifs_api / "v2/auth/temporary")
+            json_obj = await self.client.get_json(self.domain, )
         self.token = json_obj["token"]
         self.headers = {"Authorization": f"Bearer {self.token}"}