Exit gracefully when run as non-root

Patch for pull request #33
2017-05-18 01:58:08 -04:00
parent df7167a945
commit 433d30861d
2 changed files with 2 additions and 3 deletions
--- a/Wifite.py
+++ b/Wifite.py
@@ -23,8 +23,7 @@ class Wifite(object):
        if os.getuid() != 0:
            Color.pl('{!} {R}error: {O}wifite{R} must be run as {O}root{W}')
            Color.pl('{!} {O}re-run as: sudo ./Wifite.py{W}')
-            exit(0)
-            return
+            Configuration.exit_gracefully(0)

        Configuration.initialize(load_interface=False)