avoiding 'str doesn't support the buffer api' exception

1 files changed, 4 insertions, 1 deletions

diff --git a/cmd2.py b/cmd2.py
index 3611e60..3216ec9 100755
--- a/cmd2.py
+++ b/cmd2.py
@@ -1438,7 +1438,10 @@ class OutputTrap(Borg):
         self.trap.seek(0)
         result = self.trap.read()
         self.trap.truncate(0)
-        return result.strip('\x00')        
+        try:
+            return result.strip('\x00') #TODO: understand this
+        except TypeError:
+            return result
     def tearDown(self):
         sys.stdout = self.old_stdout