73b706f0eb
- Created Alembic migrations for SQLite database - Set up database initialization on app startup - Fixed linting issues with Ruff - Updated README with comprehensive documentation - Configured startup tasks and health checks
69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
import logging
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from app.api.api_v1.api import api_router
|
|
from app.core.config import settings
|
|
from app.db.init_db import init_db
|
|
|
|
# Setup logging
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
app = FastAPI(
|
|
title=settings.PROJECT_NAME,
|
|
description=settings.PROJECT_DESCRIPTION,
|
|
version=settings.VERSION,
|
|
openapi_url="/openapi.json",
|
|
docs_url="/docs",
|
|
redoc_url="/redoc",
|
|
)
|
|
|
|
# Add CORS middleware
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Include API router
|
|
app.include_router(api_router)
|
|
|
|
@app.get("/", tags=["Home"])
|
|
async def root():
|
|
"""
|
|
Root endpoint returning basic information about the API.
|
|
"""
|
|
return {
|
|
"title": settings.PROJECT_NAME,
|
|
"description": settings.PROJECT_DESCRIPTION,
|
|
"version": settings.VERSION,
|
|
"docs_url": "/docs",
|
|
"health_check": "/health",
|
|
}
|
|
|
|
@app.get("/health", tags=["Health"])
|
|
async def health_check():
|
|
"""
|
|
Health check endpoint to verify the service is running.
|
|
"""
|
|
return {"status": "healthy"}
|
|
|
|
# Initialize the database on startup
|
|
@app.on_event("startup")
|
|
async def startup_event():
|
|
"""Run database migrations on startup"""
|
|
logger.info("Running startup tasks")
|
|
try:
|
|
init_db()
|
|
except Exception as e:
|
|
logger.error(f"Error during startup: {str(e)}")
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True) |