From 18f1e656155ad52ec119d4d9800f9c66f4d8c26d Mon Sep 17 00:00:00 2001
From: Hans van den Heuvel <hans1.vandenheuvel@wur.nl>
Date: Tue, 3 Mar 2020 18:43:55 +0100
Subject: [PATCH] Added possibility to read Tab separated files.

---
 Convert-EUProcessingFactorsDB/mcra.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/Convert-EUProcessingFactorsDB/mcra.py b/Convert-EUProcessingFactorsDB/mcra.py
index 1052b05..e29d4ad 100644
--- a/Convert-EUProcessingFactorsDB/mcra.py
+++ b/Convert-EUProcessingFactorsDB/mcra.py
@@ -179,6 +179,13 @@ class DataSheet:
                     if key not in kwargs:
                         kwargs[key] = value
                 self.sheet = pd.read_csv(self.file.path, **kwargs)
+            elif self.file.extension == '.tsv':
+                # Some defaults:
+                default_args = {'comment': '#', 'dtype': str, 'sep': '\t'}
+                for key, value in default_args.items():
+                    if key not in kwargs:
+                        kwargs[key] = value
+                self.sheet = pd.read_csv(self.file.path, **kwargs)
             elif self.file.extension == '.xlsx':
                 self.sheet = pd.read_excel(self.file.path, **kwargs)
             elif self.file.extension == '.md':
@@ -203,6 +210,17 @@ class DataSheet:
             self.sheet.to_csv(
                 path_or_buf=self.file.path,
                 **kwargs)
+        elif self.file.extension == '.tsv':
+            if self.type is None:
+                self.type = 'pandas'
+            # Some defaults:
+            default_args = {'index': False, 'sep': '\t'}
+            for key, value in default_args.items():
+                if key not in kwargs:
+                    kwargs[key] = value
+            self.sheet.to_csv(
+                path_or_buf=self.file.path,
+                **kwargs)
         elif self.file.extension == '.xlsx':
             if self.type is None:
                 self.type = 'pandas'
-- 
GitLab