Refactor error logging in database functions for consistency and clarity

This commit is contained in:
JSC
2025-07-25 11:09:50 +02:00
parent d8aaedf851
commit c5c3e8442e

View File

@@ -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