Fix

akn/AllDownloaderBot/admins.py

@@ -1,3 +1,4 @@
+from datetime import datetime as dt
 from pyrogram import *
 from pyrogram.types import *
 from pyrogram.enums import *
@@ -36,6 +37,59 @@ async def extract_userid(message, text: str):
         return entity.user.id
     return None
 
+cached_admins = {}
+
+async def arz_update_admin_cache(client, chat_id):
+    admin_list = []
+    async for m in client.get_chat_members(chat_id, filter=ChatMembersFilter.ADMINISTRATORS):
+        if not m.user.is_bot and not m.user.is_deleted:
+            admin_list.append(m.user.id)
+    cached_admins[chat_id] = admin_list
+    return admin_list
+
+@Client.on_callback_query(filters.regex(r"^arzunmute_(\d+)_(\d+)$"))
+async def arzunmute_callbackx(client, callback):
+    chat_id = int(callback.matches[0].group(1))
+    user_id = int(callback.matches[0].group(2))
+    
+    admins = cached_admins.get(chat_id, [])
+    if not admins:
+        chat_data = await db.alldl_bot.find_one({"chat_ids": chat_id})
+        admins = chat_data.get("chat_admins", []) if chat_data else []
+    
+    if callback.from_user.id not in admins:
+        await callback.answer("⚠️ Only admins can unmute!", show_alert=True)
+        return
+    try:
+        await client.restrict_chat_member(chat_id, user_id, unmute_permissions)
+        await callback.message.edit_text(
+            f"✅ User {user_id} unmuted by @{callback.from_user.username}",
+            reply_markup=None
+        )
+    except Exception as e:
+        await callback.message.edit_text(f"❌ Failed: {str(e)}")
+
+@Client.on_callback_query(filters.regex(r"^arzunban_(\d+)_(\d+)$"))
+async def arzunban_callbackx(client, callback):
+    chat_id = int(callback.matches[0].group(1))
+    user_id = int(callback.matches[0].group(2))
+    
+    admins = cached_admins.get(chat_id, [])
+    if not admins:
+        chat_data = await db.alldl_bot.find_one({"chat_ids": chat_id})
+        admins = chat_data.get("chat_admins", []) if chat_data else []
+    
+    if callback.from_user.id not in admins:
+        await callback.answer("⚠️ Only admins can unmute!", show_alert=True)
+        return
+    try:
+        await client.unban_chat_member(chat_id, user_id)
+        await callback.message.edit_text(
+            f"✅ User {user_id} unbanned by @{callback.from_user.username}",
+            reply_markup=None
+        )
+    except Exception as e:
+        await callback.message.edit_text(f"❌ Failed: {str(e)}")
 
 async def extract_user_and_reason(message, sender_chat=False):
     args = message.text.strip().split()
@@ -76,6 +130,32 @@ async def extract_user_and_reason(message, sender_chat=False):
 async def extract_user(message):
     return (await extract_user_and_reason(message))[0]
 
+@Client.on_message(
+    ~filters.scheduled & filters.command(["reload"]) & ~filters.forwarded
+)
+async def arz_reload(client: Client, message: Message):
+    if not message.from_user:
+        return
+    
+    bot_permissions = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
+    if not bot_permissions.can_restrict_members:
+        return await message.reply_text("❌ I need admin permissions to restrict members")
+
+    admins = await arz_update_admin_cache(client, message.chat.id)
+    await db.alldl_bot.update_one(
+        {"bot_id": client.me.id},
+        {
+            "$addToSet": {"chat_ids": message.chat.id},
+            "$set": {
+                "chat_admins": admins,
+                "last_updated": dt.now()
+            }
+        },
+        upsert=True
+    )
+    await message.reply_text(
+        "🔄 Admin list reloaded successfully!",
+    )
 
 @Client.on_message(
     ~filters.scheduled
@@ -182,18 +262,13 @@ async def arz_ban_me(client: Client, message: Message):
     if reason:
         msg += f"**Reason:** {reason}"
     await message.chat.ban_member(user_id)
-    await db.alldl_bot.update_one(
-        {"bot_id": client.me.id},
-        {"$set": {"ban_user": user_id}},
-        upsert=True,
-    )
     await message.reply_text(
         msg,
         reply_markup=InlineKeyboardMarkup(
             [
                 [
                     InlineKeyboardButton(
-                        "Unban", callback_data=f"arzunban_{user_id}"
+                        "⚫ Unban", callback_data=f"arzunban_{message.chat.id}_{user_id}"
                     )
                 ]
             ]
@@ -244,7 +319,7 @@ async def arz_mute(client: Client, message: Message):
             [
                 [
                     InlineKeyboardButton(
-                        "Unmute", callback_data=f"arzunmute_{user_id}"
+                        "⚫ Unmute", callback_data=f"arzunmute_{message.chat.id}_{user_id}"
                     )
                 ]
             ]
@@ -278,6 +353,7 @@ async def arz_unmute(client: Client, message: Message):
     )
     if reason:
         msg += f"**Reason:** {reason}"
+    await message.chat.restrict_member(user_id, unmute_permissions)
     await message.chat.unban_member(user_id)
     await db.alldl_bot.update_one(
         {"bot_id": client.me.id},