write to sys.stdout in python 2.X, not sys.stdout.buffer

This commit is contained in:
Jacob Fenton
2016-03-10 09:57:29 -08:00
parent 00e56855b6
commit e06133e49a
+2 -2
View File
@@ -63,8 +63,8 @@ def to_csv(pdf, types, encoding):
]
cols = first_columns + list(sorted(set(fields) - set(first_columns)))
w = unicodecsv.DictWriter(sys.stdout.buffer,
stdout = (sys.stdout.buffer if sys.version_info[0] >= 3 else sys.stdout)
w = unicodecsv.DictWriter(stdout,
fieldnames=cols, encoding=encoding)
w.writeheader()
w.writerows(objs)