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

Corrected after_pick() logic

Browse Source
This commit is contained in:
Anon 2023-03-12 20:16:14 -07:00
parent eed87f229d
commit 70a9f7457e
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/main.py
View File

@ -158,11 +158,13 @@ class YandereBot(FediBot.YandereBot):
} }
def after_pick(self, picked): def after_pick(self, picked):
self.print_header_stats(picked["picked"]) if picked:
self.currentProfileIndex[self.currentIndexCount] += 1 self.currentProfileIndex[self.currentIndexCount] += 1
self.currentIndexCount += 1 self.currentIndexCount += 1
os.remove(picked["picked"]["full_path"]) self.print_header_stats(picked["picked"])
return super(YandereBot, self).after_pick(picked) os.remove(picked["picked"]["full_path"])
else:
self.print_header_stats(None)
def post(self, picked): def post(self, picked):
# Attempt post # Attempt post