diff --git a/app/core/database.py b/app/core/database.py index b682543..0b3fa1d 100644 --- a/app/core/database.py +++ b/app/core/database.py @@ -24,12 +24,13 @@ engine: AsyncEngine = create_async_engine( async def get_db() -> AsyncGenerator[AsyncSession, None]: + """Get a database session for dependency injection.""" logger = get_logger(__name__) async with AsyncSession(engine) as session: try: yield session - except Exception as e: - logger.exception("Database session error: %s", e) + except Exception: + logger.exception("Database session error") await session.rollback() raise finally: @@ -37,12 +38,13 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]: async def init_db() -> None: + """Initialize the database and create tables if they do not exist.""" logger = get_logger(__name__) try: logger.info("Initializing database tables") async with engine.begin() as conn: await conn.run_sync(SQLModel.metadata.create_all) logger.info("Database tables created successfully") - except Exception as e: - logger.exception("Failed to initialize database: %s", e) + except Exception: + logger.exception("Failed to initialize database") raise