Update Akeno/plugins/admin.py

Akeno/plugins/admin.py

@@ -10,6 +10,7 @@
 import asyncio
 import time
 from time import time as waktu
+from datetime import timedelta
 
 from pyrogram import Client
 from pyrogram import Client as ren
@@ -99,6 +100,67 @@ async def extract_user_and_reason(message, sender_chat=False):
 async def extract_user(message):
     return (await extract_user_and_reason(message))[0]
 
+@app.on_message(filters.command("ban") & filters.me)
+async def test_ban_command(client: Client, message: Message):
+    if not message.reply_to_message and len(message.command) < 2:
+        return await message.reply("Please reply to a message or provide a user ID, along with a reason.")
+
+    if message.reply_to_message:
+        user_id = message.reply_to_message.from_user.id
+    else:
+        try:
+            user_id = int(message.command[1])
+        except ValueError:
+            return await message.reply("Invalid user ID provided.")
+
+    ban_duration = None
+    delete_message = False
+    reason = None
+    if len(message.command) > 2:
+        for arg in message.command:
+            if arg.startswith("h:"):
+                try:
+                    hours = int(arg[2:])
+                    ban_duration = timedelta(hours=hours)
+                except ValueError:
+                    return await message.reply("Invalid hour duration format.")
+            elif arg.startswith("m:"):
+                try:
+                    minutes = int(arg[2:])
+                    ban_duration = timedelta(minutes=minutes)
+                except ValueError:
+                    return await message.reply("Invalid minute duration format.")
+            elif arg.startswith("s:"):
+                try:
+                    seconds = int(arg[2:])
+                    ban_duration = timedelta(seconds=seconds)
+                except ValueError:
+                    return await message.reply("Invalid second duration format.")
+
+        if "{del}" in message.command:
+            delete_message = True
+            message.command.remove("{del}")
+        if len(message.command) > 2:
+            reason = " ".join(message.command[2:])
+
+    try:
+        if ban_duration:
+            until_date = message.date + ban_duration
+            await client.ban_chat_member(message.chat.id, user_id, until_date=until_date)
+        else:
+            await client.ban_chat_member(message.chat.id, user_id)
+        if delete_message:
+            await message.delete()
+            if message.reply_to_message:
+                await message.reply_to_message.delete()
+        response_text = f"User {user_id} has been banned."
+        if ban_duration:
+            response_text += f" Duration: {ban_duration}"
+        if reason:
+            response_text += f" Reason: {reason}"
+        await message.reply(response_text)
+    except Exception as e:
+        await message.reply(f"Failed to ban user: {e}")
 
 async def list_admins(client: Client, chat_id: int):
     global admins_in_chat