Fix: Python3 str/bytes confusion

2 files changed, 5 insertions, 2 deletions

diff --git a/ttystatus/area.py b/ttystatus/area.py
index 370f422..2bea1c6 100644
--- a/ttystatus/area.py
+++ b/ttystatus/area.py
@@ -77,10 +77,11 @@ class AreaManager(object):
             if i > 0:
                 parts.append(down)
             parts.append(cr)
-            parts.append(line[:max_chars])
+            parts.append(line[:max_chars].encode())
             parts.append(erase)
 
-        self._terminal.write(''.join(parts))
+        output = b''.join(parts)
+        self._terminal.write(output)
 
     def clear_area(self, num_lines):
         '''Clear area reserved for message needing a given number of lines.
diff --git a/ttystatus/tty.py b/ttystatus/tty.py
index 83561b6..d3df316 100644
--- a/ttystatus/tty.py
+++ b/ttystatus/tty.py
@@ -95,6 +95,8 @@ class PhysicalTerminal(object):
 
         '''
 
+        if not isinstance(raw_data, bytes):
+            raw_data = raw_data.encode()
         if self._terminal is not None:
             try:
                 self._terminal.write(raw_data)