File size: 1,019 Bytes
db78256 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
from io import BytesIO
from .. import LOGGER
from ..helper.ext_utils.bot_utils import cmd_exec, new_task
from ..helper.telegram_helper.message_utils import send_message, send_file
@new_task
async def run_shell(_, message):
cmd = message.text.split(maxsplit=1)
if len(cmd) == 1:
await send_message(message, "No command to execute was given.")
return
cmd = cmd[1]
stdout, stderr, _ = await cmd_exec(cmd, shell=True)
reply = ""
if len(stdout) != 0:
reply += f"*Stdout*\n<code>{stdout}</code>\n"
LOGGER.info(f"Shell - {cmd} - {stdout}")
if len(stderr) != 0:
reply += f"*Stderr*\n<code>{stderr}</code>"
LOGGER.error(f"Shell - {cmd} - {stderr}")
if len(reply) > 3000:
with BytesIO(str.encode(reply)) as out_file:
out_file.name = "shell_output.txt"
await send_file(message, out_file)
elif len(reply) != 0:
await send_message(message, reply)
else:
await send_message(message, "No Reply")
|