!fixup python3 compatibility

tartley · Jun 16, 2018 · 9c703ca · 9c703ca
1 parent 6486095
commit 9c703ca
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/colorama/ansitowin32.py b/colorama/ansitowin32.py
@@ -12,6 +12,8 @@
 if windll is not None:
     winterm = WinTerm()
 
+is_python2 = sys.version_info.major == 2
+
 
 def is_stream_closed(stream):
     return not hasattr(stream, 'closed') or stream.closed
@@ -173,7 +175,7 @@ def write_and_convert(self, text):
 
     def write_plain_text(self, text, start, end):
         if start < end:
-            if isinstance(text, unicode):
+            if is_python2 and isinstance(text, unicode):
                 self.wrapped.write(text[start:end].encode('ascii', 'ignore'))
             else:
                 self.wrapped.write(text[start:end])