Files
URL-shortener/app/main.py
T

36 lines
776 B
Python

# app/main.py
import logging
from fastapi import FastAPI
from contextlib import asynccontextmanager
from app.core.config import settings
from app.db.session import engine
from app.db.base import Base
from app.api.routes import router
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - %(levelname)s - %(name)s - %(message)s",
)
import os
from contextlib import asynccontextmanager
@asynccontextmanager
async def lifespan(app: FastAPI):
if os.getenv("TESTING") != "1":
Base.metadata.create_all(bind=engine)
yield
def create_app() -> FastAPI:
app = FastAPI(
title=settings.app_name,
debug=settings.debug,
)
app.include_router(router)
return app
app = create_app()
app.router.lifespan_context = lifespan