From 53ccc9899b3ddf5fc71c135db505d03a68a0e243 Mon Sep 17 00:00:00 2001
From: Richard Mwewa <74001397+rly0nheart@users.noreply.github.com>
Date: Sun, 25 Dec 2022 03:02:42 +0200
Subject: [PATCH] Update main.py

---
 octosuite/main.py | 44 +++++++++++++++++++++++---------------------
 1 file changed, 23 insertions(+), 21 deletions(-)

diff --git a/octosuite/main.py b/octosuite/main.py
index b09bd73..f456271 100644
--- a/octosuite/main.py
+++ b/octosuite/main.py
@@ -18,29 +18,31 @@ def octosuite():
                 if args.method == argument:
                     method()
                     print("\n")
-        
-        """
-        Main loop keeps octosuite running, this will break if Octosuite detects a KeyboardInterrupt (Ctrl+C)
-        or if the 'exit' command is entered.
-        """
-        xprint(banner()[0], banner()[1])
-        while True:
-            command_input = Prompt.ask(f"{white}┌──({red}{getpass.getuser()}{white}@{red}octosuite{white})\n├──[~{green}{os.getcwd()}{white}]\n└╼ {reset}")
+                else:
+                    pass
+        else:
             """
-            Iterate over the command_map and check if the user input matches any command in it [command_map],
-            if there's a match, we return its method. If no match is found, we ignore it.
+            Main loop keeps octosuite running, this will break if Octosuite detects a KeyboardInterrupt (Ctrl+C)
+            or if the 'exit' command is entered.
             """
-            if command_input[:2] == 'cd':
-                os.chdir(command_input[3:])
-            elif command_input[:2] == 'ls':
-                os.system(f'dir {command_input[3:]}' if os.name == 'nt' else f'ls {command_input[3:]}')
-            else:
-                for command, method in run.command_map:
-                    if command_input == command:
-                        method()
-                        print("\n")
-                    else:
-                        pass
+            xprint(banner()[0], banner()[1])
+            while True:
+                command_input = Prompt.ask(f"{white}┌──({red}{getpass.getuser()}{white}@{red}octosuite{white})\n├──[~{green}{os.getcwd()}{white}]\n└╼ {reset}")
+                """
+                Iterate over the command_map and check if the user input matches any command in it [command_map],
+                if there's a match, we return its method. If no match is found, we ignore it.
+                """
+                if command_input[:2] == 'cd':
+                    os.chdir(command_input[3:])
+                elif command_input[:2] == 'ls':
+                    os.system(f'dir {command_input[3:]}' if os.name == 'nt' else f'ls {command_input[3:]}')
+                else:
+                    for command, method in run.command_map:
+                        if command_input == command:
+                            method()
+                            print("\n")
+                        else:
+                            pass
         
     except KeyboardInterrupt:
         logging.warning(ctrl_c)