aravagarwal/CodeCloakPII

1def set_password(args):2	password = args.password3	while not password  :4		password1 = getpass("" if args.quiet else "Provide password: ")5		password_repeat = getpass("" if args.quiet else "Repeat password:  ")6		if password1 != password_repeat:7			print("Passwords do not match, try again")8		elif len(password1) < 4:9			print("Please provide at least 4 characters")10		else:11			password = password112 13	password_hash = passwd(password)14	cfg = BaseJSONConfigManager(config_dir=jupyter_config_dir())15	cfg.update('jupyter_notebook_config', {16		'NotebookApp': {17			'password': password_hash,18		}19	})20	if not args.quiet:21		print("password stored in config dir: %s" % jupyter_config_dir())22 23def main(argv):24	parser = argparse.ArgumentParser(argv[0])25	subparsers = parser.add_subparsers()26	parser_password = subparsers.add_parser('password', help='sets a password for your notebook server')27	parser_password.add_argument("password", help="password to set, if not given, a password will be queried for (NOTE: this may not be safe)",28			nargs="?")29	parser_password.add_argument("--quiet", help="suppress messages", action="store_true")30	parser_password.set_defaults(function=set_password)31	args = parser.parse_args(argv[1:])32	args.function(args)33