o 0 i@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_helpargsZ auth_parserd/home/app/PaddleOCR-VL-test/.venv_paddleocr/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) AuthLoginrrrr r!Rr"r zLog outz --token-namez3Optional: Name of the access token to log out from.cSr'r) AuthLogoutrrrr r![r"whoamiz;Find out which huggingface.co account you are logged in as.cSr'r) AuthWhoamirrrr r!ar"switchzSwitch between access tokensz0Optional: Name of the access token to switch to.cSr'r) AuthSwitchrrrr r!or"listzList all stored access tokenscSr'r)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__ staticmethodrr5rrrr r:src@eZdZddZdS)BaseAuthCommandcCs||_t|_dSr)rr_api)selfrrrr __init__vs zBaseAuthCommand.__init__N)r6r7r8r>rrrr r;u r;c@r:)r(cCs tt|jj|jjddS)N)tokenadd_to_git_credential)rset_verbosity_infor rr@rAr=rrr run|s  z AuthLogin.runNr6r7r8rDrrrr r({r?r(c@r:)r)cCstt|jjddS)N) token_name)rrBr rrFrCrrr rDszAuthLogout.runNrErrrr r)r?r)c@s@eZdZddZdeefddZdeedeefddZd S) r-cCsJt|jj}|dur|}|durtdtt||jjddS)Nz!No token name provided. Aborting.)rA) rrBrrF_select_token_nameprintexitr rA)r=rFrrr rDszAuthSwitch.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_tuirH enumerateinputlowerintlen ValueError)r= token_namesirFchoiceindexrrr rGs,        zAuthSwitch._select_token_namerWcCsDdd|D}z tjd|ddWSty!tdYdSw)NcSsg|]}t||dqS)namer).0rFrrr sz5AuthSwitch._select_token_name_tui..zSelect a token to switch to:)messagechoicesdefaultzToken selection cancelled.)rselectexecuteKeyboardInterruptrMinfo)r=rWr`rrr rPs  z!AuthSwitch._select_token_name_tuiN) r6r7r8rDrr4rGrrPrrrr r-s r-c@r:)r/cCsttdSr)rrBr rCrrr rDs z AuthList.runNrErrrr r/r?r/c@r:)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: r\cSsg|]}|dqSr[r)r]orgrrr r^sz"AuthWhoami.run..orgszorgs: ,zhttps://huggingface.coz(Authenticated through private endpoint: r) rrHrIr<r*rboldjoinrrredresponsetext)r=r@rergerrr rDs& zAuthWhoami.runNrErrrr r+r?r+N))__doc__argparsertypingrrZrequests.exceptionsrZhuggingface_hub.commandsrZhuggingface_hub.constantsrZhuggingface_hub.hf_apirZ_loginr r r r utilsrrrZ _cli_utilsrZ get_loggerr6rMZ InquirerPyrZInquirerPy.base.controlrrO ImportErrorrr;r(r)r-r/r+rrrr s2          ; 3