chore: user-updated dev.py

DragMusic/plugins/tools/dev.py

@@ -10,27 +10,30 @@ from time import time
 from pyrogram import filters
 from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message, CallbackQuery
 
-from DragMusic import app
-from config import OWNER_ID, BANNED_USERS
-from DragMusic.utils.database import is_banned_user, remove_banned_user, is_gbanned_user, remove_gban_user
-from DragMusic.misc import SUDOERS
-from DragMusic.core.mongo import mongodb
-from DragMusic.utils.decorators.admins import AdminActual
-from DragMusic.utils.pastebin import paste
-
+from AnonXMusic import app
+from config import OWNER_ID
+from AnonXMusic.core.mongo import mongodb
 
 async def aexec(code, client, message):
+    afkdb = mongodb.afk
+
+    # Create a local execution environment
+    local_vars = {
+        "client": client,
+        "message": message,
+        "afkdb": afkdb,
+        "get_afk_users": lambda: afkdb.find({"user_id": {"$gt": 0}}).to_list(1000),
+    }
+
+    # Prepare the async wrapper function
     exec(
-        "async def __aexec(client, message, mongodb):\n"
-        + "    try:\n"
-        + "        p = print\n"
-        + "        reply = message.reply_to_message\n"
-        + "        chat = message.chat.id\n"
-        + "".join(f"        {l}\\n" for l in code.split("\n"))
-        + "    except Exception as e:\n"
-        + "        print(traceback.format_exc())"
+        f"async def __aexec(client, message):\n"
+        + "".join(f"    {l}\n" for l in code.split("\n")),
+        globals(),
+        local_vars,
     )
-    return await locals()["__aexec"](client, message, mongodb)
+
+    return await local_vars["__aexec"](client, message)
 
 
 async def edit_or_reply(msg: Message, **kwargs):
@@ -38,25 +41,6 @@ async def edit_or_reply(msg: Message, **kwargs):
     await func(**kwargs)
 
 
-@app.on_message(
-    filters.command(["eval", "e"]) & SUDOERS & ~filters.forwarded & ~filters.via_bot
-)
-async def executor(client, message: Message):
-    if len(message.command) < 2:
-        return await message.reply_text("<b>» ᴡʜᴀᴛ ʏᴏᴜ ᴡᴀɴɴᴀ ᴇxᴇᴄᴜᴛᴇ ?</b>")
-    try:
-        cmd = message.text.split(" ", maxsplit=1)[1]
-    except IndexError:
-        return await message.delete()
-    t1 = time.time()
-    try:
-        await aexec(cmd, client, message)
-    except Exception as e:
-        await message.reply_text(e)
-    t2 = time.time()
-    await message.reply_text(f"<b>» ᴛᴏᴏᴋ {round(t2 - t1, 3)} sᴇᴄᴏɴᴅs ᴛᴏ ᴇxᴇᴄᴜᴛᴇ.</b>")
-
-
 @app.on_message(filters.text & filters.user(OWNER_ID) & ~filters.forwarded & ~filters.via_bot)
 async def executor(client: app, message: Message):
     """Executes Python or shell commands based on plain text input."""
@@ -143,28 +127,3 @@ async def close_command(_, CallbackQuery: CallbackQuery):
     
     await CallbackQuery.message.delete()
     await CallbackQuery.answer()
-
-
-@app.on_message(filters.command("checkban") & filters.user(OWNER_ID))
-async def check_ban_status(client, message):
-    """Check if owner is banned and unban if needed."""
-    user_id = message.from_user.id
-    
-    # Check if user is banned locally
-    is_banned = await is_banned_user(user_id)
-    # Check if user is globally banned
-    is_gbanned = await is_gbanned_user(user_id)
-    
-    if is_banned:
-        await remove_banned_user(user_id)
-        await message.reply_text("✅ You were locally banned. You have been unbanned.")
-    elif is_gbanned:
-        await remove_gban_user(user_id)
-        await message.reply_text("✅ You were globally banned. You have been unbanned.")
-    else:
-        await message.reply_text("✅ You are not banned.")
-    
-    # Also remove from BANNED_USERS set if present
-    if user_id in BANNED_USERS:
-        BANNED_USERS.remove(user_id)
-        await message.reply_text("✅ Removed from BANNED_USERS set.")