Ensure SSH command output is output in correct encoding

- Fixes #72
robotpy · Nov 10, 2023 · 55dc796 · 55dc796
1 parent 8f3909a
commit 55dc796
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/robotpy_installer/sshcontroller.py b/robotpy_installer/sshcontroller.py
@@ -6,6 +6,7 @@
 from os.path import exists, join, expanduser, split as splitpath
 from pathlib import PurePath, PurePosixPath
 import socket
+import sys
 import typing
 
 
@@ -91,7 +92,11 @@ def exec_cmd(
                     if get_output:
                         buffer.write(line)
                     if print_output:
-                        print(line, end="")
+                        try:
+                            print(line, end="")
+                        except UnicodeEncodeError:
+                            eline = line.encode(sys.stdout.encoding, "backslashreplace").decode(sys.stdout.encoding)
+                            print(eline, end="")
 
             retval = channel.recv_exit_status()