dotnet
/
MisskeyBooruImageBot_v2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Don't update timestamp if post failed.

This commit is contained in:
dotnet 2023-10-13 18:33:23 -04:00
parent 030c8c7128
commit 952c47f2ea
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
gelbooru_poster.py
View File

@ -228,14 +228,18 @@ class BotInstance:
def bot_process(self, log_file): def bot_process(self, log_file):
# Get a random image making sure it's not in the saved image list # Get a random image making sure it's not in the saved image list
attempts = 10
while True: while True:
if attempts <= 0:
return False
image_url, image_src, image_rating, cur_page_number = self.get_random_image(max_page_number=self.max_page_number) image_url, image_src, image_rating, cur_page_number = self.get_random_image(max_page_number=self.max_page_number)
self.max_page_number = cur_page_number self.max_page_number = cur_page_number
if image_url is None: if image_url is None:
attempts -= 1
continue continue
break break
# Download and post the image to Misskey # Download and post the image to Misskey
self.post_image(image_url, image_src, image_rating, log_file) return self.post_image(image_url, image_src, image_rating, log_file)
def generate_config(defaults): def generate_config(defaults):
if os.path.exists("config.json"): if os.path.exists("config.json"):
@ -318,7 +322,11 @@ def main():
try: try:
bot_instance = BotInstance(cfg_name, cfg_tmp, log_file) bot_instance = BotInstance(cfg_name, cfg_tmp, log_file)
bot_instance.bot_process(log_file) if not bot_instance.bot_process(log_file):
print("\n" + cfg_name + ": Error: ", file=log_file)
print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()), file=log_file)
print("Error running bot", file=log_file)
continue
# Save the saved image list to config.json # Save the saved image list to config.json
config[cfg_name]["max_page_number"] = bot_instance.max_page_number config[cfg_name]["max_page_number"] = bot_instance.max_page_number
# Save the last run time # Save the last run time