mirror_maker/misskey-boorubot
1
0
Fork 1
mirror of https://github.com/himanshugoel2797/MisskeyBooruImageBot.git synced 2024-10-22 23:19:06 +00:00
Code Issues Projects Releases Packages Wiki Activity

Eliminate need to keep a list of saved images

Browse source
This commit is contained in:
Himanshu Goel 2022-10-28 22:29:31 -04:00
parent 161978a69d
commit f0c45fc288
1 changed files with 10 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
gelbooru_poster.py
View file

@ -30,7 +30,6 @@ class BotInstance:
self.bot_message = config["bot_message"] self.bot_message = config["bot_message"]
self.misskey_url = config["misskey_url"] self.misskey_url = config["misskey_url"]
self.misskey_token = config["misskey_token"] self.misskey_token = config["misskey_token"]
self.saved_images = config["saved_images"]
self.max_page_number = config["max_page_number"] self.max_page_number = config["max_page_number"]
# Get a random image from Gelbooru # Get a random image from Gelbooru
@ -66,7 +65,15 @@ class BotInstance:
# Download and post the image to Misskey # Download and post the image to Misskey
def post_image(self, image_url, image_rating, log_file): def post_image(self, image_url, image_rating, log_file):
if image_url not in self.saved_images: image_found = False
file_presence_check = requests.post(self.misskey_url + "drive/files/find", json = {"name": os.path.split(image_url)[-1], "i": self.misskey_token})
if file_presence_check.status_code != 200:
image_found = False
else:
file_presence_json = file_presence_check.json()
image_found = len(file_presence_json) > 0
if not image_found:
# Submit a /drive/files/upload-from-url request to Misskey # Submit a /drive/files/upload-from-url request to Misskey
upload_from_url_request = requests.post(self.misskey_url + "drive/files/upload-from-url", json = {"url": image_url, "isSensitive": image_rating != 'general', "i": self.misskey_token}) upload_from_url_request = requests.post(self.misskey_url + "drive/files/upload-from-url", json = {"url": image_url, "isSensitive": image_rating != 'general', "i": self.misskey_token})
# If error, print error and exit # If error, print error and exit
@ -115,9 +122,7 @@ class BotInstance:
continue continue
break break
# Download and post the image to Misskey # Download and post the image to Misskey
if self.post_image(image_url, image_rating, log_file): self.post_image(image_url, image_rating, log_file)
# Add the image to the saved image list
self.saved_images.append(image_url)
def generate_config(defaults): def generate_config(defaults):
if os.path.exists("config.json"): if os.path.exists("config.json"):
@ -131,7 +136,6 @@ def generate_config(defaults):
'bot_message': defaults['bot_message'], 'bot_message': defaults['bot_message'],
'misskey_url': defaults['misskey_url'], 'misskey_url': defaults['misskey_url'],
'misskey_token': defaults['misskey_token'], 'misskey_token': defaults['misskey_token'],
'saved_images': [],
'max_page_number': defaults['max_page_number'] 'max_page_number': defaults['max_page_number']
} }
@ -195,7 +199,6 @@ def main():
bot_instance = BotInstance(cfg_name, cfg_tmp) bot_instance = BotInstance(cfg_name, cfg_tmp)
bot_instance.bot_process(log_file) bot_instance.bot_process(log_file)
# Save the saved image list to config.json # Save the saved image list to config.json
config[cfg_name]["saved_images"] = bot_instance.saved_images
config[cfg_name]["max_page_number"] = bot_instance.max_page_number config[cfg_name]["max_page_number"] = bot_instance.max_page_number
# If error, print error and continue # If error, print error and continue