1 files changed, 32 insertions, 0 deletions
diff --git a/src/lib/Bcfg2/Reporting/Storage/__init__.py b/src/lib/Bcfg2/Reporting/Storage/__init__.py
new file mode 100644
index 000000000..85356fcfe
--- /dev/null
+++ b/src/lib/Bcfg2/Reporting/Storage/__init__.py
@@ -0,0 +1,32 @@
+"""
+Public storage routines
+"""
+
+import traceback
+
+from Bcfg2.Reporting.Storage.base import StorageError, \
+    StorageImportError
+
+def load_storage(storage_name, setup):
+    """
+    Try to load the storage.  Raise StorageImportError on failure
+    """
+    try:
+        mod_name = "%s.%s" % (__name__, storage_name)
+        mod = getattr(__import__(mod_name).Reporting.Storage, storage_name)
+    except ImportError:
+        try:
+            mod = __import__(storage_name)
+        except:
+            raise StorageImportError("Unavailable")
+    try:
+        cls = getattr(mod, storage_name)
+        return cls(setup)
+    except:
+        raise StorageImportError("Storage unavailable: %s" %
+            traceback.format_exc().splitlines()[-1])
+
+def load_storage_from_config(setup):
+    """Load the storage in the config... eventually"""
+    return load_storage('DjangoORM', setup)
+