osm_lcm/data_utils/database/database.py

   1 # -*- coding: utf-8 -*-
   2
   3 # Copyright 2020 Whitestack, LLC
   4 # *************************************************************
   5 #
   6 # This file is part of OSM LCM module
   7 # All Rights Reserved to Whitestack, LLC
   8 #
   9 # Licensed under the Apache License, Version 2.0 (the "License"); you may
  10 # not use this file except in compliance with the License. You may obtain
  11 # a copy of the License at
  12 #
  13 #         http://www.apache.org/licenses/LICENSE-2.0
  14 #
  15 # Unless required by applicable law or agreed to in writing, software
  16 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  17 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  18 # License for the specific language governing permissions and limitations
  19 # under the License.
  20 #
  21 # For those usages not covered by the Apache License, Version 2.0 please
  22 # contact: fbravo@whitestack.com or agarcia@whitestack.com
  23 ##
  24
  25 import logging
  26 from osm_common import dbmemory, dbmongo
  27 from osm_common.dbbase import DbException
  28
  29
  30 class Database:
  31     class __Database:
  32         def __init__(self, config):
  33             self.logger = logging.getLogger("lcm")
  34             try:
  35                 # TODO check database version
  36                 if config["database"]["driver"] == "mongo":
  37                     self.db = dbmongo.DbMongo()
  38                     self.db.db_connect(config["database"])
  39                 elif config["database"]["driver"] == "memory":
  40                     self.db = dbmemory.DbMemory()
  41                     self.db.db_connect(config["database"])
  42                 else:
  43                     raise Exception(
  44                         "Invalid configuration param '{}' at '[database]':'driver'".format(
  45                             config["database"]["driver"]
  46                         )
  47                     )
  48             except (DbException) as e:
  49                 self.logger.critical(str(e), exc_info=True)
  50                 raise Exception(str(e))
  51
  52         def __str__(self):
  53             return repr(self) + self.db
  54
  55     instance = None
  56
  57     def __init__(self, config=None):
  58         if not Database.instance:
  59             Database.instance = Database.__Database(config)
  60
  61     def __getattr__(self, name):
  62         return getattr(self.instance, name)