瀏覽代碼

Merge pull request #139 from alexandrevicenzi/pwd

Check if an argument is given
Jason Antic 6 年之前
父節點
當前提交
eb71687443
共有 2 個文件被更改,包括 19 次插入4 次删除
  1. 1 1
      run_notebook.sh
  2. 18 3
      set_password.py

+ 1 - 1
run_notebook.sh

@@ -1 +1 @@
-jupyter notebook --port=8888 --no-browser --allow-root --ip=0.0.0.0 --NotebookApp.token="" --NotebookApp.password="$(python set_password.py $NOTEBOOK_PASSWORD)"
+jupyter notebook --port=8888 --no-browser --allow-root --ip=0.0.0.0 --NotebookApp.token="" --NotebookApp.password="$(./set_password.py $NOTEBOOK_PASSWORD)"

+ 18 - 3
set_password.py

@@ -1,4 +1,19 @@
+#!/usr/bin/env python3
+
 import sys
-from IPython.lib import passwd
-password = passwd(sys.argv[1])
-print(password)
+
+from notebook.auth import passwd
+
+
+def run():
+    password = sys.argv[1:].pop()
+
+    if not password:
+        print('Error: Missing or empty password.', file=sys.stderr)
+    else:
+        encoded = passwd(password)
+        print(encoded)
+
+
+if __name__ == '__main__':
+    run()