Set fallback series for non-unicode encoding

Rockhopper-Technologies · Mar 13, 2020 · 9329164 · 9329164
1 parent 125631f
commit 9329164
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/enlighten/_counter.py b/enlighten/_counter.py
@@ -12,6 +12,7 @@
 """
 
 import platform
+import sys
 import time
 
 try:
@@ -31,6 +32,14 @@
 else:
     SERIES_STD = u' ▏▎▍▌▋▊▉█'
 
+# Test for non-unicode terminals
+try:
+    SERIES_STD.encode(sys.__stdout__.encoding)
+except UnicodeEncodeError:  # pragma: no cover(Non-unicode Terminal)
+    SERIES_STD = u' |'
+except AttributeError:  # pragma: no cover(Non-standard Terminal)
+    pass
+
 COLORS = ('black', 'red', 'green', 'yellow', 'blue', 'magenta', 'cyan', 'white',
           'bright_black', 'bright_red', 'bright_green', 'bright_yellow',
           'bright_blue', 'bright_magenta', 'bright_cyan', 'bright_white')