[hitomi] fix error when number of tag results is multiple of 25

(#2870)
mikf · Aug 28, 2022 · 2eb0ddd · 2eb0ddd
1 parent 3cebf78
commit 2eb0ddd
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/gallery_dl/extractor/hitomi.py b/gallery_dl/extractor/hitomi.py
@@ -174,20 +174,24 @@ def items(self):
         }
 
         offset = 0
+        total = None
         while True:
             headers["Referer"] = "{}/{}/{}.html?page={}".format(
                 self.root, self.type, self.tag, offset // 100 + 1)
             headers["Range"] = "bytes={}-{}".format(offset, offset+99)
-            nozomi = self.request(nozomi_url, headers=headers).content
+            response = self.request(nozomi_url, headers=headers)
 
-            for gallery_id in decode_nozomi(nozomi):
+            for gallery_id in decode_nozomi(response.content):
                 gallery_url = "{}/galleries/{}.html".format(
                     self.root, gallery_id)
                 yield Message.Queue, gallery_url, data
 
-            if len(nozomi) < 100:
-                return
             offset += 100
+            if total is None:
+                total = text.parse_int(
+                    response.headers["content-range"].rpartition("/")[2])
+            if offset >= total:
+                return
 
 
 @memcache(maxage=1800)