Update chatbot/plugins/autobanch.py

chatbot/plugins/autobanch.py

@@ -23,6 +23,7 @@ import asyncio
 import io
 import os
 import re
+import logging
 
 from pyrogram import *
 from pyrogram.enums import ChatMemberStatus
@@ -33,60 +34,82 @@ from pyrogram.errors import *
 from database import db
 from logger import LOGS
 
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 @Client.on_callback_query(filters.regex("^unbanch_"))
 async def unbanch_usert(client: Client, cb: CallbackQuery):
     data = cb.data.split("_")
-    user_id = int(data[1])
+    if len(data) != 2:
+        await cb.answer("Invalid data format.", True)
+        return
+    try:
+        user_id = int(data[1])
+    except ValueError:
+        await cb.answer("Invalid user ID.", True)
+        return
     try:
         if cb.from_user.id == 1191668125:
             await client.unban_chat_member(
-                "RendyProjects",
+                "@RendyProjects",
                 user_id,
             )
             await cb.edit_message_text(
-                f"you {user_id} have been unbanned from Rendy Projects",
+                f"User {user_id} has been unbanned from Rendy Projects.",
                 disable_web_page_preview=True,
             )
+            logger.info(f"User {user_id} has been unbanned by admin {cb.from_user.id}.")
         else:
-            await cb.answer("Only Developer", True)
+            await cb.answer("Only the Developer can perform this action.", True)
+            logger.warning(f"Unauthorized unban attempt by user {cb.from_user.id}.")
     except Exception as e:
         await cb.answer(f"Error: {e}", True)
-        
-@Client.on_chat_member_updated(filters.chat("RendyProjects"))
+        logger.error(f"Error unbanning user {user_id}: {e}")
+
+@Client.on_chat_member_updated(filters.chat("@RendyProjects"))
 async def auto_banned_ch(client: Client, event: ChatMemberUpdated):
-    if event.old_chat_member and event.old_chat_member.status == ChatMemberStatus.LEFT:
+    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_{event.from_user.id}"
+                            callback_data=f"unbanch_{user_id}"
                         )
                     ]
                 ]
             )
-            text_ban = f"User {event.from_user.first_name} was banned from {event.chat.title}."
+            text_ban = f"User {user_first_name} (ID: {user_id}) was banned from {event.chat.title}."
             await client.ban_chat_member(
                 event.chat.id,
-                event.from_user.id,
+                user_id,
             )
             await client.send_message(
-                1191668125,
+                "@xtdevs",
                 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(1191668125, str(e))
+            await client.send_message("@zt", f"Error banning user {user_id}: {e}")
+            logger.error(f"Error banning user {user_id}: {e}")
 
-    elif event.new_chat_member and event.new_chat_member.status == ChatMemberStatus.MEMBER: 
+    elif new_status == ChatMemberStatus.MEMBER and event.new_chat_member.status == ChatMemberStatus.MEMBER:
         try:
-            if await db.is_gbanned(event.from_user.id):
-                text_ban = f"User {event.from_user.first_name} was banned from {event.chat.title}."
+            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."
                 await client.ban_chat_member(
                     event.chat.id,
-                    event.from_user.id,
+                    user_id,
                 )
-                await client.send_message(1191668125, text_ban)
+                await client.send_message("@xtdevs", text_ban)
+                logger.info(f"User {user_id} was globally banned and has been banned from the chat.")
         except Exception as e:
-            await client.send_message(1191668125, str(e))
+            await client.send_message("@xtdevs", f"Error banning user {user_id}: {e}")
+            logger.error(f"Error banning user {user_id}: {e}")