Corrected after_pick() logic

2023-03-12 20:16:14 -07:00 · 2023-03-12 20:16:14 -07:00 · 70a9f7457e
commit 70a9f7457e
parent eed87f229d
1 changed files with 7 additions and 5 deletions
--- 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"])
+		if picked:
 			self.currentProfileIndex[self.currentIndexCount] += 1
 			self.currentIndexCount += 1
+			self.print_header_stats(picked["picked"])
 			os.remove(picked["picked"]["full_path"])
-		return super(YandereBot, self).after_pick(picked)
+		else:
+			self.print_header_stats(None)

 	def post(self, picked):
 		# Attempt post