Data loader

4 files changed, 95 insertions, 0 deletions

diff --git a/lib/commandmanager.py b/lib/commandmanager.py
new file mode 100644
index 0000000..3a06f99
--- /dev/null
+++ b/lib/commandmanager.py
@@ -0,0 +1,37 @@
+from lib.exceptions import CmdNotFoundException
+
+class Command:
+    def __init__(self):
+        pass
+
+    def execute(self, loader, params):
+        pass
+
+    def usage(self, cmd_name):
+        return cmd_name
+
+class AdminCommand(Command):
+    pass
+
+class NetworkCommand(Command):
+    pass
+
+class DomainCommand(Command):
+    pass
+
+class CommandManager:
+    def __init__(self):
+        self.cmds = {
+            'admin'  : AdminCommand(),
+            'network': NetworkCommand(),
+            'domain' : DomainCommand(),
+            }
+        
+    def find(self, cmd):
+        if cmd in self.cmds:
+            return self.cmds[cmd]
+        else:
+            raise CmdNotFoundException(cmd + 'is not a valid command.')
+
+    def commands(self):
+        return self.cmds.keys()
diff --git a/lib/completionmanager.py b/lib/completionmanager.py
new file mode 100644
index 0000000..fe6f099
--- /dev/null
+++ b/lib/completionmanager.py
@@ -0,0 +1,9 @@
+class CompletionManager:
+    def __init__(self):
+        pass
+
+    def generate_commands(self):
+        pass
+
+    def generate_parameters(self):
+        pass
diff --git a/lib/exceptions.py b/lib/exceptions.py
new file mode 100644
index 0000000..63c7475
--- /dev/null
+++ b/lib/exceptions.py
@@ -0,0 +1,8 @@
+class CmdNotFoundException(Exception):
+    pass
+
+class CommandException(Exception):
+    def __init__(self, message, print_usage=True):
+        Exception.__init__(self, message)
+        self.print_usage = print_usage
+    
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
+
+        
+