Updated all

akn/AllDownloaderBot/admins.py

@@ -71,6 +71,11 @@ async def xarzunmute_cbok(client, callback):
             return
         user_mention = callback.from_user.mention
         await client.restrict_chat_member(chat_id, user_id, unmute_permissions)
+        await db.alldl_bot.update_one(
+            {"bot_id": client.me.id},
+            {"$pull": {"mute_user": user_id}},
+            upsert=True
+        )
         await callback.message.edit_text(
             f"✅ User {user_id} unmuted by {user_mention}",
             reply_markup=None
@@ -228,6 +233,9 @@ async def arz_ban_user(client: Client, message: Message):
 )
 async def arz_unban_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 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.")
     if user_id == client.me.id:
@@ -351,6 +359,9 @@ async def arz_mute(client: Client, message: Message):
 )
 async def arz_unmute(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 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.")
     if user_id == client.me.id:
@@ -376,7 +387,7 @@ async def arz_unmute(client: Client, message: Message):
     await db.alldl_bot.update_one(
         {"bot_id": client.me.id},
         {"$pull": {"mute_user": user_id}},
-        upsert=True,
+        upsert=True
     )
     await message.reply_text(msg)
 
@@ -387,7 +398,7 @@ async def arz_unmute(client: Client, message: Message):
 )
 async def arz_mute_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.")