소스 검색

Check for missing and empty password

In #139 only empty password was being caught.

This patch includes the case where password might
not even be set.
Alexandre Vicenzi 6 년 전
부모
커밋
6f6b622a5c
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)