from collections.abc import AsyncGenerator
from typing import Any, TextIO
from app import App
from app.components.http import HTTPComponent
from app.components.lifespan import LifespanComponent
from app.subroutines.http import HTMLResponse
app = App()
lifespan = app.use_component(LifespanComponent())
http = app.use_component(HTTPComponent())
@lifespan.on_context
async def my_context() -> AsyncGenerator[Any, None]:
try:
print("Start!")
yield
finally:
print("Stop!")
lifespan.add_managed_context(open("teapot.log", "w"), name="teapot_log")
@http.route("/teapot", get=True, post=True, put=True, delete=True)
async def teapot() -> HTMLResponse:
resp = """
I'm a Teapot!!!
I'm a Teapot!!!
I've already told you I'm a teapot.
"""
log = lifespan.get_context("teapot_log", TextIO)
_ = log.write("teapot\n")
log.flush()
return HTMLResponse(status=418, content=resp)