summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorLars Wirzenius <liw@liw.fi>2010-06-05 13:55:42 +1200
committerLars Wirzenius <liw@liw.fi>2010-06-05 13:55:42 +1200
commit2461bdc0a28bd697cd6babb95f3e73d192342bc1 (patch)
tree4fde546f8aaeca6b424845759ecaada3a38eef66
parent20048bb5d257f019454ecbebb5ccda385b11694b (diff)
downloadpython-ttystatus-2461bdc0a28bd697cd6babb95f3e73d192342bc1.tar.gz
Make Messages.finish() to leave last message on screen.
-rw-r--r--ttystatus/messager.py3
-rw-r--r--ttystatus/messager_tests.py2
2 files changed, 3 insertions, 2 deletions
diff --git a/ttystatus/messager.py b/ttystatus/messager.py
index a54f0ce..aa14509 100644
--- a/ttystatus/messager.py
+++ b/ttystatus/messager.py
@@ -72,4 +72,5 @@ class Messager(object):
def finish(self):
'''Finalize output.'''
- self.clear()
+ if self.output.isatty():
+ self.output.write('\n')
diff --git a/ttystatus/messager_tests.py b/ttystatus/messager_tests.py
index c086fc1..8907201 100644
--- a/ttystatus/messager_tests.py
+++ b/ttystatus/messager_tests.py
@@ -77,4 +77,4 @@ class MessagerTests(unittest.TestCase):
def test_finish_removes_message(self):
self.messager.write('foo')
self.messager.finish()
- self.assertEqual(self.output.getvalue(), 'foo\r \r')
+ self.assertEqual(self.output.getvalue(), 'foo\n')