Fix golden bug

1 files changed, 9 insertions, 5 deletions

diff --git a/taulubot.py b/taulubot.py
index c0c5c8e..5030e71 100644
--- a/taulubot.py
+++ b/taulubot.py
@@ -39,7 +39,13 @@ def run_taulu(chat_id, user_id):
 	# download to a file with user_id in the name for duplicate recognition
 	filepath = "photo/" + str(abs(user_id)) + ".jpg"
 	finalpath = "photo/" + str(abs(user_id)) + "final.png"
-	if os.path.exists(finalpath):
+
+	# every 100th image has golden borders :)
+	golden_chance = random.randrange(1, 100)
+	logging.warning("Random gold number was: {}".format(golden_chance))
+	is_golden = (golden_chance == 1)
+
+	if os.path.exists(finalpath) and not is_golden:
 		logging.warning("User profile already found. Skipping download and edit...")
 	else:
 		logging.warning("Downloading picture...")
@@ -50,10 +56,8 @@ def run_taulu(chat_id, user_id):
 		im = Image.open(filepath)
 		im.thumbnail(img_size, Image.ANTIALIAS)
 
-		# every 100th image has golden borders :)
-		golden_chance = random.randrange(1, 100)
-		logging.warning("Random gold number was: {}".format(golden_chance))
-		if golden_chance == 1:
+		if is_golden:
+			logging.warning("Golden taulu!")
 			im.paste(golden_im, (0, 0), golden_im)
 			bot.sendMessage(chat_id, "Onneksi olkoon! Kultainen taulu ilmestyy vain kerran tuhannessa vuodessa!")
 		else: