osm_lcm/data_utils/filesystem/filesystem.py

# -*- coding: utf-8 -*-

# Copyright 2020 Whitestack, LLC
# *************************************************************
#
# This file is part of OSM LCM module
# All Rights Reserved to Whitestack, LLC
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
#
# For those usages not covered by the Apache License, Version 2.0 please
# contact: fbravo@whitestack.com or agarcia@whitestack.com
##

import logging
from osm_common.fsbase import FsException
from osm_common import fslocal, fsmongo


class Filesystem:
    class __Filesystem:
        def __init__(self, config):
            self.logger = logging.getLogger("lcm")
            try:
                if config["storage"]["driver"] == "local":
                    self.fs = fslocal.FsLocal()
                    self.fs.fs_connect(config["storage"])
                elif config["storage"]["driver"] == "mongo":
                    self.fs = fsmongo.FsMongo()
                    self.fs.fs_connect(config["storage"])
                else:
                    raise Exception(
                        "Invalid configuration param '{}' at '[storage]':'driver'".format(
                            config["storage"]["driver"]
                        )
                    )
            except FsException as e:
                self.logger.critical(str(e), exc_info=True)
                raise Exception(str(e))

        def __str__(self):
            return repr(self) + self.fs

    instance = None

    def __init__(self, config=None):
        if not Filesystem.instance:
            Filesystem.instance = Filesystem.__Filesystem(config)

    def __getattr__(self, name):
        return getattr(self.instance, name)