forked from s_ranjbar/city_retrofit
68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
"""
|
|
Persistence (Postgresql) configuration
|
|
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
|
Copyright © 2022 Concordia CERC group
|
|
Project Coder Peter Yefi peteryefi@gmail.com
|
|
"""
|
|
|
|
import logging
|
|
import os
|
|
from pathlib import Path
|
|
from dotenv import load_dotenv
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
|
|
Models = declarative_base()
|
|
|
|
|
|
class Configuration:
|
|
"""
|
|
Configuration class to hold common persistence configuration
|
|
"""
|
|
|
|
def __init__(self, db_name: str, dotenv_path: str, app_env='TEST'):
|
|
"""
|
|
:param db_name: database name
|
|
:param app_env: application environment, test or production
|
|
:param dotenv_path: the absolute path to dotenv file
|
|
"""
|
|
try:
|
|
# load environmental variables
|
|
if not Path(dotenv_path).exists():
|
|
error_message = f'dotenv file doesn\'t exists at {dotenv_path}'
|
|
logging.error(error_message)
|
|
raise FileNotFoundError(error_message)
|
|
load_dotenv(dotenv_path=dotenv_path)
|
|
|
|
self._db_name = db_name
|
|
self._db_host = os.getenv(f'{app_env}_DB_HOST')
|
|
self._db_user = os.getenv(f'{app_env}_DB_USER')
|
|
self._db_pass = os.getenv(f'{app_env}_DB_PASSWORD')
|
|
self._db_port = os.getenv(f'{app_env}_DB_PORT')
|
|
self.hub_token = os.getenv('HUB_TOKEN')
|
|
except KeyError as err:
|
|
logging.error('Error with credentials: %s', err)
|
|
|
|
@property
|
|
def connection_string(self):
|
|
"""
|
|
Returns a connection string postgresql
|
|
:return: connection string
|
|
"""
|
|
if self._db_pass:
|
|
return f'postgresql://{self._db_user}:{self._db_pass}@{self._db_host}:{self._db_port}/{self._db_name}'
|
|
return f'postgresql://{self._db_user}@{self._db_host}:{self._db_port}/{self._db_name}'
|
|
|
|
@property
|
|
def db_user(self):
|
|
"""
|
|
retrieve the configured username
|
|
"""
|
|
return self._db_user
|
|
|
|
@property
|
|
def db_name(self):
|
|
"""
|
|
retrieve the configured database name
|
|
"""
|
|
return self._db_name
|