使用不支持查询参数的python fast api代理外部网站

Question

我正在尝试使用 python 快速 API 框架代理外部网站（在不同容器上运行的花卉监控 URL）：

client = AsyncClient(base_url=f'http://containername:7800/monitor')

@app.get(“/monitor/{path:path}”)
async def tile_request(path: str):
    req = client.build_request("GET", path)
    r = await client.send(req, stream=True)
    return StreamingResponse(
        r.aiter_raw(),
        background=BackgroundTask(r.aclose),
        headers=r.headers
   )

它能够为每个路径代理容器URL。为了前任。

http://python_server:8001/monitor/dashboard --

超文本传输协议：//python_server： 8001/监视器/任务--

它工作得很好。但是当PATH在URL中有一些查询参数时，它会失败。

为了前任。

http://python_server:8001/monitor/dashboard?json=1&_=1641485992460 --> redirects to http://containername:7800/monitor/dashboard 

（请注意，没有查询参数附加到 URL）。

任何人都可以帮助我们如何使用任何查询参数代理此外部网站的任何路径。

Answer 1

此代码适用于我并在生产中使用：

import httpx
from httpx import AsyncClient
from fastapi import Request
from fastapi.responses import StreamingResponse
from starlette.background import BackgroundTask

app = FastAPI()
HTTP_SERVER = AsyncClient(base_url="http://localhost:8000/")

async def _reverse_proxy(request: Request):
    url = httpx.URL(path=request.url.path, query=request.url.query.encode("utf-8"))
    rp_req = HTTP_SERVER.build_request(
        request.method, url, headers=request.headers.raw, content=await request.body()
    )
    rp_resp = await HTTP_SERVER.send(rp_req, stream=True)
    return StreamingResponse(
        rp_resp.aiter_raw(),
        status_code=rp_resp.status_code,
        headers=rp_resp.headers,
        background=BackgroundTask(rp_resp.aclose),
    )

app.add_route("/{path:path}", _reverse_proxy, ["GET", "POST"])