create data directory for reporting Transports

1 files changed, 14 insertions, 2 deletions

diff --git a/src/lib/Bcfg2/Reporting/Transport/base.py b/src/lib/Bcfg2/Reporting/Transport/base.py
index cca7beda0..ea2098b52 100644
--- a/src/lib/Bcfg2/Reporting/Transport/base.py
+++ b/src/lib/Bcfg2/Reporting/Transport/base.py
@@ -2,17 +2,21 @@
 The base for all server -> collector Transports
 """
 
-import os.path
-import logging 
+import os
+import sys
+import logging
+
 
 class TransportError(Exception):
     """Generic TransportError"""
     pass
 
+
 class TransportImportError(TransportError):
     """Raised when a transport fails to import"""
     pass
 
+
 class TransportBase(object):
     """The base for all transports"""
 
@@ -22,6 +26,14 @@ class TransportBase(object):
         self.logger = logging.getLogger(clsname)
         self.logger.debug("Loading %s transport" % clsname)
         self.data = os.path.join(setup['repo'], 'Reporting', clsname)
+        if not os.path.exists(self.data):
+            self.logger.info("%s does not exist, creating" % self.data)
+            try:
+                os.makedirs(self.data)
+            except OSError:
+                self.logger.warning("Could not create %s: %s" %
+                                    (self.data, sys.exc_info()[1]))
+                self.logger.warning("The transport may not function properly")
         self.setup = setup
         self.timeout = 2