diff --git a/.gitignore b/.gitignore index edb9376..e4b477d 100644 --- a/.gitignore +++ b/.gitignore @@ -164,6 +164,4 @@ cython_debug/ # User Files audit_log.txt -config.ini - - +config.ini \ No newline at end of file diff --git a/admin_dashboard.py b/admin_dashboard.py index 1bea334..5c4783e 100644 --- a/admin_dashboard.py +++ b/admin_dashboard.py @@ -1,4 +1,5 @@ import time +import sys from colorama import Fore, Style from common_functions import clear_screen, log_action from user_management import user_management @@ -28,7 +29,7 @@ def admin_dashboard(): print("\nLogging out...") log_action("ADMIN", "Logged Out") time.sleep(2) - exit() + sys.exit() else: print(Fore.RED + "\nInvalid option. Please try again." + Style.RESET_ALL) time.sleep(2) \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 293caac..c8159ac 100644 --- a/requirements.txt +++ b/requirements.txt @@ -109,4 +109,4 @@ Werkzeug==3.0.1 wheel==0.42.0 whichcraft==0.6.1 zope.event==5.0 -zope.interface==6.1 \ No newline at end of file +zope.interface==6.1 diff --git a/sudo_user_login.py b/sudo_user_login.py index 5904ea7..6d1590f 100644 --- a/sudo_user_login.py +++ b/sudo_user_login.py @@ -1,4 +1,5 @@ import time +import sys import getpass from common_functions import clear_screen, verify_password, log_action from colorama import Fore, Style @@ -51,7 +52,7 @@ def attempt_login(self, username, password, attempts): return username else: self.print_insufficient_clearance(username) - exit() + sys.exit() else: self.print_incorrect_password(username, attempts) else: