From 55a7973b93741fb8d4f19f9aa072ec274f59de8a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Petr=20Dlouh=C3=BD?= <petr.dlouhy@email.cz>
Date: Sun, 4 Dec 2016 01:29:27 +0100
Subject: [PATCH] Python3: fix csvlog

---
 linkcheck/logger/csvlog.py | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/linkcheck/logger/csvlog.py b/linkcheck/logger/csvlog.py
index d91a13d4..13c1bc8e 100644
--- a/linkcheck/logger/csvlog.py
+++ b/linkcheck/logger/csvlog.py
@@ -19,7 +19,10 @@ A CSV logger.
 """
 import csv
 import os
-from io import BytesIO
+try:
+    from cStringIO import StringIO
+except ImportError: # Python 3
+    from io import StringIO
 from . import _Logger
 from .. import strformat
 
@@ -69,7 +72,7 @@ class CSVLogger (_Logger):
         else:
             # write empty string to initialize file output
             self.write(u"")
-        self.queue = BytesIO()
+        self.queue = StringIO()
         self.writer = csv.writer(self.queue, dialect=self.dialect,
                delimiter=self.separator, lineterminator=self.linesep,
                quotechar=self.quotechar)
@@ -119,12 +122,21 @@ class CSVLogger (_Logger):
         self.writerow(map(strformat.unicode_safe, row))
         self.flush()
 
+    def encode_row_s(self, row_s):
+        if isinstance(row_s, str):
+            return row_s  # Python 3
+        else:
+            return row_s.encode("utf-8", self.codec_errors)  # Python 2
+
     def writerow (self, row):
         """Write one row in CSV format."""
-        self.writer.writerow([s.encode("utf-8", self.codec_errors) for s in row])
+        self.writer.writerow([self.encode_row_s(s) for s in row])
         # Fetch UTF-8 output from the queue ...
         data = self.queue.getvalue()
-        data = data.decode("utf-8")
+        try:
+            data = data.decode("utf-8")
+        except AttributeError:
+            pass
         # ... and write to the target stream
         self.write(data)
         # empty queue