Data loader

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
+
+        
+