From dc3a45297b6efd621c7e56a6aad282c5fb75f9e0 Mon Sep 17 00:00:00 2001 From: dotnet Date: Fri, 13 Oct 2023 19:05:59 -0400 Subject: [PATCH] Image number selection design fix. --- gelbooru_poster.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/gelbooru_poster.py b/gelbooru_poster.py index 8d3c121..0e28123 100644 --- a/gelbooru_poster.py +++ b/gelbooru_poster.py @@ -55,8 +55,7 @@ class BotInstance: def get_random_image(self, max_page_number = 100): # Get a random page number page_number = random.randint(0, max_page_number) - # Get a random image from the page - image_number = random.randint(0, 100) + # Get the JSON data from the API if self.gelbooru_tags_exclude != "": gelbooru_tags_exclude = " " + self.gelbooru_tags_exclude @@ -70,9 +69,9 @@ class BotInstance: # Make sure there are images on the page if 'post' not in gelbooru_json: return None, None, None, max_pages - # Make sure the image number is valid - if image_number > len(gelbooru_json['post']): - image_number = random.randint(0, len(gelbooru_json['post'])) + + # Choose a random image from the page + image_number = random.randint(0, len(gelbooru_json['post'])) # Save json to file for debugging #with open("gelbooru.json", "w") as gelbooru_json_file: