fix update

akn/AllDownloaderBot/admins.py

@@ -1,5 +1,6 @@
 from pyrogram import *
 from pyrogram.types import *
+from pyrogram.enums import *
 from akn.utils.database import db
 
 unmute_permissions = ChatPermissions(
@@ -181,6 +182,11 @@ 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(
@@ -227,6 +233,11 @@ async def arz_mute(client: Client, message: Message):
     if reason:
         msg += f"**Reason:** {reason}"
     await message.chat.restrict_member(user_id, ChatPermissions())
+    await db.alldl_bot.update_one(
+        {"bot_id": client.me.id},
+        {"$addToSet": {"mute_user": user_id}},
+        upsert=True,
+    )
     await message.reply_text(
         msg,
         reply_markup=InlineKeyboardMarkup(
@@ -238,4 +249,64 @@ async def arz_mute(client: Client, message: Message):
                 ]
             ]
         )
-    )
+    )
+
+@Client.on_message(
+    ~filters.scheduled
+    & filters.command(["unmute"])
+    & ~filters.forwarded
+)
+async def arz_unmute(client: Client, message: Message):
+    user_id, reason = await extract_user_and_reason(message, sender_chat=True)
+    if not user_id:
+        return await message.edit_text("I can't find that user.")
+    if user_id == client.me.id:
+        return await message.edit_text("I can't unmute myself.")
+    if user_id == 1191668125:
+        return await message.edit_text("I can't unmute my developer!")
+    try:
+        mention = (await client.get_users(user_id)).mention
+    except IndexError:
+        mention = (
+            message.reply_to_message.sender_chat.title
+            if message.reply_to_message
+            else "Anon"
+        )
+    msg = (
+        f"**Unmuted User:** <a href='tg://user?id={user_id}'>{mention}</a>\n"
+        f"**Unmuted By:** [{message.from_user.mention}](tg://user?id={message.from_user.id if message.from_user else 0})\n"
+    )
+    if reason:
+        msg += f"**Reason:** {reason}"
+    await message.chat.unban_member(user_id)
+    await db.alldl_bot.update_one(
+        {"bot_id": client.me.id},
+        {"$pull": {"mute_user": user_id}},
+        upsert=True,
+    )
+    await message.reply_text(msg)
+
+@Client.on_message(filters.incoming & filters.group, group=-1)
+async def check_user_mute(client: Client, message: Message):
+    if message.from_user and message.from_user.is_bot:
+        return
+    if message.chat.type == ChatType.PRIVATE:
+        return
+    user = message.from_user.id
+    bot = (await client.get_chat_member(message.chat.id, client.me.id)).privileges
+    if not bot.can_restrict_members:
+        return
+    muted_users = await db.alldl_bot.find_one({"bot_id": client.me.id})
+    if muted_users and user in muted_users["mute_user"]:
+        await client.restrict_chat_member(
+            message.chat.id,
+            user,
+            permissions=ChatPermissions(
+                can_send_messages=False,
+                can_send_media_messages=False,
+                can_send_polls=False,
+                can_change_info=False,
+                can_invite_users=False,
+                can_pin_messages=False,
+            ),
+        )

akn/AllDownloaderBot/azhelp.py

@@ -15,6 +15,22 @@ async def arzhelp_command(_, message):
         [
             InlineKeyboardButton("🧹 Clean", callback_data="rhelp_clean"),
             InlineKeyboardButton("🔒 Lock", callback_data="rhelp_lock")
+        ],
+        [
+            InlineKeyboardButton("📥 Downloader", callback_data="rhelp_downloader"),
+        ],
+        [  
+            InlineKeyboardButton("📊 Stats", callback_data="rhelp_stats")
+        ],
+        [
+            InlineKeyboardButton("🔒 Force Subscribe", callback_data="rhelp_forcesub"),
+            InlineKeyboardButton("🛡️ Blacklist", callback_data="rhelp_blacklist")
+        ],
+        [
+            InlineKeyboardButton("⚙️ Bot Settings", callback_data="rhelp_botsettings"),
+        ],
+        [
+            InlineKeyboardButton("❌ Close", callback_data="cclose"),
         ]
     ])