feat(sol006): LCM migration to SOL006

[osm/LCM.git] / osm_lcm / data_utils / filesystem / filesystem.py

diff --git a/osm_lcm/data_utils/filesystem/filesystem.py b/osm_lcm/data_utils/filesystem/filesystem.py
new file mode 100644 (file)
index 0000000..bc26c96
--- /dev/null
+++ b/osm_lcm/data_utils/filesystem/filesystem.py
@@ -0,0 +1,58 @@
+# -*- 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)