Coverage for toardb/utils/database.py: 100%

1# SPDX-FileCopyrightText: 2021 Forschungszentrum Jülich GmbH 

2# SPDX-License-Identifier: MIT 

3 

4""" 

5Database connection 

6""" 

7 

8from sqlalchemy import create_engine 

9from sqlalchemy.orm import sessionmaker 

10 

11DATABASE_URL = "postgresql://postgres:postgres@postgres:5432/postgres" 

12 

13engine = create_engine(DATABASE_URL) 

14ToarDbSession = sessionmaker(autocommit=False, autoflush=False, bind=engine) 

15 

16 

17# Dependency 

18def get_engine(): # pragma: no cover 

19 assert engine is not None 

20 return engine 

21 

22 

23# Dependency 

24def get_db(): # pragma: no cover 

25 try: 

26 db = ToarDbSession() 

27 yield db 

28 finally: 

29 db.close()