XindeX 习语录¶

from fastapi import FastAPI, Response, Request, Query
from fastapi.responses import ORJSONResponse, StreamingResponse
from brotli_asgi import BrotliMiddleware
from httpx import AsyncClient
from bs4 import BeautifulSoup
from asyncache import cached
from cachetools import TTLCache

app = FastAPI(default_response_class=ORJSONResponse)
app.add_middleware(BrotliMiddleware, quality=3)

client = AsyncClient(http2=True, follow_redirects=True,
                     base_url="http://jhsjk.people.cn",
                     headers={"Accept-Encoding": "gzip", "User-Agent": ""})

@app.get("/search")
@cached(TTLCache(15, 60))  # 1 minute caching, max 15 items
async def search(fuzzy: bool = False, keywords="", sort_by: SortBy = ...,
                 num: int | None = Query(None, example=10, ge=1), reverse: bool = False):
    result = await find_all(fuzzy, keywords)
    if sort_by is not SortBy.no_sort:
        result.sort(key=itemgetter(sort_by.value), reverse=reverse)
    return result if num is None else result[:num]

@app.get("/image/{article_id}")
async def fetch_image(article_id: int = Query(alias="articleId")):
    if url := await get_image(article_id):
        response = await client.get(url)
        return StreamingResponse(response.iter_bytes(8192),
                                 media_type=response.headers["Content-Type"])