a h@s<dZddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z dd lmZmZmZmZdd lmZmZmZd d lmZeeZz dd lmZddlmZdZ Wne!ydZ Yn0Gddde 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@seZdZeedddZdS) AuthCommands)parsercs|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_helpargsZ auth_parserT/var/www/html/assistant/venv/lib/python3.9/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.)actionrcSst|Sr) AuthLoginrrrr r!Rr"r zLog outz --token-namez3Optional: Name of the access token to log out from.cSst|Sr) AuthLogoutrrrr r![r"whoamiz;Find out which huggingface.co account you are logged in as.cSst|Sr) AuthWhoamirrrr r!ar"switchzSwitch between access tokensz0Optional: Name of the access token to switch to.cSst|Sr) AuthSwitchrrrr r!or"listzList all stored access tokenscSst|Sr)AuthListrrrr r!rr") add_parseradd_subparsers set_defaults add_argumentstr)rZauth_subparsersZ login_parserZ logout_parserZ whoami_parserZauth_switch_parserZauth_list_parserrrr register_subcommand;sT z AuthCommands.register_subcommandN)__name__ __module__ __qualname__ staticmethodrr4rrrr r:src@seZdZddZdS)BaseAuthCommandcCs||_t|_dSr)rr_api)selfrrrr __init__vszBaseAuthCommand.__init__N)r5r6r7r<rrrr r9usr9c@seZdZddZdS)r'cCs tt|jj|jjddS)N)tokenadd_to_git_credential)rset_verbosity_infor rr=r>r;rrr run|s z AuthLogin.runNr5r6r7rArrrr r'{sr'c@seZdZddZdS)r(cCstt|jjddS)N) token_name)rr?r rrCr@rrr rAszAuthLogout.runNrBrrrr r(sr(c@s>eZdZddZeedddZeeeedddZd S) r,cCsJt|jj}|dur |}|dur6tdtt||jjddS)Nz!No token name provided. Aborting.)r>) rr?rrC_select_token_nameprintexitr r>)r;rCrrr rAszAuthSwitch.run)returncCstt}|s tddStr.||Stdt|dD]\}}t|d|q@zXt d}| dkrxWdSt |d}d|krt |krnn ||WStdWq\t ytd Yq\0q\dS) Nz+No stored tokens found. Please login first.zAvailable stored tokens:rz. z=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_tuirE enumerateinputlowerintlen ValueError)r; token_namesirCchoiceindexrrr rDs&       zAuthSwitch._select_token_name)rTrGcCsHdd|D}ztjd|ddWStyBtdYdS0dS)NcSsg|]}t||dqS)namer).0rCrrr r"z5AuthSwitch._select_token_name_tui..zSelect a token to switch to:)messagechoicesdefaultzToken selection cancelled.)rselectexecuteKeyboardInterruptrJinfo)r;rTr]rrr rMs   z!AuthSwitch._select_token_name_tuiN) r5r6r7rArr3rDrrMrrrr r,s r,c@seZdZddZdS)r.cCsttdSr)rr?r r@rrr rAsz AuthList.runNrBrrrr r.sr.c@seZdZddZdS)r*c Cst}|durtdtz^|j|}t|ddd|dD}|rbttdd|tdkrxtd tWnFt y}z.t|tt |j j td WYd}~n d}~00dS) Nz Not logged inrYcSsg|] }|dqSrXr)rZorgrrr r[r"z"AuthWhoami.run..orgszorgs: ,zhttps://huggingface.coz(Authenticated through private endpoint: r) rrErFr:r)rboldjoinrrredresponsetext)r;r=rbrderrr rAs   zAuthWhoami.runNrBrrrr r*sr*N))__doc__argparsertypingrrZrequests.exceptionsrZhuggingface_hub.commandsrZhuggingface_hub.constantsrZhuggingface_hub.hf_apirZ_loginr r r r utilsrrrZ _cli_utilsrZ get_loggerr5rJZ InquirerPyrZInquirerPy.base.controlrrL ImportErrorrr9r'r(r,r.r*rrrr s.           ; 3