Spaces:
Sleeping
Sleeping
File size: 1,663 Bytes
426ff68 10fde30 426ff68 |
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 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
import asyncio
import os
from io import BytesIO
from hydrogram import filters
from hydrogram.types import Message
from bot import TelegramBot
from bot.config import Telegram
MAX_MESSAGE_LENGTH = 4096
@TelegramBot.on_message(filters.command("bash") & filters.user(Telegram.OWNER_ID))
async def execution(client, message):
status_message = await message.reply_text("`Processing ...`")
try:
cmd = message.text.split(" ", maxsplit=1)[1]
except IndexError:
return await status_message.edit("No command provided!")
reply_to_ = message.reply_to_message or message
process = await asyncio.create_subprocess_shell(
cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE
)
stdout, stderr = await process.communicate()
stderr_output = stderr.decode().strip() or "😂"
stdout_output = stdout.decode().strip() or "😐"
output = (
f"<b>QUERY:</b>\n<u>Command:</u>\n<code>{cmd}</code>\n"
f"<u>PID</u>: <code>{process.pid}</code>\n\n"
f"<b>stderr</b>: \n<code>{stderr_output}</code>\n\n"
f"<b>stdout</b>: \n<code>{stdout_output}</code>"
)
if len(output) > MAX_MESSAGE_LENGTH:
with BytesIO(output.encode()) as out_file:
out_file.name = "exec.txt"
await reply_to_.reply_document(
document=out_file,
caption=cmd[: MAX_MESSAGE_LENGTH // 4 - 1],
disable_notification=True,
quote=True,
)
os.remove("exec.txt")
else:
await reply_to_.reply_text(output, quote=True)
await status_message.delete()
|