Code Coverage

Cobertura Coverage Report > osm_lcm.data_utils.database >

database.py

Trend

File Coverage summary

NameClassesLinesConditionals
database.py
100%
1/1
69%
18/26
100%
0/0

Coverage Breakdown by Class

NameLinesConditionals
database.py
69%
18/26
N/A

Source

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 1 import logging
26 1 from osm_common import dbmemory, dbmongo
27 1 from osm_common.dbbase import DbException
28
29
30 1 class Database:
31 1     class __Database:
32 1         def __init__(self, config):
33 1             self.logger = logging.getLogger("lcm")
34 1             try:
35                 # TODO check database version
36 1                 if config["database"]["driver"] == "mongo":
37 0                     self.db = dbmongo.DbMongo()
38 0                     self.db.db_connect(config["database"])
39 1                 elif config["database"]["driver"] == "memory":
40 1                     self.db = dbmemory.DbMemory()
41 1                     self.db.db_connect(config["database"])
42                 else:
43 0                     raise Exception(
44                         "Invalid configuration param '{}' at '[database]':'driver'".format(
45                             config["database"]["driver"]
46                         )
47                     )
48 0             except DbException as e:
49 0                 self.logger.critical(str(e), exc_info=True)
50 0                 raise Exception(str(e))
51
52 1         def __str__(self):
53 0             return repr(self) + self.db
54
55 1     instance = None
56
57 1     def __init__(self, config=None):
58 1         if not Database.instance:
59 1             Database.instance = Database.__Database(config)
60
61 1     def __getattr__(self, name):
62 0         return getattr(self.instance, name)