|
|
import time |
|
|
from datetime import datetime as dt |
|
|
from akn.manage.parameter import generate_random_string |
|
|
from pyrogram import Client as ren |
|
|
from pyrogram import * |
|
|
from pyrogram.types import * |
|
|
from akn.utils.database import db |
|
|
from akn.utils.logger import LOGS |
|
|
from config import * |
|
|
|
|
|
def create_bot_controls(bot_uuid_str, restart_count=0): |
|
|
return InlineKeyboardMarkup([ |
|
|
[ |
|
|
InlineKeyboardButton("🔄 Restart", callback_data=f"restartdl_{bot_uuid_str}"), |
|
|
InlineKeyboardButton("🛑 Stop", callback_data=f"stopbotdl_{bot_uuid_str}") |
|
|
], |
|
|
[ |
|
|
InlineKeyboardButton("📊 Stats", callback_data=f"botstatsdl_{bot_uuid_str}"), |
|
|
InlineKeyboardButton("⚙️ Settings", callback_data=f"botsettingsdl_{bot_uuid_str}") |
|
|
], |
|
|
[ |
|
|
InlineKeyboardButton("🔙 Back", callback_data="my_bots") |
|
|
] |
|
|
]) |
|
|
|
|
|
@ren.on_callback_query(filters.regex(r"^stopbotdl_(\w+)$")) |
|
|
async def handle_stop_botx(client: Client, callback: CallbackQuery): |
|
|
try: |
|
|
bots_uuid = callback.matches[0].group(1) |
|
|
user_id = callback.from_user.id |
|
|
|
|
|
bot_data = await db.alldl_bot.find_one( |
|
|
{"user_id": user_id, "bots.uuid": bots_uuid}, |
|
|
{"bots.$": 1} |
|
|
) |
|
|
|
|
|
if not bot_data or not bot_data.get("bots"): |
|
|
return await callback.answer("❌ Bot not found!", show_alert=True) |
|
|
|
|
|
bot = bot_data["bots"][0] |
|
|
|
|
|
if not bot.get("is_active"): |
|
|
return await callback.answer("⚠️ Bot is already stopped.", show_alert=True) |
|
|
|
|
|
await db.alldl_bot.update_one( |
|
|
{"user_id": user_id, "bots.uuid": bots_uuid}, |
|
|
{"$set": { |
|
|
"bots.$.is_active": False, |
|
|
"bots.$.stopped_at": dt.now().strftime("%Y-%m-%d %H:%M:%S") |
|
|
}} |
|
|
) |
|
|
|
|
|
await callback.message.edit_reply_markup( |
|
|
create_bot_controls(bots_uuid, restart_count=bot.get("restart_count", 0)) |
|
|
) |
|
|
await callback.answer("🛑 Bot has been 24 hours after stopped successfully!") |
|
|
|
|
|
except Exception as e: |
|
|
LOGS.error(f"Stop handler error for UUID {bots_uuid}: {str(e)}") |
|
|
await callback.answer("⚠️ Failed to stop the bot!", show_alert=True) |
|
|
|
|
|
@ren.on_callback_query(filters.regex(r"^(restartdl)_(\w+)$")) |
|
|
async def handle_restart_botx(client: Client, callback: CallbackQuery): |
|
|
try: |
|
|
_, bots_uuid = callback.matches[0].groups() |
|
|
user_id = callback.from_user.id |
|
|
|
|
|
bot_data = await db.alldl_bot.find_one( |
|
|
{"user_id": user_id, "bots.uuid": bots_uuid}, |
|
|
{"bots.$": 1} |
|
|
) |
|
|
|
|
|
if not bot_data or not bot_data.get("bots"): |
|
|
return await callback.answer("❌ Bot not found!", show_alert=True) |
|
|
|
|
|
bot = bot_data["bots"][0] |
|
|
|
|
|
if not bot.get("is_active"): |
|
|
return await callback.answer("❌ Bot is not active!", show_alert=True) |
|
|
|
|
|
if bot.get("status") != "approved": |
|
|
return await callback.answer("❌ Only approved bots can be restarted!", show_alert=True) |
|
|
|
|
|
try: |
|
|
client_name = generate_random_string(12) |
|
|
new_client = Client( |
|
|
name=client_name, |
|
|
api_id=API_ID, |
|
|
api_hash=API_HASH, |
|
|
bot_token=bot["bot_token"], |
|
|
plugins={"root": "akn.AllDownloaderBot"} |
|
|
) |
|
|
await new_client.start() |
|
|
await db.alldl_bot.update_one( |
|
|
{"user_id": user_id, "bots.uuid": bots_uuid}, |
|
|
{"$set": { |
|
|
"bots.$.is_active": True, |
|
|
"bots.$.last_restart": dt.now().strftime("%Y-%m-%d %H:%M:%S"), |
|
|
"bots.$.restart_count": bot.get("restart_count", 0) + 1 |
|
|
}} |
|
|
) |
|
|
|
|
|
await callback.message.edit_reply_markup( |
|
|
create_bot_controls(bots_uuid, restart_count=bot.get("restart_count", 0) + 1) |
|
|
) |
|
|
await callback.answer("✅ Bot restarted successfully!") |
|
|
except Exception as e: |
|
|
LOGS.error(f"Restart error for UUID {bots_uuid}: {str(e)}") |
|
|
await callback.answer("❌ Failed to restart the bot!", show_alert=True) |
|
|
|
|
|
except Exception as e: |
|
|
LOGS.error(f"Handler restart_botx error: {str(e)}") |
|
|
await callback.answer("⚠️ error!", show_alert=True) |
|
|
|
