Fix updates

akn/AllDownloaderBot/admins.py

@@ -48,15 +48,16 @@ async def arz_update_admin_cache(client, chat_id):
     return admin_list
 
 @Client.on_callback_query(filters.regex(r"^arzunmute_(\d+)_(\d+)$"))
-async def arzunmute_callbackx(client, callback):
+async def xarzunmute_cbok(client, callback):
     chat_id = int(callback.matches[0].group(1))
     user_id = int(callback.matches[0].group(2))
     
     admins = cached_admins.get(chat_id, [])
+    chat_data = await db.alldl_bot.find_one({"chat_ids": chat_id})
+    admins = chat_data.get("chat_admins", []) if chat_data else []
     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 []
-    
+        await callback.answer("⚠️ You can /reload again admins", show_alert=True)
+        return
     if callback.from_user.id not in admins:
         await callback.answer("⚠️ Only admins can unmute!", show_alert=True)
         return
@@ -70,15 +71,14 @@ async def arzunmute_callbackx(client, callback):
         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):
+async def xarzunban_cbok(client, callback):
     chat_id = int(callback.matches[0].group(1))
     user_id = int(callback.matches[0].group(2))
-    
-    admins = cached_admins.get(chat_id, [])
+    chat_data = await db.alldl_bot.find_one({"chat_ids": chat_id})
+    admins = chat_data.get("chat_admins", []) if chat_data else []
     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 []
-    
+        await callback.answer("⚠️ You can /reload again admins", show_alert=True)
+        return
     if callback.from_user.id not in admins:
         await callback.answer("⚠️ Only admins can unmute!", show_alert=True)
         return
@@ -138,7 +138,7 @@ async def arz_reload(client: Client, message: Message):
         return
     
     bot_permissions = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot_permissions.can_restrict_members:
+    if not bot_permissions or 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)
@@ -165,7 +165,7 @@ async def arz_reload(client: Client, message: Message):
 async def arz_ban_user(client: Client, message: Message):
     user_id, reason = await extract_user_and_reason(message, sender_chat=True)
     bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_restrict_members:
+    if not bot or not bot.can_restrict_members:
         return await message.reply_text("I don't have enough permissions")
     if not user_id:
         return await message.reply_text("I can't find that user.")
@@ -283,7 +283,7 @@ async def arz_ban_me(client: Client, message: Message):
 async def arz_mute(client: Client, message: Message):
     user_id, reason = await extract_user_and_reason(message, sender_chat=True)
     bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_restrict_members:
+    if not bot or not bot.can_restrict_members:
         return await message.reply_text("I don't have enough permissions")
     if not user_id:
         return await message.reply_text("I can't find that user.")
@@ -390,7 +390,7 @@ async def arz_mute_all(client: Client, message: Message):
 )
 async def arz_unmute_all(client: Client, message: Message):
     bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_restrict_members:
+    if not bot or not bot.can_restrict_members:
         return await message.reply_text("I don't have enough permissions")
     if message.chat.type == ChatType.PRIVATE:
         return await message.reply_text("This command is not available in private chats.")
@@ -412,7 +412,7 @@ async def arz_unmute_all(client: Client, message: Message):
 async def arz_mute_me(client: Client, message: Message):
     user_id = message.from_user.id
     bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
-    if not bot.can_restrict_members:
+    if not bot or not bot.can_restrict_members:
         return await message.reply_text("I don't have enough permissions")
     reason = None
     if not user_id: