o tf@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZdZdd d Zd d Ze dddZdddZdS)z- Password generation for the Jupyter Server. N)contextmanager)jupyter_config_dir)Config)ConfigFileNotFoundJSONFileConfigLoader argon2c Cs|dur*tdD]}td}td}||kr|}ntjdddqd}t||d krFd dl}|jd d d d}||}|d|St |} dt t dt dt } | |d| d|d| d| S)aGenerate hashed password and salt for use in server configuration. In the server configuration, set `c.ServerApp.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`, or 'argon2'). Returns ------- hashed_passphrase : str Hashed password, in the format 'hash_algorithm:salt:passphrase_hash'. Examples -------- >>> passwd("mypassword") # doctest: +ELLIPSIS 'argon2:...' NzEnter password: zVerify password: zPasswords do not match. stacklevelz'No matching passwords found. Giving up.rri( )Z memory_costZ time_costZ parallelism:z%0xutf-8ascii)rangegetpasswarningswarn ValueErrorrPasswordHasherhashhashlibnewstrsalt_lenrandom getrandbitsupdateencode hexdigest) passphrase algorithm_p0p1msgrphZh_phhsaltr-e/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/jupyter_server/auth/security.pypasswds.     r/c Cs|dr)ddl}ddl}|}z ||dd|WS|jjy(YdSwz |dd\}}}Wn tt fy@YdSwzt |}Wn tyRYdSwt |dkr[dS| |d|d ||kS) a`Verify 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 -------- >>> myhash = passwd("mypassword") >>> passwd_check(myhash, "mypassword") True >>> passwd_check(myhash, "otherpassword") False >>> passwd_check("sha1:0e112c3ddfce:a68df677475c2b47b6e86d0467eec97ac5f4b85a", "mypassword") True zargon2:rNFrr rr) startswithrZargon2.exceptionsrverify exceptionsZVerificationErrorsplitr TypeErrorrrlenr!r"r#)Zhashed_passphraser$rr*r%r,Z pw_digestr+r-r-r. passwd_checkPs.    r7ccs|dur tjtd}tjtj|ddttj|tj|}z|}Wn t y7t }Ynw|Vt |ddd}| t j|dd Wdn1sWwYz t||WdStyt}tjd |d |tdd YdSw) 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_server_config.jsonT)exist_okwutf8)encodingr )indentzFailed to set permissions on z: r )ospathjoinrmakedirsdirnamerbasenameZ load_configrropenwritejsondumpschmod Exception traceback format_excrrRuntimeWarning) config_filemodeloaderconfigftbr-r-r.persist_configs,     rScCs>t|}t| }||j_Wd|S1swY|S)z:Ask user for password, store it in JSON configuration fileN)r/rSZIdentityProviderhashed_password)passwordrMrTrPr-r-r. set_passwords   rV)Nr)Nr8)NN)__doc__rrrFr>rrJr contextlibrZjupyter_core.pathsrZtraitlets.configrZtraitlets.config.loaderrrrr/r7rSrVr-r-r-r.s$    98