Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -940,43 +940,44 @@ def create_keyboard(
 async def vote(client, callback_query):
     message_id = callback_query.message.id
     user_id = callback_query.from_user.id
-    action = callback.matches[0].group(1)
-    if message_id not in like_counts:
-        like_counts[message_id] = set()
-    if message_id not in unlike_counts:
-        unlike_counts[message_id] = set()
+    action = callback_query.matches[0].group(1)
+
+    like_counts.setdefault(message_id, set())
+    unlike_counts.setdefault(message_id, set())
+
+    user_bl = await db.user_blacklists.find_one({"user_id": user_id})
+    if user_bl and user_bl.get("is_frozen", False):
+        await callback_query.answer("Sorry, your account is frozen.", show_alert=True)
+        return
 
     if action == "like":
-        user_bl = await db.user_blacklists.find_one({"user_id": user_id})
-        if user_bl and user_bl.get("is_frozen", False):
-            await callback_query.answer("Sorry your account is disabled", True)
-            return
         if user_id in like_counts[message_id]:
             like_counts[message_id].remove(user_id)
         else:
             like_counts[message_id].add(user_id)
-            await callback_query.answer("Thank you for your feedback!", False)
-            if user_id in unlike_counts[message_id]:
-                unlike_counts[message_id].remove(user_id)
+            unlike_counts[message_id].discard(user_id)
+            await callback_query.answer("Thank you for your feedback!", show_alert=False)
 
     elif action == "unlike":
-        user_bl = await db.user_blacklists.find_one({"user_id": user_id})
-        if user_bl and user_bl.get("is_frozen", False):
-            await callback_query.answer("Sorry your account is disabled", True)
-            return
         if user_id in unlike_counts[message_id]:
             unlike_counts[message_id].remove(user_id)
         else:
             unlike_counts[message_id].add(user_id)
-            if user_id in like_counts[message_id]:
-                like_counts[message_id].remove(user_id)
+            like_counts[message_id].discard(user_id)
+            await callback_query.answer("Got it, thank you!", show_alert=False)
 
     likes = len(like_counts[message_id])
     unlikes = len(unlike_counts[message_id])
     updated_keyboard = create_keyboard(likes=likes, unlikes=unlikes)
-    await callback_query.edit_message_reply_markup(reply_markup=updated_keyboard)
+
+    try:
+        await callback_query.edit_message_reply_markup(reply_markup=updated_keyboard)
+    except MessageNotModified:
+        pass
+
     await callback_query.answer()
 
+
 @Client.on_message(
     ~filters.scheduled
     & filters.command(["start"])