Skip to content
Snippets Groups Projects
Commit 81b23fd8 authored by Waleed Akbar's avatar Waleed Akbar
Browse files

Minor changes in Generic DB class.

parent c8d99dcb
No related branches found
No related tags found
2 merge requests!294Release TeraFlowSDN 4.0,!267Resolve "(CTTC) Generalize Service Database management"
...@@ -30,7 +30,6 @@ class Database: ...@@ -30,7 +30,6 @@ class Database:
if self.db_engine is None: if self.db_engine is None:
LOGGER.error('Unable to get SQLAlchemy DB Engine...') LOGGER.error('Unable to get SQLAlchemy DB Engine...')
raise Exception('Failed to initialize the database engine.') raise Exception('Failed to initialize the database engine.')
self.db_name = DB_NAME
self.db_model = model self.db_model = model
self.db_table = model.__name__ self.db_table = model.__name__
self.Session = sessionmaker(bind=self.db_engine) self.Session = sessionmaker(bind=self.db_engine)
......
...@@ -25,7 +25,7 @@ class Engine: ...@@ -25,7 +25,7 @@ class Engine:
if crdb_uri is None: if crdb_uri is None:
CRDB_NAMESPACE = get_setting('CRDB_NAMESPACE') CRDB_NAMESPACE = get_setting('CRDB_NAMESPACE')
CRDB_SQL_PORT = get_setting('CRDB_SQL_PORT') CRDB_SQL_PORT = get_setting('CRDB_SQL_PORT')
CRDB_DATABASE = get_setting('CRDB_DATABASE', default=None) CRDB_DATABASE = get_setting('CRDB_DATABASE')
CRDB_USERNAME = get_setting('CRDB_USERNAME') CRDB_USERNAME = get_setting('CRDB_USERNAME')
CRDB_PASSWORD = get_setting('CRDB_PASSWORD') CRDB_PASSWORD = get_setting('CRDB_PASSWORD')
CRDB_SSLMODE = get_setting('CRDB_SSLMODE') CRDB_SSLMODE = get_setting('CRDB_SSLMODE')
......
...@@ -25,7 +25,6 @@ from common.proto.telemetry_frontend_pb2 import CollectorId, Collector, Collecto ...@@ -25,7 +25,6 @@ from common.proto.telemetry_frontend_pb2 import CollectorId, Collector, Collecto
from common.proto.telemetry_frontend_pb2_grpc import TelemetryFrontendServiceServicer from common.proto.telemetry_frontend_pb2_grpc import TelemetryFrontendServiceServicer
from telemetry.database.TelemetryModel import Collector as CollectorModel from telemetry.database.TelemetryModel import Collector as CollectorModel
# from telemetry.database.Telemetry_DB import TelemetryDB
from telemetry.database.Telemetry_DB import TelemetryDB from telemetry.database.Telemetry_DB import TelemetryDB
from confluent_kafka import Consumer as KafkaConsumer from confluent_kafka import Consumer as KafkaConsumer
......
...@@ -25,4 +25,5 @@ def test_verify_databases_and_tables(): ...@@ -25,4 +25,5 @@ def test_verify_databases_and_tables():
# TelemetryDBobj.verify_tables() # TelemetryDBobj.verify_tables()
TelemetryDBobj.create_database() TelemetryDBobj.create_database()
TelemetryDBobj.create_tables() TelemetryDBobj.create_tables()
TelemetryDBobj.verify_tables() TelemetryDBobj.verify_tables()
\ No newline at end of file
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment