MCP Vulnerability Management System

from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker, Session from sqlalchemy.ext.declarative import declarative_base import logging from app.core.config import settings logger = logging.getLogger(__name__) # Create database engine engine = create_engine( settings.DATABASE_URL, pool_pre_ping=True, pool_recycle=3600 ) # Create session factory SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) def get_db() -> Session: """Get a database session""" db = SessionLocal() try: yield db except Exception as e: logger.error(f"Database error: {str(e)}") db.rollback() raise finally: db.close() def init_db(): """Initialize the database""" try: from app.models import Base Base.metadata.create_all(bind=engine) logger.info("Database initialized successfully") except Exception as e: logger.error(f"Error initializing database: {str(e)}") raise