B 3\@sdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z m Z mZddlmZmZmZddlmZdZdd d Zd d ZedddZdddZdS)z' Password generation for the Notebook. )contextmanagerN) cast_bytes str_to_bytes cast_unicode)ConfigConfigFileNotFoundJSONFileConfigLoader)jupyter_config_dir sha1cCs|dkrNxDtdD]0}td}td}||kr:|}PqtdqWtdt|}dttdt d t}| t |d t |d d |||fS) aGenerate hashed password and salt for use in notebook configuration. In the notebook configuration, set `c.NotebookApp.password` to the generated string. Parameters ---------- passphrase : str Password to hash. If unspecified, the user is asked to input and verify a password. algorithm : str Hashing algorithm to use (e.g, 'sha1' or any argument supported by :func:`hashlib.new`). Returns ------- hashed_passphrase : str Hashed password, in the format 'hash_algorithm:salt:passphrase_hash'. Examples -------- >>> passwd('mypassword') 'sha1:7cf3:b7d6da294ea9592a9480c8f52e63cd42cfb9dd12' NzEnter password: zVerify password: zPasswords do not match.z'No matching passwords found. Giving up.z%0xzutf-8ascii:)rangegetpassprint ValueErrorhashlibnewstrsalt_lenrandomZ getrandbitsupdaterrjoin hexdigest) passphrase algorithmiZp0Zp1hsaltr"5lib/python3.7/site-packages/notebook/auth/security.pypasswds    r$c Csy|dd\}}}Wnttfk r.dSXyt|}Wntk rRdSXt|dkrddS|t|dt|d||kS)aVerify that a given passphrase matches its hashed version. Parameters ---------- hashed_passphrase : str Hashed password, in the format returned by `passwd`. passphrase : str Passphrase to validate. Returns ------- valid : bool True if the passphrase matches the hash. Examples -------- >>> from notebook.auth.security import passwd_check >>> passwd_check('sha1:0e112c3ddfce:a68df677475c2b47b6e86d0467eec97ac5f4b85a', ... 'mypassword') True >>> passwd_check('sha1:0e112c3ddfce:a68df677475c2b47b6e86d0467eec97ac5f4b85a', ... 'anotherpassword') False rFrzutf-8r) splitr TypeErrorrrlenrrr)Zhashed_passphraserrr!Z pw_digestr r"r"r# passwd_checkEs r)c cs|dkrtjtd}ttj|tj|}y |}Wntk rXt }YnX|Vt j |ddd}| t tj|ddWdQRXyt||Wn>tk r}z t}td||ftWdd}~XYnXdS) zContext manager that can be used to modify a config object On exit of the context manager, the config will be written back to disk, by default with user-only (600) permissions. Nzjupyter_notebook_config.jsonwutf8)encodingr%)indentz#Failed to set permissions on %s: %s)ospathrr rbasenamedirnameZ load_configrrioopenwriterjsondumpschmod Exception traceback format_excwarningswarnRuntimeWarning) config_filemodeloaderconfigfetbr"r"r#persist_configps   "rFc Cs(t|}t|}||j_WdQRXdS)zCAsk user for password, store it in notebook json configuration fileN)r$rFZ NotebookApppassword)rGr?Zhashed_passwordrBr"r"r# set_passwords rH)Nr )Nr*)NN)__doc__ contextlibrrrr3r6r/rr:r<Zipython_genutils.py3compatrrrZtraitlets.configrrrZjupyter_core.pathsr rr$r)rFrHr"r"r"r#s$   -+