From 70a9f7457e6d355693d2955fc541dc436dfade05 Mon Sep 17 00:00:00 2001 From: Anon Date: Sun, 12 Mar 2023 20:16:14 -0700 Subject: [PATCH] Corrected after_pick() logic --- src/main.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) 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