[pixiv][danbooru] set '_ugoira_original' for ugoira results

mikf · Oct 11, 2024 · f8bb0fd · f8bb0fd
1 parent 5dd1791
commit f8bb0fd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/gallery_dl/extractor/danbooru.py b/gallery_dl/extractor/danbooru.py
@@ -93,6 +93,7 @@ def items(self):
 
             if post["extension"] == "zip":
                 if self.ugoira:
+                    post["_ugoira_original"] = False
                     post["_ugoira_frame_data"] = post["frames"] = \
                         self._ugoira_frames(post)
                     post["_http_adjust_extension"] = False

diff --git a/gallery_dl/extractor/pixiv.py b/gallery_dl/extractor/pixiv.py
@@ -150,6 +150,7 @@ def _extract_ugoira(self, work):
         work["_http_adjust_extension"] = False
 
         if self.load_ugoira == "original":
+            work["_ugoira_original"] = True
             base, sep, _ = url.rpartition("_ugoira")
             base = base.replace("/img-zip-ugoira/", "/img-original/", 1) + sep
 
@@ -173,6 +174,7 @@ def _extract_ugoira(self, work):
                 for num in range(len(frames))
             ]
         else:
+            work["_ugoira_original"] = False
             url = url.replace("_ugoira600x600", "_ugoira1920x1080", 1)
             return ({"url": url},)