Преглед изворни кода

Merge pull request #143 from alexandrevicenzi/pwd

Check for missing and empty password
Jason Antic пре 6 година
родитељ
комит
62cf29a46f
1 измењених фајлова са 8 додато и 2 уклоњено
  1. 8 2
      set_password.py

+ 8 - 2
set_password.py

@@ -6,10 +6,16 @@ from notebook.auth import passwd
 
 
 def run():
-    password = sys.argv[1:].pop()
+    args = sys.argv[1:]
+
+    if not args:
+        print('Error: Missing password.', file=sys.stderr)
+        return
+
+    password = args[0]
 
     if not password:
-        print('Error: Missing or empty password.', file=sys.stderr)
+        print('Error: Empty password.', file=sys.stderr)
     else:
         encoded = passwd(password)
         print(encoded)