From 6afc37d9a4b9b46d6cfc221c97a10eb06903edb2 Mon Sep 17 00:00:00 2001
From: Richard Mwewa <74001397+rly0nheart@users.noreply.github.com>
Date: Fri, 20 May 2022 16:52:52 +0200
Subject: [PATCH] Update octosuite

---
 octosuite | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/octosuite b/octosuite
index 8acc468..4cb901e 100644
--- a/octosuite
+++ b/octosuite
@@ -1,17 +1,19 @@
 #!/usr/bin/env python3
 
 import logging
-from src.main import *
-from lib.colors import red,white,reset
+from utils.colors import Color
+from utils.misc import Banner
+from core.main import Octosuite, Attributes, logMsg
 
+print(Banner.nameLogo)
 if __name__ == '__main__':
     try:
-        octosuite().on_start()
+        Octosuite().onStart()
        
     except KeyboardInterrupt:
-        logging.warning('Session terminated with (Ctrl+C)')
-        exit(f'\n{white}[{red} x {white}] Session terminated with ({red}Ctrl{white}+{red}C{reset}{white}).{reset}')
+        logging.warning(logMsg.Ctrl)
+        print(Attributes.warning, logMsg.Ctrl)
         
     except Exception as e:
-        logging.error(f'Session terminated on error: {e}')
-        exit(f'\n{white}[{red} ! {white}] Session terminated on error: {red}{e}{reset}')
+        logging.error(logMsg.Error.format(e))
+        print(Attributes.error, logMsg.Exception.format(f'{Color.red}{e}{Color.reset}'))