o h@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) aContains 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 )_SubParsersAction)ListOptional) HTTPError)BaseHuggingfaceCLICommand)ENDPOINT)HfApi) auth_list auth_switchloginlogout)get_stored_tokens get_tokenlogging)ANSI)inquirerChoiceTFc@seZdZedefddZdS) AuthCommandsparsercs|jdddjdd}jfddd|jdd d}|jd td d |jd ddd|jddd|jddd}|jdtdd |jddd|jddd}|jddd|jddd}|jdtdd |jd ddd|jddd|jddd}|jd dddS)!Nauthz,Manage authentication (login, logout, etc.).)helpzAuthentication subcommandscsSN) print_helpargs auth_parserN/var/www/html/ai/venv/lib/python3.10/site-packages/huggingface_hub/cli/auth.pyBz2AuthCommands.register_subcommand..)funcr z8Log in using a token from huggingface.co/settings/tokensz--tokenz;Token generated from https://huggingface.co/settings/tokens)typerz--add-to-git-credential store_truez.Optional: Save token to git credential helper.)actionrcSt|Sr) AuthLoginrr r r!r"Rr#r zLog outz --token-namez3Optional: Name of the access token to log out from.cSr(r) AuthLogoutrr r r!r"[r#whoamiz;Find out which huggingface.co account you are logged in as.cSr(r) AuthWhoamirr r r!r"ar#switchzSwitch between access tokensz0Optional: Name of the access token to switch to.cSr(r) AuthSwitchrr r r!r"or#listzList all stored access tokenscSr(r)AuthListrr r r!r"rr#) add_parseradd_subparsers set_defaults add_argumentstr)rauth_subparsers login_parser logout_parser whoami_parserauth_switch_parserauth_list_parserr rr!register_subcommand;sT z AuthCommands.register_subcommandN)__name__ __module__ __qualname__ staticmethodrr<r r r r!r:src@eZdZddZdS)BaseAuthCommandcCs||_t|_dSr)rr_api)selfrr r r!__init__vs zBaseAuthCommand.__init__N)r=r>r?rEr r r r!rBu rBc@rA)r)cCs tt|jj|jjddS)N)tokenadd_to_git_credential)rset_verbosity_infor rrGrHrDr r r!run|s  z AuthLogin.runNr=r>r?rKr r r r!r){rFr)c@rA)r*cCstt|jjddS)N) token_name)rrIr rrMrJr r r!rKszAuthLogout.runNrLr r r r!r*rFr*c@s@eZdZddZdeefddZdeedeefddZd S) r.cCsJt|jj}|dur|}|durtdtt||jjddS)Nz!No token name provided. Aborting.)rH) rrIrrM_select_token_nameprintexitr rH)rDrMr r r!rKszAuthSwitch.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_tuirO enumerateinputlowerintlen ValueError)rD token_namesirMchoiceindexr r r!rNs,        zAuthSwitch._select_token_namer^cCsDdd|D}z tjd|ddWSty!tdYdSw)NcSsg|]}t||dqS)namer).0rMr r r! sz5AuthSwitch._select_token_name_tui..zSelect a token to switch to:)messagechoicesdefaultzToken selection cancelled.)rselectexecuteKeyboardInterruptrTinfo)rDr^rgr r r!rWs  z!AuthSwitch._select_token_name_tuiN) r=r>r?rKrr5rNrrWr r r r!r.s r.c@rA)r0cCsttdSr)rrIr rJr r r!rKs z AuthList.runNrLr r r r!r0rFr0c@rA)r,c Cst}|durtdtz7|j|}ttd|ddd|dD}|r5ttdd|td krCtd tWdSWdSt yi}zt|tt |j j td WYd}~dSd}~ww) Nz Not logged inzuser: rccSsg|]}|dqSrbr )rdorgr r r!resz"AuthWhoami.run..orgszorgs: ,zhttps://huggingface.coz(Authenticated through private endpoint: r) rrOrPrCr+rboldjoinrrredresponsetext)rDrGrlrner r r!rKs& zAuthWhoami.runNrLr r r r!r,rFr,N))__doc__argparsertypingrrrequests.exceptionsrhuggingface_hub.commandsrhuggingface_hub.constantsrhuggingface_hub.hf_apir_loginr r r r utilsrrr _cli_utilsr get_loggerr=rT InquirerPyrInquirerPy.base.controlrrV ImportErrorrrBr)r*r.r0r,r r r r!s2          ; 3