Corrected after_pick() logic

This commit is contained in:
Anon 2023-03-12 20:16:14 -07:00
parent eed87f229d
commit 70a9f7457e

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
self.print_header_stats(picked["picked"])
os.remove(picked["picked"]["full_path"]) os.remove(picked["picked"]["full_path"])
return super(YandereBot, self).after_pick(picked) else:
self.print_header_stats(None)
def post(self, picked): def post(self, picked):
# Attempt post # Attempt post