o ĎiD @sdZddlmZmZddlZddlmZddlmZddl m Z m Z m Z m Z ddlmZmZmZmZd d lmZmZeeZed d Zejd gdd  d.dedeeejdd fddfddZejdddgd d/deeeejdd fddfddZdeefddZ ejd d!d"gd  d.deeeejd#d fdeeejdd fddfd$d%Z!ejd&d'gdd0d(d)Z"ejd*d+gdd0d,d-Z#dS)1aContains 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 ) AnnotatedOptionalN)ENDPOINT)whoami) auth_list auth_switchloginlogout)ANSIget_stored_tokens get_tokenlogging)TokenOpt 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-credential)examplesFtokenadd_to_git_credentialzTSave to git credential helper. Useful only if you plan to run git commands directly.returncCst||ddS)z8Login using a token from huggingface.co/settings/tokens.rrN)r rrT/home/jeff/fluffinator/venv/lib/python3.10/site-packages/huggingface_hub/cli/auth.py auth_login3srr 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_logoutHs 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) token_namesirchoiceindexrrr_select_token_nameVs(       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_cmdns r2rz hf auth listcCs tdS)zList all stored access tokens.N)rrrrr auth_list_cmds r3rzhf auth whoamicCst}|durtdtt|}ttd|ddd|dD}|r3ttdd |td kr@td tdSdS) z;Find out which huggingface.co account you are logged in as.Nz Not logged inzuser: namecSsg|]}|dqS)r4r).0orgrrr szauth_whoami..orgszorgs: ,zhttps://huggingface.coz(Authenticated through private endpoint: ) r r#r0r1rr boldjoinr)rinfor8rrr auth_whoamisr=)NF)N)rN)$__doc__typingrrr0huggingface_hub.constantsrhuggingface_hub.hf_apir_loginrrr r utilsr r r r _cli_utilsrr get_logger__name__r!auth_clicommandboolOptionrstrrr.r2r3r=rrrrs