FEAT: added adapter for .tsv files

Author: gdementen

larray_editor/arrayadapter.py

@@ -2918,14 +2918,16 @@ def open(cls, fpath):
 
 
 class CsvFileAdapter(TextFileAdapter):
+    DELIMITER = ','
+
     def __init__(self, data, attributes):
         # we know the module is loaded but it is not in the current namespace
         csv = sys.modules['csv']
         TextFileAdapter.__init__(self, data=data, attributes=attributes)
         if self._nbytes > 0:
             first_line = self._get_lines(0, 1)
             assert len(first_line) == 1
-            reader = csv.reader([first_line[0]])
+            reader = csv.reader([first_line[0]], delimiter=self.DELIMITER)
             self._colnames = next(reader)
         else:
             self._colnames = []
@@ -2952,7 +2954,7 @@ def get_values(self, h_start, v_start, h_stop, v_stop):
         # we know the module is loaded but it is not in the current namespace
         csv = sys.modules['csv']
         # Note that csv reader actually needs a line-based input
-        reader = csv.reader(lines)
+        reader = csv.reader(lines, delimiter=self.DELIMITER)
         return [line[h_start:h_stop] for line in reader]
 
 
@@ -2963,6 +2965,17 @@ def open(cls, fpath):
         return open(fpath, 'rt')
 
 
+class TsvFileAdapter(CsvFileAdapter):
+    DELIMITER = '\t'
+
+
+@path_adapter_for('.tsv', 'csv')
+class TsvPathAdapter(TsvFileAdapter):
+    @classmethod
+    def open(cls, fpath):
+        return open(fpath, 'rt')
+
+
 # This is a Path adapter (it handles Path objects) because pyreadstat has no
 # object representing open files
 @path_adapter_for('.sas7bdat', 'pyreadstat')