Coverage for dormatory/api/dependencies.py: 56%

1""" 

2Database dependencies for FastAPI. 

3 

4This module provides database session management and dependency injection 

5for the DORMATORY API. 

6""" 

7 

8from typing import Generator 

9from sqlalchemy.orm import Session 

10 

11from dormatory.models.dormatory_model import create_engine_and_session 

12 

13# Create engine and session factory 

14engine, SessionLocal = create_engine_and_session() 

15 

16 

17def get_db() -> Generator[Session, None, None]: 

18 """ 

19 Database dependency for FastAPI. 

20  

21 Yields: 

22 Database session 

23 """ 

24 db = SessionLocal() 

25 try: 

26 yield db 

27 finally: 

28 db.close()