From 132884506979c46182bfac268c24834be47d7380 Mon Sep 17 00:00:00 2001
From: Anon <yanderefedi@proton.me>
Date: Sun, 19 Mar 2023 19:27:22 -0700
Subject: [PATCH] Updated per backend change

---
 src/main.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/main.py b/src/main.py
index 6d247b4..1f8a5c4 100755
--- a/src/main.py
+++ b/src/main.py
@@ -169,10 +169,12 @@ class YandereBot(FediBot.YandereBot):
 		else:
 			self.print_header_stats(None)
 
-	def post(self):
+	def post(self, callback=None):
+		picked = None
 		# Attempt post
 		try:
-			return super(YandereBot, self).post()
+			picked = self.pick()
+			return super(YandereBot, self).post(lambda: picked)
 		
 		# Invalid post (move to next profile)
 		except FediBot.InvalidPost as e:
@@ -181,7 +183,8 @@ class YandereBot(FediBot.YandereBot):
 		
 		# Invalid post (remove downloaded files)
 		except (FileTooLarge, InvalidMimeType) as e:
-			os.remove(picked["full_path"])
+			if picked is not None:
+				os.remove(picked["full_path"])
 			print("Unable to post:", e)
 
 		# Other Exceptions