diff --git a/app/core/client/client.py b/app/core/client/client.py index 77cd9c6..0b127f1 100644 --- a/app/core/client/client.py +++ b/app/core/client/client.py @@ -61,9 +61,9 @@ class BOT(Client): @staticmethod def add_cmd(cmd: str | list): def the_decorator(func): + path = inspect.stack()[1][1] @wraps(func) def wrapper(): - path = inspect.stack()[1][1] if isinstance(cmd, list): for _cmd in cmd: Config.CMD_DICT[_cmd] = {"func": func, "path": path} diff --git a/app/plugins/dev/exec.py b/app/plugins/dev/exec.py index 040f877..a1fbe3b 100644 --- a/app/plugins/dev/exec.py +++ b/app/plugins/dev/exec.py @@ -1,4 +1,5 @@ import asyncio +import inspect import sys import traceback from io import StringIO @@ -51,4 +52,4 @@ async def executor(bot: bot, message: Message) -> Message | None: if Config.DEV_MODE: - Config.CMD_DICT["exec"] = executor + Config.CMD_DICT["exec"] = {"func": executor, "path":inspect.stack()[0][1]} diff --git a/app/plugins/dev/loader.py b/app/plugins/dev/loader.py index 5f872be..b2d2194 100644 --- a/app/plugins/dev/loader.py +++ b/app/plugins/dev/loader.py @@ -1,4 +1,5 @@ import importlib +import inspect import sys import traceback @@ -26,4 +27,4 @@ async def loader(bot: bot, message: Message) -> Message | None: if Config.DEV_MODE: - Config.CMD_DICT["load"] = loader + Config.CMD_DICT["load"] ={"func": loader, "path":inspect.stack()[0][1]} diff --git a/app/plugins/dev/shell.py b/app/plugins/dev/shell.py index c19066b..16b912b 100644 --- a/app/plugins/dev/shell.py +++ b/app/plugins/dev/shell.py @@ -1,4 +1,5 @@ import asyncio +import inspect from pyrogram.enums import ParseMode @@ -52,5 +53,5 @@ async def live_shell(bot: bot, message: Message) -> Message | None: if Config.DEV_MODE: - Config.CMD_DICT["shell"] = live_shell - Config.CMD_DICT["sh"] = run_cmd + Config.CMD_DICT["shell"] ={"func": live_shell, "path":inspect.stack()[0][1]} + Config.CMD_DICT["sh"] = {"func": run_cmd, "path":inspect.stack()[0][1]}