#462 fix

arekit/contrib/utils/data/ext.py

@@ -1,19 +1,24 @@
 from arekit.contrib.utils.data.readers.base import BaseReader
 from arekit.contrib.utils.data.readers.csv_pd import PandasCsvReader
 from arekit.contrib.utils.data.writers.base import BaseWriter
+from arekit.contrib.utils.data.writers.csv_native import NativeCsvWriter
 from arekit.contrib.utils.data.writers.csv_pd import PandasCsvWriter
 from arekit.contrib.utils.data.writers.json_opennre import OpenNREJsonWriter
 
+
 PANDAS_CSV_EXTENSION = ".tsv.gz"
+OPENNRE_EXTENSION = ".jsonl"
 
 
 def create_writer_extension(writer):
     assert(isinstance(writer, BaseWriter))
 
     if isinstance(writer, OpenNREJsonWriter):
-        return ".json"
+        return OPENNRE_EXTENSION
     if isinstance(writer, PandasCsvWriter):
         return PANDAS_CSV_EXTENSION
+    if isinstance(writer, NativeCsvWriter):
+        return ".csv"
 
     raise NotImplementedError()
 
@@ -23,5 +28,7 @@ def create_reader_extension(writer):
 
     if isinstance(writer, PandasCsvReader):
         return PANDAS_CSV_EXTENSION
+    if isinstance(writer, OpenNREJsonWriter):
+        return OPENNRE_EXTENSION
 
     raise NotImplementedError()

arekit/contrib/utils/data/readers/jsonl.py

@@ -10,6 +10,3 @@ def read(self, target):
             for line in f.readlines():
                 rows.append(line)
         return JsonlBasedRowsStorage(rows)
-
-    def target_extension(self):
-        return ".jsonl"