Allow making directories when they exist

selinon · Mar 31, 2022 · c570226 · c570226
1 parent ea3ee5b
commit c570226
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/selinon/storages/filesystem.py b/selinon/storages/filesystem.py
@@ -36,7 +36,7 @@ def is_connected(self):
 
     def connect(self):
         if not os.path.isdir(self.path):
-            os.makedirs(self.path)
+            os.makedirs(self.path, exist_ok=True)
         self._connected = True
 
     def disconnect(self):
@@ -50,7 +50,7 @@ def retrieve(self, flow_name, task_name, task_id):
     def store(self, node_args, flow_name, task_name, task_id, result):  # noqa
         base_path = self._construct_base_path(flow_name, task_name)
         if not os.path.isdir(base_path):
-            os.makedirs(base_path)
+            os.makedirs(base_path, exist_ok=True)
 
         path = self._construct_path(flow_name, task_name, task_id)
         with open(path, 'w') as result_file: