mirror of
https://github.com/TeamPGM/PagerMaid-Pyro.git
synced 2024-11-22 05:54:15 +00:00
64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
import io
|
|
import sys
|
|
import traceback
|
|
from typing import Optional
|
|
|
|
from pagermaid import bot
|
|
from pagermaid.services import client as httpx_client
|
|
|
|
|
|
async def run_eval(cmd: str, message=None) -> str:
|
|
old_stderr = sys.stderr
|
|
old_stdout = sys.stdout
|
|
redirected_output = sys.stdout = io.StringIO()
|
|
redirected_error = sys.stderr = io.StringIO()
|
|
stdout, stderr, exc = None, None, None
|
|
try:
|
|
await aexec(cmd, message, bot)
|
|
except Exception: # noqa
|
|
exc = traceback.format_exc()
|
|
stdout = redirected_output.getvalue()
|
|
stderr = redirected_error.getvalue()
|
|
sys.stdout = old_stdout
|
|
sys.stderr = old_stderr
|
|
if exc:
|
|
evaluation = exc
|
|
elif stderr:
|
|
evaluation = stderr
|
|
elif stdout:
|
|
evaluation = stdout
|
|
else:
|
|
evaluation = "Success"
|
|
return evaluation
|
|
|
|
|
|
async def aexec(code, event, client):
|
|
exec(
|
|
(
|
|
(
|
|
("async def __aexec(e, client): " + "\n msg = message = e")
|
|
+ "\n reply = message.reply_to_message if message else None"
|
|
)
|
|
+ "\n chat = e.chat if e else None"
|
|
)
|
|
+ "".join(f"\n {x}" for x in code.split("\n"))
|
|
)
|
|
|
|
return await locals()["__aexec"](event, client)
|
|
|
|
|
|
async def paste_pb(
|
|
content: str, private: bool = True, sunset: int = 3600
|
|
) -> Optional[str]:
|
|
data = {
|
|
"c": content,
|
|
}
|
|
if private:
|
|
data["p"] = "1"
|
|
if sunset:
|
|
data["sunset"] = sunset
|
|
result = await httpx_client.post("https://fars.ee", data=data)
|
|
if result.is_error:
|
|
return None
|
|
return result.headers.get("location")
|