Writing unicode str in WriteCSV under Py2.7

maet3608 · Aug 31, 2020 · b07bac1 · b07bac1
1 parent c7666e1
commit b07bac1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nutsflow/sink.py b/nutsflow/sink.py
@@ -620,6 +620,7 @@ def __rrshift__(self, iterable):
         for row in iterable:
             row = row if is_iterable(row) else [row]
             row = [row[i] for i in cols] if cols else row
-            self.writer.writerow([self.fmtfunc(r) for r in row])
+            row = [six.u(self.fmtfunc(r)) for r in row]
+            self.writer.writerow(row)
             if self.flush:
                 self.csvfile.flush()