From 4a5dfc7d7640e316ccc1543c7293e12a5830e304 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mike=20F=C3=A4hrmann?= Date: Sun, 1 Dec 2024 18:17:25 +0100 Subject: [PATCH] [rule34] fix 'favorite' extraction (#6573) --- gallery_dl/extractor/gelbooru_v02.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/gallery_dl/extractor/gelbooru_v02.py b/gallery_dl/extractor/gelbooru_v02.py index aad5752c..295a3376 100644 --- a/gallery_dl/extractor/gelbooru_v02.py +++ b/gallery_dl/extractor/gelbooru_v02.py @@ -82,16 +82,17 @@ class GelbooruV02Extractor(booru.BooruExtractor): params["pid"] = self.page_start * self.per_page data = {} - while True: - num_ids = 0 - page = self.request(url, params=params).text + find_ids = re.compile(r"\sid=\"p(\d+)").findall - for data["id"] in text.extract_iter(page, '" id="p', '"'): - num_ids += 1 + while True: + page = self.request(url, params=params).text + pids = find_ids(page) + + for data["id"] in pids: for post in self._api_request(data): yield post.attrib - if num_ids < self.per_page: + if len(pids) < self.per_page: return params["pid"] += self.per_page