diff --git a/src/main.py b/src/main.py index fd05442..8d2a517 100755 --- a/src/main.py +++ b/src/main.py @@ -158,11 +158,13 @@ class YandereBot(FediBot.YandereBot): } def after_pick(self, picked): - self.print_header_stats(picked["picked"]) - self.currentProfileIndex[self.currentIndexCount] += 1 - self.currentIndexCount += 1 - os.remove(picked["picked"]["full_path"]) - return super(YandereBot, self).after_pick(picked) + if picked: + self.currentProfileIndex[self.currentIndexCount] += 1 + self.currentIndexCount += 1 + self.print_header_stats(picked["picked"]) + os.remove(picked["picked"]["full_path"]) + else: + self.print_header_stats(None) def post(self, picked): # Attempt post