Spaces:
Runtime error
Runtime error
File size: 2,040 Bytes
8e1c132 387690c 8e1c132 9fd5b73 83f55da 8e1c132 c2d1089 387690c 8e1c132 6c377a5 |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import asyncio
import os
from contextlib import asynccontextmanager
from threading import Thread
from alembic import command
from alembic.config import Config
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from src.config import DatabaseConfig, logger
from src.controllers import api_router
from src.models import Base
def run_upgrade(connection, alembic_config: Config):
alembic_config.attributes["connection"] = connection
command.upgrade(alembic_config, "head")
async def run_migrations():
logger.info("Running migrations if any...")
alembic_config = Config("alembic.ini")
alembic_config.set_main_option(
"sqlalchemy.url", os.getenv("SQLALCHEMY_DATABASE_URI")
)
async with DatabaseConfig.get_engine().begin() as session:
await session.run_sync(run_upgrade, alembic_config)
@asynccontextmanager
async def lifespan(app: FastAPI):
try:
logger.info("Starting up the application...")
await run_migrations()
logger.info("Application started successfully...")
yield
except Exception as e:
logger.error(f"Error during startup: {str(e)}")
raise
finally:
logger.info("Shutting down the application...")
logger.info("Application shutdown complete.")
app = FastAPI(lifespan=lifespan)
app.add_middleware(
CORSMiddleware,
# allow_origins=os.getenv(
# "CORS_ALLOW_ORIGINS", "http://localhost, http://127.0.0.1"
# ).split(", "),
allow_origins=["*"],
allow_credentials=False,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/")
async def check_health():
return {"response": "Service is healthy!"}
@app.delete("/api/v1/database")
async def delete_all_data():
async with DatabaseConfig.get_engine().begin() as session:
await session.run_sync(Base.metadata.drop_all)
await session.run_sync(Base.metadata.create_all)
return {"response": "All data deleted successfully!"}
app.include_router(api_router, prefix="/api/v1") |