o Ďi'@sdZddlZddlmZmZddlZddlmZmZm Z m Z ddl m Z ddl mZmZmZmZmZmZmZmZedd Zed d Zed d Zejed d ejedd GdddeejZejdddgdejdddddfdededeeeejdd fdededee ejdd fdeeeejdd fd dfd!d"Z!ejd#d$gdejddfdededed%ee ejd&d fd df d'd(Z"ejd)d*gddejfd+ed,ededed df d-d.Z#ejd/d0d1gddddejfded2eeeejd3d fdeee ejd4d fdeded df d5d6Z$ejdd7d8gdddejdfdedeeej%d9d fd:edededee ejd;d fd dfdgddejfdedeeej%d?d fdeded df d@dAZ'ejddBdCgddddejfded eeej%dDd fdEeeeejdFdGdHd fd:ededed dfdIdJZ(ejdKdLgddejfdededed dfdMdNZ)ejd#dOgdddejfded eeej%dPd fdQee ejdRdSdTd fdeded df dUdVZ*dS)Wa Contains commands to interact with repositories on the Hugging Face Hub. Usage: # create a new dataset repo on the Hub hf repo create my-cool-dataset --repo-type=dataset # create a private model repo on the Hub hf repo create my-cool-model --private N) AnnotatedOptional)CLIErrorHfHubHTTPErrorRepositoryNotFoundErrorRevisionNotFoundError)ANSI) PrivateOpt RepoIdArgRepoType RepoTypeOpt RevisionOptTokenOpt get_hf_api typer_factoryzManage repos on the Hub.)helpz"Manage tags for a repo on the Hub.z&Manage branches for a repo on the Hub.tag)namebranchc@seZdZdZdZdZdS) GatedChoicesautomanualfalseN)__name__ __module__ __qualname__rrrrrT/home/jeff/fluffinator/venv/lib/python3.10/site-packages/huggingface_hub/cli/repo.pyr3srcreatezhf repo create my-modelz7hf repo create my-dataset --repo-type dataset --private)examplesFrepo_id repo_type space_sdkzEHugging Face Spaces SDK type. Required when --type is set to 'space'.privatetokenexist_okz-Do not raise an error if repo already exists.resource_group_idzoResource group in which to create the repo. Resource groups is only available for Enterprise Hub organizations.returnc CsTt|d}|j||j|||||d}tdt|jdtdt|dS)zCreate a new repo on the Hub.r%)r!r"r$r%r&r'r#Successfully created on the Hub.zYour repo is now available at N)r create_repovalueprintrboldr!) r!r"r#r$r%r&r'apirepo_urlrrr repo_create9s  r2deletezhf repo delete my-model missing_okz=If set to True, do not raise an error if repo does not exist.cCs6t|d}|j||j|dtdt|ddS)z>Delete a repo from the Hub. This is an irreversible operation.r))r!r"r4Successfully deleted r+N)r delete_repor-r.rr/)r!r"r%r4r0rrr repo_deletegs r7movez:hf repo move old-namespace/my-model new-namespace/my-modelfrom_idto_idcCsBt|d}|j|||jdtdt|dt|ddS)z8Move a repository from a namespace to another namespace.r))r9r:r"zSuccessfully moved z to r+N)r move_repor-r.rr/)r9r:r%r"r0rrr repo_move}s &r<settingsz#hf repo settings my-model --privatez&hf repo settings my-model --gated autogatedz$The gated status for the repository.z)Whether the repository should be private.cCsBt|d}|j||r |jnd||jdtdt|ddS)z$Update the settings of a repository.r)N)r!r>r$r"z%Successfully updated the settings of r+)rupdate_repo_settingsr-r.rr/)r!r>r$r%r"r0rrr repo_settingss  r@z"hf repo branch create my-model devz4hf repo branch create my-model dev --revision abc123z!The name of the branch to create.revisionz?If set to True, do not raise an error if branch already exists.c CsLt|d}|j||||j|dtdt|d|jdt|dS)z*Create a new branch for a repo on the Hub.r))r!rrAr"r&r* branch on  N)r create_branchr-r.rr/)r!rrAr%r"r&r0rrr branch_creates ,rEz"hf repo branch delete my-model devz!The name of the branch to delete.c CsHt|d}|j|||jdtdt|d|jdt|dS)z'Delete a branch from a repo on the Hub.r))r!rr"r5rBrCN)r delete_branchr-r.rr/)r!rr%r"r0rrr branch_deletes ,rGz hf repo tag create my-model v1.0z3hf repo tag create my-model v1.0 -m "First release"zThe name of the tag to create.messagez-mz --messagez%The description of the tag to create.c Cs |j}t|d}tdt|d|dt|z |j|||||dWnKty@}z t|d|d|d}~wt yT}z td |d|d}~wt yr}z|j j d krmtd |d |d |d}~wwtdt|dt|dS)zCreate a tag for a repo.r)zYou are about to create tag  on rC)r!r tag_messagerAr" ' ' not found.Nz Revision 'iTag 'z' already exists on ''.Tag z created on ) r-rr.rr/ create_tagrr capitalizerrresponse status_code) r!rrHrAr%r" repo_type_strr0errr tag_creates$ & $rVlistzhf repo tag list my-modelc Cs|j}t|d}z |j||d}Wnty*}z t|d|d|d}~wwt|jdkrs   (       '           %