o Rh@s<dZddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z dd lmZmZmZmZdd lmZmZmZd d lmZeeZzdd lmZddlmZdZ Wn e!yddZ YnwGddde Z"GdddZ#Gddde#Z$Gddde#Z%Gddde#Z&Gddde#Z'Gddde#Z(dS) aDContains commands to authenticate to the Hugging Face Hub and interact with your repositories. Usage: # login and save token locally. huggingface-cli login --token=hf_*** --add-to-git-credential # switch between tokens huggingface-cli auth switch # list all tokens huggingface-cli auth list # logout from a specific token, if no token-name is provided, all tokens will be deleted from your machine. huggingface-cli logout --token-name=your_token_name # find out which huggingface.co account you are logged in as huggingface-cli whoami )_SubParsersAction)ListOptional) HTTPError)BaseHuggingfaceCLICommand)ENDPOINT)HfApi) auth_list auth_switchloginlogout)get_stored_tokens get_tokenlogging)ANSI)inquirerChoiceTFc@seZdZedefddZdS) UserCommandsparsercCs|jddd}|jdtdd|jddd d |jd d d |jddd}|jdd d |jddd}|jdtdd|jdd d |jddd}|jdd}|jddd}|jdtdd|jddd d |jdd d |jddd}|jdd d dS) Nr z8Log in using a token from huggingface.co/settings/tokens)helpz--tokenz;Token generated from https://huggingface.co/settings/tokens)typerz--add-to-git-credential store_truez.Optional: Save token to git credential helper.)actionrcSt|SN) LoginCommandargsr!_/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/huggingface_hub/commands/user.pyHz2UserCommands.register_subcommand..)funcwhoamiz;Find out which huggingface.co account you are logged in as.cSrr) WhoamiCommandrr!r!r"r#Jr$r zLog outz --token-namez3Optional: Name of the access token to log out from.cSrr) LogoutCommandrr!r!r"r#Rr$authz%Other authentication related commandszAuthentication subcommandsswitchzSwitch between access tokensz0Optional: Name of the access token to switch to.cSrr)AuthSwitchCommandrr!r!r"r#ar$listzList all stored access tokenscSrr)AuthListCommandrr!r!r"r#cr$) add_parser add_argumentstr set_defaultsadd_subparsers)r login_parser whoami_parser logout_parser auth_parserauth_subparsersauth_switch_parserauth_list_parserr!r!r"register_subcommand;sJ z UserCommands.register_subcommandN)__name__ __module__ __qualname__ staticmethodrr:r!r!r!r"r:src@eZdZddZdS)BaseUserCommandcCs||_t|_dSr)r r_api)selfr r!r!r"__init__gs zBaseUserCommand.__init__N)r;r<r=rCr!r!r!r"r@f r@c@r?)rcCs tt|jj|jjddS)N)tokenadd_to_git_credential)rset_verbosity_infor r rErFrBr!r!r"runms  zLoginCommand.runNr;r<r=rIr!r!r!r"rlrDrc@r?)r(cCstt|jjddS)N) token_name)rrGr r rKrHr!r!r"rIvszLogoutCommand.runNrJr!r!r!r"r(urDr(c@s@eZdZddZdeefddZdeedeefddZd S) r+cCsJt|jj}|dur|}|durtdtt||jjddS)Nz!No token name provided. Aborting.)rF) rrGr rK_select_token_nameprintexitr rF)rBrKr!r!r"rI|szAuthSwitchCommand.runreturncCstt}|stddStr||Stdt|dD] \}}t|d|q z,t d}| dkr=WdSt |d}d|krOt |krVnn||WStd Wn t yhtd 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.)r,rkeysloggererror_inquirer_py_available_select_token_name_tuirM enumerateinputlowerintlen ValueError)rB token_namesirKchoiceindexr!r!r"rLs,        z$AuthSwitchCommand._select_token_namer\cCsDdd|D}z tjd|ddWSty!tdYdSw)NcSsg|]}t||dqS)namer).0rKr!r!r" sz.zSelect a token to switch to:)messagechoicesdefaultzToken selection cancelled.)rselectexecuteKeyboardInterruptrRinfo)rBr\rer!r!r"rUs  z(AuthSwitchCommand._select_token_name_tuiN) r;r<r=rIrr0rLrrUr!r!r!r"r+{s r+c@r?)r-cCsttdSr)rrGr rHr!r!r"rIs zAuthListCommand.runNrJr!r!r!r"r-rDr-c@r?)r'c Cst}|durtdtz3|j|}t|ddd|dD}|r1ttdd|tdkr?td tWdSWdSt ye}zt|tt |j j td WYd}~dSd}~ww) Nz Not logged inracSsg|]}|dqSr`r!)rborgr!r!r"rcsz%WhoamiCommand.run..orgszorgs: ,zhttps://huggingface.coz(Authenticated through private endpoint: r) rrMrNrAr&rboldjoinrrredresponsetext)rBrErjrler!r!r"rIs&  zWhoamiCommand.runNrJr!r!r!r"r'rDr'N))__doc__argparsertypingrrrequests.exceptionsrhuggingface_hub.commandsrhuggingface_hub.constantsrhuggingface_hub.hf_apir_loginr r r r utilsrrr _cli_utilsr get_loggerr;rR InquirerPyrInquirerPy.base.controlrrT ImportErrorrr@rr(r+r-r'r!r!r!r"s2          , 3