|
5 | 5 | import argparse
|
6 | 6 | import sys
|
7 | 7 |
|
| 8 | + |
8 | 9 | def set_password(args):
|
9 |
| - password = args.password |
10 |
| - while not password : |
11 |
| - password1 = getpass("" if args.quiet else "Provide password: ") |
12 |
| - password_repeat = getpass("" if args.quiet else "Repeat password: ") |
13 |
| - if password1 != password_repeat: |
14 |
| - print("Passwords do not match, try again") |
15 |
| - elif len(password1) < 4: |
16 |
| - print("Please provide at least 4 characters") |
17 |
| - else: |
18 |
| - password = password1 |
| 10 | + password = args.password |
| 11 | + while not password: |
| 12 | + password1 = getpass("" if args.quiet else "Provide password: ") |
| 13 | + password_repeat = getpass("" if args.quiet else "Repeat password: ") |
| 14 | + if password1 != password_repeat: |
| 15 | + print("Passwords do not match, try again") |
| 16 | + elif len(password1) < 4: |
| 17 | + print("Please provide at least 4 characters") |
| 18 | + else: |
| 19 | + password = password1 |
| 20 | + |
| 21 | + password_hash = passwd(password) |
| 22 | + cfg = BaseJSONConfigManager(config_dir=jupyter_config_dir()) |
| 23 | + cfg.update("jupyter_server_config", {"ServerApp": {"password": password_hash,}}) |
| 24 | + if not args.quiet: |
| 25 | + print("password stored in config dir: %s" % jupyter_config_dir()) |
19 | 26 |
|
20 |
| - password_hash = passwd(password) |
21 |
| - cfg = BaseJSONConfigManager(config_dir=jupyter_config_dir()) |
22 |
| - cfg.update('jupyter_server_config', { |
23 |
| - 'ServerApp': { |
24 |
| - 'password': password_hash, |
25 |
| - } |
26 |
| - }) |
27 |
| - if not args.quiet: |
28 |
| - print("password stored in config dir: %s" % jupyter_config_dir()) |
29 | 27 |
|
30 | 28 | def main(argv):
|
31 |
| - parser = argparse.ArgumentParser(argv[0]) |
32 |
| - subparsers = parser.add_subparsers() |
33 |
| - parser_password = subparsers.add_parser('password', help='sets a password for your jupyter server') |
34 |
| - parser_password.add_argument("password", help="password to set, if not given, a password will be queried for (NOTE: this may not be safe)", |
35 |
| - nargs="?") |
36 |
| - parser_password.add_argument("--quiet", help="suppress messages", action="store_true") |
37 |
| - parser_password.set_defaults(function=set_password) |
38 |
| - args = parser.parse_args(argv[1:]) |
39 |
| - args.function(args) |
40 |
| - |
| 29 | + parser = argparse.ArgumentParser(argv[0]) |
| 30 | + subparsers = parser.add_subparsers() |
| 31 | + parser_password = subparsers.add_parser( |
| 32 | + "password", help="sets a password for your jupyter server" |
| 33 | + ) |
| 34 | + parser_password.add_argument( |
| 35 | + "password", |
| 36 | + help="password to set, if not given, a password will be queried for (NOTE: this may not be safe)", |
| 37 | + nargs="?", |
| 38 | + ) |
| 39 | + parser_password.add_argument( |
| 40 | + "--quiet", help="suppress messages", action="store_true" |
| 41 | + ) |
| 42 | + parser_password.set_defaults(function=set_password) |
| 43 | + args = parser.parse_args(argv[1:]) |
| 44 | + args.function(args) |
| 45 | + |
| 46 | + |
41 | 47 | if __name__ == "__main__":
|
42 |
| - main(sys.argv) |
| 48 | + main(sys.argv) |
0 commit comments