Coverage for app / database.py: 75%

1import sqlalchemy as _sql 

2from sqlalchemy.orm import declarative_base 

3import sqlalchemy.orm as _orm 

4from sqlalchemy import create_engine 

5import os 

6from dotenv import load_dotenv 

7 

8# Load .env if present 

9load_dotenv() 

10 

11DATABASE_URL = os.getenv("DATABASE_URL") 

12 

13engine = create_engine(DATABASE_URL) 

14 

15SessionLocal = _orm.sessionmaker(autocommit=False, autoflush=False, bind=engine) 

16 

17Base = declarative_base() 

18 

19 

20def get_db(): 

21 db = SessionLocal() 

22 try: 

23 yield db 

24 finally: 

25 db.close()