""" Generate router """ import logging from contextlib import asynccontextmanager from fastapi import APIRouter from api.schema import SuccessDetail from .schema import RequirementsRequest @asynccontextmanager async def lifespan(app: APIRouter): logger = logging.getLogger("uvicorn") logger.info("Starting requirements router...") yield logger.info("Shutting down requirements router...") router = APIRouter( lifespan=lifespan, ) @router.get( "/", status_code=200, response_model=SuccessDetail, ) async def home(): """ Requirements home page. """ return {"success": "Welcome to the requirements submodule!"} @router.post( "/create", status_code=200, response_model=SuccessDetail, ) async def create_requirements(requirements: RequirementsRequest): """ Create requirements. """ return {"success": "Requirements created successfully!"}