:vjdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZmZdZd dZd Ze d d Zdd Zy)z- Password generation for the Jupyter Server. N)contextmanager)jupyter_config_dir)Config)ConfigFileNotFoundJSONFileConfigLoader c`|gtdD]L}tjd}tjd}||k(r|}n&tjddNd}t ||d k(r0d dl}|j d d d }|j|}|d|Stj|} dttzdztjdtzz} | j|jd| jdz|d| d| j!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.argon2ri( ) memory_cost time_cost parallelism:z%0xutf-8ascii)rangegetpasswarningswarn ValueErrorrPasswordHasherhashhashlibnewstrsalt_lenrandom getrandbitsupdateencode hexdigest) passphrase algorithm_p0p1msgrphh_phhsalts m/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_server/auth/security.pypasswdr4s&4qA!34B!45BRx  MM3 B  >?[$q 00c|jdr.ddl}ddl}|j} |j |dd|S |jdd\}}} tj|}t|dk(ry|j|jd|jd z|j|k(S#|j j $rYywxYw#ttf$rYywxYw#t$rYywxYw) 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) startswithrargon2.exceptionsrverify exceptionsVerificationErrorsplitr TypeErrorr r!lenr&r'r()hashed_passphraser)rr/r*r2 pw_digestr1s r3 passwd_checkrBPs6##I.  " " $ 99.qr2J? ?%6%<% >? ;;=I %%%  22    " s5CC C5CC C21C25 DDc#K|(tjjtd}tjtjj |dt tjj|tjj |} |j}|t|dd5}|jtj|d ddd tj||y#t$r t}YnwxYw#1swY:xYw#t $r9t#j$}t'j(d |d |t*d YywxYww) 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 )ospathjoinrmakedirsdirnamerbasename load_configrropenwritejsondumpschmod Exception traceback format_excrrRuntimeWarning) config_filemodeloaderconfigftbs r3persist_configr_s!ggll#5#79UV KK ,t< !"''"2"2;"?Q\A] ^F##% L k3 0A  6!,- 1  d#  1 0   ! ! # +K=B4 @.]^  sfB%E;(D8E; 'D*1E;:D6E;D'$E;&D''E;*D3/E;6?E85E;7E88E;ct|}t|5}||j_ddd|S#1swY|SxYw)z:Ask user for password, store it in JSON configuration fileN)r4r_IdentityProviderhashed_password)passwordrYrbr\s r3 set_passwordrds=X&O  $2A/ %  % s3=)Nr)Ni)NN)__doc__rr rRrIr$rVr contextlibrjupyter_core.pathsrtraitlets.configrtraitlets.config.loaderrrr#r4rBr_rdr5r3rksZ %1#L 61r5&p  @r5