Update chatbot/plugins/autobanch.py

chatbot/plugins/autobanch.py

@@ -66,34 +66,11 @@ async def unbanch_usert(client: Client, cb: CallbackQuery):
         await cb.answer(f"Error: {e}", True)
         logger.error(f"Error unbanning user {user_id}: {e}")
 
-@Client.on_message(filters.left_chat_member & filters.channel)
-async def left_ch(client: Client, message: Message):
-    keyboard_button = InlineKeyboardMarkup(
-        [
-            [
-                InlineKeyboardButton(
-                    text="⚠️ Unban",
-                    callback_data=f"unbanch_{message.from_user.id}"
-                )
-            ]
-        ]
+def create_button_unban(user_id):
+    return InlineKeyboardMarkup(
+        [[InlineKeyboardButton(
+            text="⚠️ Unban", callback_data=f"unbanch_{user_id}")]]
     )
-    try:
-        if message.chat.type == ChatType.CHANNEL:
-            if message.chat.id == -1001558215042:
-                text_ban = f"User {message.from_user.first_name} (ID: {message.from_user.id}) was banned from {message.chat.title}."
-                await client.ban_chat_member(
-                    "RendyProjects",
-                    message.from_user.id,
-                )
-                await client.send_message(
-                    "KillerXSupport",
-                    text_ban,
-                    reply_markup=keyboard_button
-                )
-    except Exception as e:
-        await client.send_message("KillerXSupport", f"Error {e}")
-        logger.error(f"Error {e}")
 
 @Client.on_chat_member_updated(
     filters.chat("RendyProjects")
@@ -102,36 +79,25 @@ async def auto_banned_ch(client: Client, event: ChatMemberUpdated):
     logger.info(f"Chat member update: {event}")
     old_status = event.old_chat_member.status if event.old_chat_member else None
     new_status = event.new_chat_member.status if event.new_chat_member else None
-    user_id = event.from_user.id
-    user_first_name = event.from_user.first_name
-    if old_status == ChatMemberStatus.LEFT and new_status != ChatMemberStatus.MEMBER:
-        try:
-            keyboard_button = InlineKeyboardMarkup(
-                [
-                    [
-                        InlineKeyboardButton(
-                            text="⚠️ Unban",
-                            callback_data=f"unbanch_{user_id}"
-                        )
-                    ]
-                ]
-            )
-            text_ban = f"User {user_first_name} (ID: {user_id}) was banned from {event.chat.title}."
-            await client.ban_chat_member(
-                event.chat.id,
-                user_id,
-            )
-            await client.send_message(
-                "KillerXSupport",
-                text_ban,
-                reply_markup=keyboard_button
-            )
-            logger.info(f"User {user_id} has been banned for leaving the chat.")
-        except Exception as e:
-            await client.send_message("KillerXSupport", f"Error banning user {user_id}: {e}")
-            logger.error(f"Error banning user {user_id}: {e}")
+    if event.chat.type == ChatType.CHANNEL:
+        if old_status == ChatMemberStatus.MEMBER:
+            try:
+                text_ban = f"User {event.from_user.first_name} (ID: {event.from_user.id}) was banned from {event.chat.title}."
+                await client.ban_chat_member(
+                    event.chat.id,
+                    event.from_user.id,
+                )
+                await client.send_message(
+                    "KillerXSupport",
+                    text_ban,
+                    reply_markup=create_button_unban(user_id)
+                )
+                logger.info(f"User {user_id} has been banned for leaving the chat.")
+            except Exception as e:
+                await client.send_message("KillerXSupport", f"Error banning user {user_id}: {e}")
+                logger.error(f"Error banning user {user_id}: {e}")
 
-    elif new_status == ChatMemberStatus.MEMBER and event.new_chat_member.status == ChatMemberStatus.MEMBER:
+    if new_status == ChatMemberStatus.MEMBER and event.new_chat_member.status == ChatMemberStatus.MEMBER:
         try:
             if await db.is_gbanned(user_id):
                 text_ban = f"User {user_first_name} (ID: {user_id}) was banned from {event.chat.title} due to global ban."