U 4Af@sddlZddlmZddlmZmZddlmZmZm Z ddl m Z ddl m Z Gdd d e ZGd d d Zeeeeefeeed d dZGdddZGdddeZGdddeZGdddeZGdddeZdS)N)ArgumentParser)ListUnion)HfFolder create_repowhoami) HTTPError)BaseTransformersCLICommandc@seZdZeedddZdS) UserCommands)parsercCs|jddd}|jddd|jddd}|jd dd|jd d d}|jd dd|jd dd}|jdd}|jddd}|jdtdd|jdtdd|jddddd|jddddS)Nloginz6Log in using the same credentials as on huggingface.co)helpcSst|SN) LoginCommandargsr>/tmp/pip-unpacked-wheel-zw5xktn0/transformers/commands/user.pyz2UserCommands.register_subcommand..)funcrz;Find out which huggingface.co account you are logged in as.cSst|Sr) WhoamiCommandrrrrrrZlogoutzLog outcSst|Sr) LogoutCommandrrrrr!rrepoz_Deprecated: use `huggingface-cli` instead. Commands to interact with your huggingface.co repos.zPDeprecated: use `huggingface-cli` instead. huggingface.co repos related commandscreatezNDeprecated: use `huggingface-cli` instead. Create a new repo on huggingface.conamezYName for your model's repo. Will be namespaced under your username to build the model id.)typerz--organizationz!Optional: organization namespace.z-yz--yes store_truez"Optional: answer Yes to the prompt)actionrcSst|Sr)RepoCreateCommandrrrrr5r) add_parser set_defaultsadd_subparsers add_argumentstr)r Z login_parserZ whoami_parserZ logout_parserZ repo_parserZrepo_subparsersZrepo_create_parserrrrregister_subcommands2z UserCommands.register_subcommandN)__name__ __module__ __qualname__ staticmethodrr&rrrrr sr c@sDeZdZdZdZdZdZdZeddZ edd Z ed d Z d S) ANSIz; Helper for en.wikipedia.org/wiki/ANSI_escape_code zzzzcCs|j||jSr)_bold_resetclssrrrboldBsz ANSI.boldcCs|j|j||jSr)r,_redr-r.rrrredFszANSI.redcCs|j||jSr)_grayr-r.rrrgrayJsz ANSI.grayN) r'r(r)__doc__r,r2r4r- classmethodr1r3r5rrrrr+8s  r+)rowsheadersreturncCs|ddt||fD}dt|j|}g}||j|||jdd|D|D]}||j|q\d|S)z Inspired by: - stackoverflow.com/a/8356620/593036 - stackoverflow.com/questions/9535954/printing-lists-as-tabular-data cSsg|]}tdd|DqS)css|]}tt|VqdSr)lenr%).0xrrr Vsz&tabulate...)max)r<colrrr Vsztabulate..z{{:{}}} cSsg|] }d|qS)-r)r<wrrrrAZs )zipr;formatappendjoin)r8r9Z col_widthsZ row_formatlinesrowrrrtabulateOsrKc@seZdZddZdS)BaseUserCommandcCs ||_dSrr)selfrrrr__init__aszBaseUserCommand.__init__N)r'r(r)rNrrrrrL`srLc@seZdZddZdS)rcCsttddS)NzERROR! `huggingface-cli login` uses an outdated login mechanism that is not compatible with the Hugging Face Hub backend anymore. Please use `huggingface-cli login instead.printr+r3rMrrrrunfs zLoginCommand.runNr'r(r)rRrrrrresrc@seZdZddZdS)rc Csttdt}|dkr,tdtz2t|\}}t||r\ttdd|WnDt k r}z&t|tt|j j tdW5d}~XYnXdS)NzxWARNING! `transformers-cli whoami` is deprecated and will be removed in v5. Please use `huggingface-cli whoami` instead. Not logged inzorgs: ,r ) rPr+r3r get_tokenexitrr1rHrresponsetext)rMtokenuserZorgserrrrRqs$ zWhoamiCommand.runNrSrrrrrpsrc@seZdZddZdS)rcCsttddS)NzERROR! `transformers-cli logout` uses an outdated logout mechanism that is not compatible with the Hugging Face Hub backend anymore. Please use `huggingface-cli logout instead.rOrQrrrrRs zLogoutCommand.runNrSrrrrrsrc@seZdZddZdS)r c Csttdt}|dkr.tdtdz*tddgd}tt | Wnt k rttdYnXz*tddgd}tt | Wn"t k rttd YnXtd t |\}}|j jdk r|j jn|}|d |j j}td t||j jsXtd }|d ksX|dksX|dksXtdtzt||d}WnFtk r} z&t| tt| jjtdW5d} ~ XYnXtdtdt|tdtd|td dS)NzlWARNING! Managing repositories through transformers-cli is deprecated. Please use `huggingface-cli` instead.rTr gitz --versionzutf-8z9Looks like you do not have git installed, please install.zgit-lfszLooks like you do not have git-lfs installed, please install. You can install from https://git-lfs.github.com/. Then run `git lfs install` (you only have to do this once)./zYou are about to create zProceed? [Y/n] yyesZAbort)Zrepo_idrZz Your repo now lives at:z zK You can clone it locally with the command below, and commit/push as usual.z git clone )rPr+r3rrVrW subprocess check_outputdecoder5stripFileNotFoundErrorrrZ organizationrr1rainputlowerrrrXrY) rMrZstdoutr[_ namespaceZ full_namechoiceurlr\rrrrRsX    zRepoCreateCommand.runNrSrrrrr sr )rbargparsertypingrrZhuggingface_hub.hf_apirrrZrequests.exceptionsrr^r r r+r%intrKrLrrrr rrrrs   &