jonigata/PoseMaker2

1from fastapi import FastAPI, Request, Response2 3filenames = ["js/poseMaker.js"]4contents = '\n'.join([f"<script type='text/javascript' src='{x}'></script>" for x in filenames])5 6app = FastAPI()7 8@app.middleware("http")9async def insert_js(request: Request, call_next):10    path = request.scope['path']  # get the request route11    response = await call_next(request)12    13    if path == "/":14        response_body = ""15        async for chunk in response.body_iterator:16            response_body += chunk.decode()17 18        response_body = response_body.replace("</body>", contents + "</body>")19 20        del response.headers["content-length"]21 22        return Response(23            content=response_body,24            status_code=response.status_code, 25            headers=dict(response.headers),26            media_type=response.media_type27        )28 29    return response