From 96bb45f11b085ebcc52c19782bc24450652c91b8 Mon Sep 17 00:00:00 2001 From: dotnet Date: Fri, 13 Oct 2023 18:05:36 -0400 Subject: [PATCH] Limit page count to 200 as per gelbooru API limits --- gelbooru_poster.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/gelbooru_poster.py b/gelbooru_poster.py index 511d138..b4f160b 100644 --- a/gelbooru_poster.py +++ b/gelbooru_poster.py @@ -62,6 +62,8 @@ class BotInstance: gelbooru_tags_exclude = "" gelbooru_json = requests.get(self.gelbooru_url + urllib.parse.quote_plus(self.gelbooru_tags + gelbooru_tags_exclude) + "&pid=" + str(page_number)).json() max_pages = gelbooru_json['@attributes']['count'] // 100 + (1 if gelbooru_json['@attributes']['count'] % 100 != 0 else 0) + if max_pages > 200: + max_pages = 200 # Make sure there are images on the page if 'post' not in gelbooru_json: return None, None, None, max_pages