osm_lcm/data_utils/filesystem/filesystem.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.fsbase import FsException
  27 from osm_common import fslocal, fsmongo
  28
  29
  30 class Filesystem:
  31     class __Filesystem:
  32         def __init__(self, config):
  33             self.logger = logging.getLogger("lcm")
  34             try:
  35                 if config["storage"]["driver"] == "local":
  36                     self.fs = fslocal.FsLocal()
  37                     self.fs.fs_connect(config["storage"])
  38                 elif config["storage"]["driver"] == "mongo":
  39                     self.fs = fsmongo.FsMongo()
  40                     self.fs.fs_connect(config["storage"])
  41                 else:
  42                     raise Exception(
  43                         "Invalid configuration param '{}' at '[storage]':'driver'".format(
  44                             config["storage"]["driver"]
  45                         )
  46                     )
  47             except (FsException) as e:
  48                 self.logger.critical(str(e), exc_info=True)
  49                 raise Exception(str(e))
  50
  51         def __str__(self):
  52             return repr(self) + self.fs
  53
  54     instance = None
  55
  56     def __init__(self, config=None):
  57         if not Filesystem.instance:
  58             Filesystem.instance = Filesystem.__Filesystem(config)
  59
  60     def __getattr__(self, name):
  61         return getattr(self.instance, name)