PagerMaid-Pyro/pagermaid/web/api/bot_info.py

import os
import signal

from fastapi import APIRouter
from fastapi.responses import JSONResponse

from pagermaid.web.api.utils import authentication
from pagermaid.common.update import update

route = APIRouter()


@route.post("/bot_update", response_class=JSONResponse, dependencies=[authentication()])
async def bot_update():
    await update()
    return {"status": 0, "msg": "更新成功，请重启 PagerMaid-Pyro 以应用更新。"}


@route.post(
    "/bot_restart", response_class=JSONResponse, dependencies=[authentication()]
)
async def bot_restart():
    os.kill(os.getpid(), signal.SIGINT)
    return {}