diff options
Diffstat (limited to 'lib/outputmanager.py')
-rw-r--r-- | lib/outputmanager.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/lib/outputmanager.py b/lib/outputmanager.py new file mode 100644 index 0000000..d9d4af1 --- /dev/null +++ b/lib/outputmanager.py @@ -0,0 +1,41 @@ +#!/usr/bin/python3 +import sys + +class OutputManager: + def __init__(self): + + self.echo_output = True + + def line_break(self): + if self.echo_output: + sys.stdout.write('\n') + sys.stdout.flush() + + return self + + def error(self, string): + if self.echo_output: + self._erase_line() + string = '[-] ' + string + self.last_line_length = len(string) + + sys.stdout.write(string) + sys.stdout.flush() + + return self + + def normal(self, string): + if self.echo_output: + self._erase_line() + self.last_line_length = len(string) + sys.stdout.write(string) + sys.stdout.flush() + + return self + + def _erase_line(self): + sys.stdout.write('\r' + ' ' * self.last_line_length + '\r') + return self + + + |