Update Akeno/utils/scripts.py

Akeno/utils/scripts.py

@@ -37,6 +37,79 @@ from pyrogram.types import Message
 META_COMMENTS = re.compile(r"^ *# *meta +(\S+) *: *(.*?)\s*$", re.MULTILINE)
 interact_with_to_delete = []
 
+def with_args(text: str):
+    def decorator(func):
+        async def wrapped(client: Client, message: Message):
+            if message.text and len(message.text.split()) == 1:
+                await message.edit(text)
+            else:
+                return await func(client, message)
+
+        return wrapped
+
+    return decorator
+
+def get_args(
+    message: Union[Message, str], use_reply: bool = None
+) -> Tuple[List[str], Dict[str, str]]:
+    """Returns list of common args and a dictionary with named args.
+
+    Args:
+        message (Union[Message, str]): Message or text.
+
+        use_reply (bool, optional): Try to get args from reply message if no args in message. Defaults to None.
+
+    Returns:
+        List[str]: List of args.
+    """
+    raw_args = get_args_raw(message, use_reply)
+
+    try:
+        args = list(filter(lambda x: len(x) > 0, shlex.split(raw_args)))
+    except ValueError:
+        return [raw_args], {}
+
+    common_args = []
+    named_args = {}
+
+    i = 0
+    while i < len(args):
+        arg = args[i]
+        if arg.startswith("-"):
+            if i + 1 < len(args) and (
+                not args[i + 1].startswith("-") or len(args[i + 1].split()) > 1
+            ):
+                named_args[arg] = args[i + 1]
+                i += 2
+            else:
+                i += 1
+        else:
+            i += 1
+        common_args.append(arg)
+    return common_args, named_args
+
+def get_args_raw(message: Union[Message, str], use_reply: bool = None) -> str:
+    """Returns text after command.
+
+    Args:
+        message (Union[Message, str]): Message or text.
+
+        use_reply (bool, optional): Try to get args from reply message if no args in message. Defaults to None.
+
+    Returns:
+        str: Text after command or empty string.
+    """
+    if isinstance(message, Message):
+        text = message.text or message.caption
+        args = text.split(maxsplit=1)[1] if len(text.split()) > 1 else ""
+
+        if use_reply and not args:
+            args = message.reply_to_message.text or message.reply_to_message.caption
+
+    elif not isinstance(message, str):
+        return ""
+
+    return args or ""
 
 async def shell_exec(
     command: str,