mirror of
https://github.com/TeamPGM/PagerMaid-Pyro.git
synced 2024-11-22 11:55:37 +00:00
65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
import asyncio
|
|
from typing import Union, Optional
|
|
|
|
from fastapi import APIRouter, Header
|
|
from fastapi.responses import JSONResponse, StreamingResponse
|
|
|
|
from pagermaid.common.status import get_status
|
|
from pagermaid.common.system import run_eval
|
|
from pagermaid.config import Config
|
|
from pagermaid.utils import execute
|
|
from pagermaid.web.api.utils import authentication
|
|
|
|
route = APIRouter()
|
|
|
|
|
|
@route.get("/log")
|
|
async def get_log(token: Optional[str] = Header(...), num: Union[int, str] = 100):
|
|
if token != Config.WEB_SECRET_KEY:
|
|
return "非法请求"
|
|
try:
|
|
num = int(num)
|
|
except ValueError:
|
|
num = 100
|
|
|
|
async def streaming_logs():
|
|
with open("pagermaid.log.txt", "r", encoding="utf-8") as f:
|
|
for log in f.readlines()[-num:]:
|
|
yield log
|
|
await asyncio.sleep(0.02)
|
|
|
|
return StreamingResponse(streaming_logs())
|
|
|
|
|
|
@route.get("/run_eval")
|
|
async def run_cmd(token: Optional[str] = Header(...), cmd: str = ""):
|
|
if token != Config.WEB_SECRET_KEY:
|
|
return "非法请求"
|
|
|
|
async def run_cmd_func():
|
|
result = (await run_eval(cmd, only_result=True)).split("\n")
|
|
for i in result:
|
|
yield i + "\n"
|
|
await asyncio.sleep(0.02)
|
|
|
|
return StreamingResponse(run_cmd_func()) if cmd else "无效命令"
|
|
|
|
|
|
@route.get("/run_sh")
|
|
async def run_sh(token: Optional[str] = Header(...), cmd: str = ""):
|
|
if token != Config.WEB_SECRET_KEY:
|
|
return "非法请求"
|
|
|
|
async def run_sh_func():
|
|
result = (await execute(cmd)).split("\n")
|
|
for i in result:
|
|
yield i + "\n"
|
|
await asyncio.sleep(0.02)
|
|
|
|
return StreamingResponse(run_sh_func()) if cmd else "无效命令"
|
|
|
|
|
|
@route.get("/status", response_class=JSONResponse, dependencies=[authentication()])
|
|
async def status():
|
|
return (await get_status()).dict()
|