diff --git a/reset_jupyter_password.py b/reset_jupyter_password.py index 71c5343..87aaa82 100644 --- a/reset_jupyter_password.py +++ b/reset_jupyter_password.py @@ -3,29 +3,33 @@ import os from hashlib import sha1 from jinja2 import Environment, FileSystemLoader from random import shuffle +from getpass import getpass username = input("Username: ") port = input("Port: ") servername = input("Servername: ") templates_folder = "templates" -passwords_file = "passwords.txt" +#passwords_file = "passwords.txt" template_env = Environment(loader=FileSystemLoader(templates_folder)) nbconfig = template_env.get_template("jupyter_notebook_config.py") -passwords = open(passwords_file).read().strip().splitlines() -passwords = [w.lower() for w in passwords] -shuffle(passwords) -password = "".join(passwords[:3]) +#passwords = open(passwords_file).read().strip().splitlines() +#passwords = [w.lower() for w in passwords] +#shuffle(passwords) +#password = "".join(passwords[:3]) userdata = {} userdata['username'] = username +userdata['port'] = port +userdata['servername'] = servername + +password = getpass(f"New password: ") + userdata['password'] = password # userdata['password_sha1'] = passwd(password) userdata['password_sha1'] = sha1(password.encode("utf-8")) -userdata['port'] = port -userdata['servername'] = servername with open("jupyter_notebook_config.temp.py", "w") as f: print(nbconfig.render(**userdata), file=f) @@ -34,4 +38,5 @@ with open("jupyter_notebook_config.temp.py", "w") as f: os.system(f"sudo service jupyterlab@{username} restart") -print(f"New password: {password}") \ No newline at end of file +print("Done!") +