Add progress bar exception for systems that do not support unicode (see

CorentinJ#50, CorentinJ#89) (CorentinJ#372)
YiDa-9527 · Jun 23, 2020 · 1453bb8 · 1453bb8
1 parent 32a59f3
commit 1453bb8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/vocoder/display.py b/vocoder/display.py
@@ -13,7 +13,12 @@ def progbar(i, n, size=16):
 
 
 def stream(message) :
-    sys.stdout.write("\r{%s}" % message)
+    try:
+        sys.stdout.write("\r{%s}" % message)
+    except:
+        #Remove non-ASCII characters from message
+        message = ''.join(i for i in message if ord(i)<128)
+        sys.stdout.write("\r{%s}" % message)
 
 
 def simple_table(item_tuples) :