2024-04-23 19:38:12 +02:00

47 lines
1.5 KiB
Python

from contextlib import asynccontextmanager
from fastapi import FastAPI, Depends
from fastapi_cache import FastAPICache
from fastapi_cache.backends.redis import RedisBackend
from app.api.nonsense import router as nonsense_router
from app.api.shakespeare import router as shakespeare_router
from app.api.stuff import router as stuff_router
from app.utils.logging import AppLogger
from app.api.user import router as user_router
from app.api.health import router as health_router
from app.redis import get_redis, get_cache
from app.services.auth import AuthBearer
logger = AppLogger().get_logger()
@asynccontextmanager
async def lifespan(app: FastAPI):
# Load the redis connection
app.state.redis = await get_redis()
try:
# Initialize the cache with the redis connection
redis_cache = await get_cache()
FastAPICache.init(RedisBackend(redis_cache), prefix="fastapi-cache")
logger.info(FastAPICache.get_cache_status_header())
yield
finally:
# close redis connection and release the resources
await app.state.redis.close()
app = FastAPI(title="Stuff And Nonsense API", version="0.6", lifespan=lifespan)
app.include_router(stuff_router)
app.include_router(nonsense_router)
app.include_router(shakespeare_router)
app.include_router(user_router)
app.include_router(health_router, prefix="/v1/public/health", tags=["Health, Public"])
app.include_router(health_router, prefix="/v1/health", tags=["Health, Bearer"], dependencies=[Depends(AuthBearer())])