PagerMaid-Pyro/pagermaid/web/__init__.py

from fastapi import FastAPI
from fastapi.responses import HTMLResponse
from starlette.middleware.cors import CORSMiddleware
from starlette.responses import RedirectResponse

from pagermaid.config import Config
from pagermaid.web.api import base_api_router
from pagermaid.web.pages import admin_app, login_page

requestAdaptor = """
requestAdaptor(api) {
    api.headers["token"] = localStorage.getItem("token");
    return api;
},
"""
responseAdaptor = """
responseAdaptor(api, payload, query, request, response) {
    if (response.data.detail == '登录验证失败或已失效，请重新登录') {
        window.location.href = '/login'
        window.localStorage.clear()
        window.sessionStorage.clear()
        window.alert('登录验证失败或已失效，请重新登录')
    }
    return payload
},
"""
icon_path = "https://xtaolabs.com/pagermaid-logo.png"
app: FastAPI = FastAPI()


def init_web():
    app.include_router(base_api_router)

    app.add_middleware(
        CORSMiddleware,
        allow_origins=Config.WEB_ORIGINS,
        allow_credentials=True,
        allow_methods=["*"],
        allow_headers=["*"],
    )

    @app.get("/", response_class=RedirectResponse)
    async def index():
        return "/admin"

    @app.get("/admin", response_class=HTMLResponse)
    async def admin():
        return admin_app.render(
            site_title="PagerMaid-Pyro 后台管理",
            site_icon=icon_path,
            requestAdaptor=requestAdaptor,
            responseAdaptor=responseAdaptor,
        )

    @app.get("/login", response_class=HTMLResponse)
    async def login():
        return login_page.render(
            site_title="登录 | PagerMaid-Pyro 后台管理",
            site_icon=icon_path,
        )