diff --git a/src/analytics/database/Analyzer_DB.py b/src/analytics/database/Analyzer_DB.py
index 99e9c62f6b4c529e3aa62368247a56c206253a56..ab0b50f2ebba8e2590f1fcb4f2801f42a9c5d208 100644
--- a/src/analytics/database/Analyzer_DB.py
+++ b/src/analytics/database/Analyzer_DB.py
@@ -53,4 +53,4 @@ class AnalyzerDB(Database):
LOGGER.error(f"Error creating filter of {model.__name__} table. ERROR: {e}")
raise OperationFailedException ("CreateKpiDescriptorFilter", extra_details=["unable to create the filter {:}".format(e)])
- return super().select_with_filter(query, session, model)
\ No newline at end of file
+ return super().select_with_filter(query, session, model)
diff --git a/src/analytics/frontend/service/__main__.py b/src/analytics/frontend/service/__main__.py
index cc324e2dc9b4a963d49b9c11f4edfb691b717f35..1df996785ec636592cf5197144d916a89257d9af 100644
--- a/src/analytics/frontend/service/__main__.py
+++ b/src/analytics/frontend/service/__main__.py
@@ -18,10 +18,7 @@ from common.Settings import get_log_level, get_metrics_port
from .AnalyticsFrontendService import AnalyticsFrontendService
from analytics.database.AnalyzerModel import Analyzer as Model
from common.tools.database.GenericDatabase import Database
-from common.Settings import get_setting
-
-DB_NAME = get_setting('CRDB_DATABASE', default=None)
terminate = threading.Event()
LOGGER = None
@@ -42,7 +39,7 @@ def main():
LOGGER.info('Starting...')
# To create DB
- kpiDBobj = Database(DB_NAME, Model)
+ kpiDBobj = Database(Model)
kpiDBobj.create_database()
kpiDBobj.create_tables()
diff --git a/src/analytics/frontend/tests/test_frontend.py b/src/analytics/frontend/tests/test_frontend.py
index f7a25f4c78547e7b1ddb8ff304c1568ca46b79e5..44e84e4683bcdcec72e572b8e4deea903bf0de65 100644
--- a/src/analytics/frontend/tests/test_frontend.py
+++ b/src/analytics/frontend/tests/test_frontend.py
@@ -123,4 +123,4 @@ def test_SelectAnalytics(analyticsFrontend_client):
# class_obj = AnalyticsFrontendServiceServicerImpl()
# for response in class_obj.StartResponseListener(analyzer_id.analyzer_id.uuid):
# LOGGER.debug(response)
-# assert isinstance(response, tuple)
\ No newline at end of file
+# assert isinstance(response, tuple)
diff --git a/src/common/tools/database/GenericDatabase.py b/src/common/tools/database/GenericDatabase.py
index 9868401ec05eabbb220f7c170a2ab66dad09e998..bc73581e30d3e102ed3e40485f8006fb8dedf313 100644
--- a/src/common/tools/database/GenericDatabase.py
+++ b/src/common/tools/database/GenericDatabase.py
@@ -23,11 +23,10 @@ from common.Settings import get_setting
from common.method_wrappers.ServiceExceptions import (OperationFailedException, AlreadyExistsException)
LOGGER = logging.getLogger(__name__)
-DB_NAME = get_setting('CRDB_DATABASE', default=None)
class Database:
def __init__(self, model):
- self.db_engine = Engine.get_engine(DB_NAME)
+ self.db_engine = Engine.get_engine()
if self.db_engine is None:
LOGGER.error('Unable to get SQLAlchemy DB Engine...')
raise Exception('Failed to initialize the database engine.')
@@ -48,7 +47,7 @@ class Database:
def create_tables(self):
try:
self.db_model.metadata.create_all(self.db_engine)
- LOGGER.debug("Tables created in the database: {:}".format(self.db_name))
+ LOGGER.debug("Tables created in the database: {:}".format(self.db_table))
except Exception as e:
LOGGER.debug("Tables cannot be created in the database. {:s}".format(str(e)))
raise OperationFailedException ("Tables can't be created", extra_details=["unable to create table {:}".format(e)])
diff --git a/src/common/tools/database/GenericEngine.py b/src/common/tools/database/GenericEngine.py
index ff3def4669bda96bb248fa71fbe970a963dc2f43..8c458285df560520a40db1ee4f7f37c4f8386f4f 100644
--- a/src/common/tools/database/GenericEngine.py
+++ b/src/common/tools/database/GenericEngine.py
@@ -20,12 +20,12 @@ CRDB_URI_TEMPLATE = 'cockroachdb://{:s}:{:s}@cockroachdb-public.{:s}.svc.cluster
class Engine:
@staticmethod
- def get_engine(db_name) -> sqlalchemy.engine.Engine:
+ def get_engine() -> sqlalchemy.engine.Engine:
crdb_uri = get_setting('CRDB_URI', default=None)
if crdb_uri is None:
CRDB_NAMESPACE = get_setting('CRDB_NAMESPACE')
CRDB_SQL_PORT = get_setting('CRDB_SQL_PORT')
- CRDB_DATABASE = db_name
+ CRDB_DATABASE = get_setting('CRDB_DATABASE', default=None)
CRDB_USERNAME = get_setting('CRDB_USERNAME')
CRDB_PASSWORD = get_setting('CRDB_PASSWORD')
CRDB_SSLMODE = get_setting('CRDB_SSLMODE')
diff --git a/src/kpi_manager/service/__main__.py b/src/kpi_manager/service/__main__.py
index 6a3d078e2d6bb5164fd76ff67ad0ba4f2fba9751..05e32bb58128975ea5d2a5f015d1e8b3977c9905 100644
--- a/src/kpi_manager/service/__main__.py
+++ b/src/kpi_manager/service/__main__.py
@@ -18,10 +18,7 @@ from .KpiManagerService import KpiManagerService
from kpi_manager.database.KpiModel import Kpi as Model
from common.tools.database.GenericDatabase import Database
-from common.Settings import get_setting
-
-DB_NAME = get_setting('CRDB_DATABASE', default=None)
terminate = threading.Event()
LOGGER = None
@@ -42,7 +39,7 @@ def main():
LOGGER.debug('Starting...')
# To create DB
- kpiDBobj = Database(DB_NAME, Model)
+ kpiDBobj = Database(Model)
kpiDBobj.create_database()
kpiDBobj.create_tables()
diff --git a/src/telemetry/frontend/service/__main__.py b/src/telemetry/frontend/service/__main__.py
index c18f7ba4c06c311807746d4a479279fb622f0054..6697ff5f10e58b494736738e631a29a20691732d 100644
--- a/src/telemetry/frontend/service/__main__.py
+++ b/src/telemetry/frontend/service/__main__.py
@@ -18,10 +18,7 @@ from common.Settings import get_log_level, get_metrics_port
from .TelemetryFrontendService import TelemetryFrontendService
from telemetry.database.TelemetryModel import Collector as Model
from common.tools.database.GenericDatabase import Database
-from common.Settings import get_setting
-
-DB_NAME = get_setting('CRDB_DATABASE', default=None)
terminate = threading.Event()
LOGGER = None
@@ -42,7 +39,7 @@ def main():
LOGGER.info('Starting...')
# To create DB
- kpiDBobj = Database(DB_NAME, Model)
+ kpiDBobj = Database(Model)
kpiDBobj.create_database()
kpiDBobj.create_tables()