1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
|
from typing import Dict
from litestar import Litestar, MediaType, Response, get
async def after_request(response: Response) -> Response:
if response.media_type == MediaType.TEXT:
return Response({"message": response.content})
return response
@get("/hello")
async def hello() -> str:
return "Hello, world"
@get("/goodbye")
async def goodbye() -> Dict[str, str]:
return {"message": "Goodbye"}
app = Litestar(route_handlers=[hello, goodbye], after_request=after_request)
# run: /hello
# run: /goodbye
|