Anon/DanbooruBot
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated per backend change

Browse Source
This commit is contained in:
Anon 2023-03-19 19:27:22 -07:00
parent 208f897501
commit 1328845069
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/main.py
View File

@ -169,10 +169,12 @@ class YandereBot(FediBot.YandereBot):
else: else:
self.print_header_stats(None) self.print_header_stats(None)
def post(self): def post(self, callback=None):
picked = None
# Attempt post # Attempt post
try: try:
return super(YandereBot, self).post() picked = self.pick()
return super(YandereBot, self).post(lambda: picked)
# Invalid post (move to next profile) # Invalid post (move to next profile)
except FediBot.InvalidPost as e: except FediBot.InvalidPost as e:
@ -181,6 +183,7 @@ class YandereBot(FediBot.YandereBot):
# Invalid post (remove downloaded files) # Invalid post (remove downloaded files)
except (FileTooLarge, InvalidMimeType) as e: except (FileTooLarge, InvalidMimeType) as e:
if picked is not None:
os.remove(picked["full_path"]) os.remove(picked["full_path"])
print("Unable to post:", e) print("Unable to post:", e)