Mkdir non-recursive if recursive mkdir fails

Author: timvaillancourt

mongodb_consistent_backup/State.py

@@ -6,6 +6,7 @@
 from time import time
 
 from mongodb_consistent_backup.Common import Lock
+from mongodb_consistent_backup.Errors import OperationError
 
 
 class StateBase(object):
@@ -22,7 +23,14 @@ def __init__(self, base_dir, config, filename="meta.bson", state_version=1, meta
         self.lock = Lock(self.state_lock, False)
 
         if not os.path.isdir(self.state_dir):
-            os.makedirs(self.state_dir)
+            # try recursive first, fallback to regular mkdir
+            try:
+                os.makedirs(self.state_dir)
+            except:
+                try:
+                    os.mkdir(self.state_dir)
+                except Exception, e:
+                    raise OperationError(e)
 
     def merge(self, new, old):
         merged = old.copy()