Refactor error logging in database functions for consistency and clarity
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user