#474 and #459 create automatically output dir

nicolay-r · May 17, 2023 · dab4f69 · dab4f69
1 parent 6af32de
commit dab4f69
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/arekit/contrib/utils/data/writers/csv_native.py b/arekit/contrib/utils/data/writers/csv_native.py
@@ -1,4 +1,7 @@
 import csv
+import os
+from os.path import dirname
+
 from arekit.common.data.storages.base import BaseRowsStorage
 from arekit.contrib.utils.data.storages.row_cache import RowCacheStorage
 from arekit.contrib.utils.data.writers.base import BaseWriter
@@ -13,6 +16,7 @@ def __init__(self, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL):
             f, delimiter=delimiter, quotechar=quotechar, quoting=quoting)
 
     def open_target(self, target):
+        os.makedirs(dirname(target), exist_ok=True)
         self.__target_f = open(target, "w")
         self.__writer = self.__create_writer_func(self.__target_f)
         pass

diff --git a/arekit/contrib/utils/data/writers/json_opennre.py b/arekit/contrib/utils/data/writers/json_opennre.py
@@ -1,6 +1,7 @@
 import json
 import logging
 import os
+from os.path import dirname
 
 from arekit.common.data import const
 from arekit.common.data.storages.base import BaseRowsStorage
@@ -58,6 +59,7 @@ def __format_row(row, text_columns):
         }
 
     def open_target(self, target):
+        os.makedirs(dirname(target), exist_ok=True)
         self.__target_f = open(target, "w")
         pass