Update code.py

code.py

@@ -5,7 +5,7 @@ from telethon.tl.types import ChannelParticipantAdmin, ChannelParticipantCreator
 
 api_id = "28810829"
 api_hash = "d3f304bbd0b69b8c30dbec4be5824748"
-bot_token = os.getenv("BOT_TOKEN")
+bot_token = os.getenv("BOT_TOKEN2")
 
 # Define the user ID that is allowed to send commands
 allowed_user_id = [1982395401, 5575183435, 7014359344]  # Replace with the actual user ID
@@ -17,11 +17,15 @@ client = TelegramClient("BBnuke", api_id, api_hash).start(bot_token=bot_token)
 
 async def ban_all_chat_members(chat_id):
     try:
+        ban_count = 0
         async for member in client.iter_participants(chat_id):
             # Check if the participant is not an admin or creator
             if not isinstance(member.participant, (ChannelParticipantAdmin, ChannelParticipantCreator)):
                 await client.edit_permissions(chat_id, member, view_messages=False)
-                #await client.kick_participant(chat_id, member)
+                ban_count += 1
+        msg = f"{ban_count} members were removed from {chat_id}"
+        for i in allowed_user_id:
+            await event.client.send_message(i, msg)
         return True
     except Exception as e:
         logger.error(f"An error occurred: {e}")
@@ -83,15 +87,15 @@ async def purge_handler(event):
 
 @client.on(events.NewMessage)
 async def handle_new_message(event):
-    if event.message.chat.id == -1001823276908:
-        # Delete the message
+    chat_id = event.message.chat.id
+    member = event.message.from_id
+    if chat_id == -1001823276908:
         await event.delete()
-        
-        # Get the user who sent the message
-        member = event.message.from_id
-        
-        # Restrict the user's permissions
-        await client.edit_permissions(event.message.chat.id, member, view_messages=False)
+        if member not in allowed_user_id:
+            try:
+                await client.kick_participant(chat_id, member)
+            except Exception as e:
+                logger.info(f"Exception: {e}")
 
 
 client.run_until_disconnected()