o 0i| @sdZddlZddlmZmZddlZddlmZddlm Z ddl m Z m Z m Z mZddlmZmZmZmZd d lmZmZmZmZeeZed d Zejd gdd   d2dedeeejdd fdeeejdd fddfddZ ejdddgd d3deee!ejdd fddfddZ"dee!fd d!Z#ejd"d#d$gd  d4deee!ejd%d fdeeejdd fddfd&d'Z$ejd(d)gdd5d*d+Z%ejd,d-d.gdej&fd/eddfd0d1Z'dS)6aContains commands to authenticate to the Hugging Face Hub and interact with your repositories. Usage: # login and save token locally. hf auth login --token=hf_*** --add-to-git-credential # switch between tokens hf auth switch # list all tokens hf auth list # logout from all tokens hf auth logout # check which account you are logged in as hf auth whoami N) AnnotatedOptional)ENDPOINT)whoami) auth_list auth_switchloginlogout)ANSIget_stored_tokens get_tokenlogging) FormatOpt OutputFormatTokenOpt typer_factoryz,Manage authentication (login, logout, etc.).)helpr )z hf auth loginzhf auth login --token $HF_TOKENz7hf auth login --token $HF_TOKEN --add-to-git-credentialzhf auth login --force)ZexamplesFtokenadd_to_git_credentialzTSave to git credential helper. Useful only if you plan to run git commands directly.forcez)Force re-login even if already logged in.returncCst||| ddS)z8Login using a token from huggingface.co/settings/tokens.)rrZskip_if_logged_inN)r )rrrrR/home/kim/smarthome/.venv/lib/python3.10/site-packages/huggingface_hub/cli/auth.py auth_login4srr zhf auth logoutz$hf auth logout --token-name my-token token_namezName of token to logoutcCst|ddS)zLogout from a specific token.rN)r rrrr auth_logoutPs rcCstt}|stddStdt|dD] \}}t|d|q z,td}|dkr6WdSt |d}d|krHt |krOnn||WStd Wn t yatd Ynwq() Nz+No stored tokens found. Please login first.zAvailable stored tokens:rz. Tz=Enter the number of the token to switch to (or 'q' to quit): qrz$Invalid selection. Please try again.z4Invalid input. Please enter a number or 'q' to quit.) listr keysloggererrorprint enumerateinputlowerintlen ValueError)Z token_namesirchoiceindexrrr_select_token_name^s(       r.switchzhf auth switchz$hf auth switch --token-name my-tokenzName of the token to switch tocCs6|durt}|durtdtt||ddS)zSwitch between access tokens.Nz!No token name provided. Aborting.)r)r.r$typerExitr)rrrrrauth_switch_cmdvs r2z list | lsz hf auth listcCs tdS)zList all stored access tokens.N)rrrrr auth_list_cmds r3rzhf auth whoamizhf auth whoami --format jsonformatcCst}|dur!|tjkrttddittdtt|}|tjkr6ttj|dtddStt d|ddd |d D}|rVtt d d |t d krctdt dSdS)z;Find out which huggingface.co account you are logged in as.Nr#z Not logged inr)indentdefaultzuser: namecSsg|]}|dqS)r7r).0orgrrr szauth_whoami..orgszorgs: ,zhttps://huggingface.coz(Authenticated through private endpoint: ) r rjsonr$dumpsr0r1rstrr boldjoinr)r4rinfor;rrr auth_whoamis"  rC)NFF)N)NF)rN)(__doc__r=typingrrr0Zhuggingface_hub.constantsrZhuggingface_hub.hf_apirZ_loginrrr r utilsr r r rZ _cli_utilsrrrrZ get_logger__name__r"Zauth_clicommandboolOptionrr?rr.r2r3tablerCrrrrs