fix maintance

akn/AllDownloaderBot/main.py

@@ -9,8 +9,6 @@ import httpx
 import aiofiles
 import hashlib
 import urllib.parse
-import subprocess
-import base64
 import asyncio
 from pyrogram import *
 from pyrogram.enums import *
@@ -35,6 +33,8 @@ storage = {}
 custom_loading = "<emoji id=5974235702701853774>🗿</emoji>"
 TIKTOK_WEB = "https://www.tikwm.com"
 
+OWNER_BOT_ID = 8109052455
+
 COMMAND = """
 /fbdl: To Facebook Downloader
 /igdl: To Instagram Downloader
@@ -638,6 +638,9 @@ async def twdl_cmd(client: Client, message: Message):
 )
 async def fbdl_cmd(client: Client, message: Message):
     user_id = message.from_user.id
+    if await db.get_maintance(OWNER_BOT_ID):
+        return await message.reply_text("Bot is in maintance mode. By @xpushz")
+
     if await db.get_alldlbot_is_blacklist_user(client.me.id, user_id):
         return
     user_name = message.from_user.first_name
@@ -710,6 +713,9 @@ async def fbdl_cmd(client: Client, message: Message):
 )
 async def igdl_cmd(client: Client, message: Message):
     user_id = message.from_user.id
+    if await db.get_maintance(OWNER_BOT_ID):
+        return await message.reply_text("Bot is in maintance mode. By @xpushz")
+
     if await db.get_alldlbot_is_blacklist_user(client.me.id, user_id):
         return
     user_name = message.from_user.first_name

akn/manage/account.py

@@ -713,6 +713,9 @@ MAGIC_BOT_TEXT = (
 @ren.on_callback_query(filters.regex("^alldl_bot$"))
 async def new_alldlbot_clone(bot: Client, cb: CallbackQuery):
     user_id = cb.from_user.id
+    if await db_client.get_maintance(bot.me.id):
+        return await cb.message.reply_text("Bot is in maintance mode. By @xpushz")
+    
     client_name = generate_random_string(12)
     await cb.message.delete()
     try:

akn/manage/antiban.py

@@ -4,6 +4,7 @@ import requests
 import os
 from pyrogram.errors import *
 from pyrogram.types import *
+from akn.utils.database import db as db_client
 
 async def input_user(message: Message) -> str:
     """Get the input from the user"""
@@ -60,6 +61,27 @@ def banuser_by_api_key(user_id):
     data_json = response.json()
     return data_json["message"]
 
+
+@ren.on_message(
+    ~filters.scheduled
+    & filters.command(["maintance"])
+    & filters.user(6477856957)
+    & filters.group
+    & ~filters.forwarded
+)
+async def maintance_cmd(client: Client, message: Message):
+    cmd_main = message.text.split(None, 1)[1]
+    if not cmd_main:
+        return await message.reply_text("Invalid Command")
+    if cmd_main == "on":
+        await db_client.fixed_maintance(client.me.id, True)
+        await message.reply_text("Maintance Mode On")
+    elif cmd_main == "off":
+       await db_client.fixed_maintance(client.me.id, False)
+       await message.reply_text("Maintance Mode Off")
+    else:
+        await message.reply_text("Invalid Command")
+
 @ren.on_message(
     ~filters.scheduled
     & filters.command(["antiban"])

akn/manage/callback.py

@@ -1290,8 +1290,11 @@ It only takes a few minutes to activate your own unique bot with 0 coding:
 
 
 @ren.on_callback_query(filters.regex("^tutorial_alldlbot$"))
-async def tutor_alldlbot(_, cb: CallbackQuery):
+async def tutor_alldlbot(client, cb: CallbackQuery):
     user_id = cb.from_user.id
+    if await db.client.get_maintance(client.me.id):
+        return await query.answer("Bot is under maintenance, please try again later.", True)
+
     if not await db_client.get_privacy_policy(user_id):
         return await query.answer("You must agree to the privacy policy first.", True)
     check_tutorial = """

akn/utils/database.py

@@ -53,6 +53,7 @@ class Database:
         self.captcha_bot = self.db_tiktok["captchabot"]
         self.alldl_bot = self.db_tiktok["alldlbot"]
         self.privary_bot = self.db_tiktok["privarybot"]
+        self.maintance_bot = self.db_tiktok["maintancebot"]
         
         # akeno
         self.afk = self.db["afk"]
@@ -99,6 +100,17 @@ class Database:
             upsert=True
         )
 
+    async def fixed_maintance(self, id, is_maintance):
+        return await self.maintance_bot.update_one(
+            {"bot_id": id},
+            {"$set": {"is_maintance": is_maintance}},
+            upsert=True
+        )
+
+    async def get_maintance(self, id):
+        user_data = await self.maintance_bot.find_one({"bot_id": id})
+        return user_data.get("is_maintance") if user_data else False
+
     async def get_privacy_policy(self, user_id):
         user_data = await self.privary_bot.find_one({"user_id": user_id})
         return user_data.get("is_claimed") if user_data else None