forked from s_ranjbar/city_retrofit
26 lines
773 B
Python
26 lines
773 B
Python
"""
|
|
Base repository class to establish db connection
|
|
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
|
Copyright © 2022 Concordia CERC group
|
|
Project Coder Peter Yefi peteryefi@gmail.com
|
|
"""
|
|
import logging
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import Session
|
|
from hub.persistence.configuration import Configuration
|
|
|
|
|
|
|
|
class Repository:
|
|
"""
|
|
Base repository class to establish db connection
|
|
"""
|
|
|
|
def __init__(self, db_name, dotenv_path: str, app_env='TEST'):
|
|
try:
|
|
self.configuration = Configuration(db_name, dotenv_path, app_env)
|
|
self.engine = create_engine(self.configuration.connection_string)
|
|
self.session = Session(self.engine)
|
|
except ValueError as err:
|
|
logging.error('Missing value for credentials: %s', err)
|