o
    h+                     @   sp  d Z ddlZddlmZ ddlmZmZ ddlmZ ddl	m
Z
 ddlmZmZmZmZ ddlmZ d	d
lmZmZmZmZmZmZmZmZ d	dlmZmZmZ ddlm Z  e!e"Z#zddl$m%Z% ddl&m'Z' dZ(W n e)yv   dZ(Y nw G dd de
Z*G dd dZ+G dd de+Z,G dd de+Z-G dd de+Z.G dd de+Z/G dd de+Z0G d d! d!e+Z1dS )"a  Contains commands to authenticate to the Hugging Face Hub and interact with your repositories.

Usage:
    # login and save token locally.
    huggingface-cli login --token=hf_*** --add-to-git-credential

    # switch between tokens
    huggingface-cli auth switch

    # list all tokens
    huggingface-cli auth list

    # logout from a specific token, if no token-name is provided, all tokens will be deleted from your machine.
    huggingface-cli logout --token-name=your_token_name

    # find out which huggingface.co account you are logged in as
    huggingface-cli whoami

    # create a new dataset repo on the Hub
    huggingface-cli repo create mydataset --type=dataset

    N)_SubParsersAction)ListOptional)	HTTPError)BaseHuggingfaceCLICommand)ENDPOINT
REPO_TYPESREPO_TYPES_URL_PREFIXESSPACES_SDK_TYPES)HfApi   )NOTEBOOK_LOGIN_PASSWORD_HTMLNOTEBOOK_LOGIN_TOKEN_HTML_ENDNOTEBOOK_LOGIN_TOKEN_HTML_START	auth_listauth_switchloginlogoutnotebook_login)get_stored_tokens	get_tokenlogging   )ANSI)inquirerChoiceTFc                   @   s   e Zd ZedefddZdS )UserCommandsparserc                 C   s  | j ddd}|jdtdd |jddd	d
 |jdd d | j ddd}|jdd d | j ddd}|jdtdd |jdd d | j ddd}|jdd}|j ddd}|jdtdd |jddd	d
 |jdd d |j ddd}|jdd d | j d d!d}|jd"d}	|	j d#d$d}
|
jd%td&d |
jd'td(d |
jd)td*d |
jd+td,td- |
jd.d/dd0d
 |
jd1d d d S )2Nr   z8Log in using a token from huggingface.co/settings/tokens)helpz--tokenz;Token generated from https://huggingface.co/settings/tokens)typer   z--add-to-git-credential
store_truez.Optional: Save token to git credential helper.)actionr   c                 S      t | S N)LoginCommandargs r(   q/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/huggingface_hub/commands/user.py<lambda>V       z2UserCommands.register_subcommand.<locals>.<lambda>)funcwhoamiz;Find out which huggingface.co account you are logged in as.c                 S   r#   r$   )WhoamiCommandr&   r(   r(   r)   r*   X   r+   r   zLog outz--token-namez3Optional: Name of the access token to log out from.c                 S   r#   r$   )LogoutCommandr&   r(   r(   r)   r*   `   r+   authz%Other authentication related commandszAuthentication subcommandsswitchzSwitch between access tokensz0Optional: Name of the access token to switch to.c                 S   r#   r$   )AuthSwitchCommandr&   r(   r(   r)   r*   o   r+   listzList all stored access tokensc                 S   r#   r$   )AuthListCommandr&   r(   r(   r)   r*   q   r+   repoz={create} Commands to interact with your huggingface.co repos.z%huggingface.co repos related commandscreatez#Create a new repo on huggingface.conamezPName for your repo. Will be namespaced under your username to build the repo id.z--typezbOptional: repo_type: set to "dataset" or "space" if creating a dataset or space, default is model.z--organizationz!Optional: organization namespace.z--space_sdkzOOptional: Hugging Face Spaces SDK type. Required when --type is set to "space".)r    r   choicesz-yz--yesz"Optional: answer Yes to the promptc                 S   r#   r$   )RepoCreateCommandr&   r(   r(   r)   r*      r+   )
add_parseradd_argumentstrset_defaultsadd_subparsersr
   )r   login_parserwhoami_parserlogout_parserauth_parserauth_subparsersauth_switch_parserauth_list_parserrepo_parserrepo_subparsersrepo_create_parserr(   r(   r)   register_subcommandI   s   z UserCommands.register_subcommandN)__name__
__module____qualname__staticmethodr   rI   r(   r(   r(   r)   r   H   s    r   c                   @      e Zd Zdd ZdS )BaseUserCommandc                 C   s   || _ t | _d S r$   )r'   r   _api)selfr'   r(   r(   r)   __init__   s   zBaseUserCommand.__init__N)rJ   rK   rL   rR   r(   r(   r(   r)   rO          rO   c                   @   rN   )r%   c                 C   s    t   t| jj| jjd d S )N)tokenadd_to_git_credential)r   set_verbosity_infor   r'   rT   rU   rQ   r(   r(   r)   run   s
   
zLoginCommand.runNrJ   rK   rL   rX   r(   r(   r(   r)   r%      rS   r%   c                   @   rN   )r/   c                 C   s   t   t| jjd d S )N)
token_name)r   rV   r   r'   rZ   rW   r(   r(   r)   rX      s   zLogoutCommand.runNrY   r(   r(   r(   r)   r/      rS   r/   c                   @   s@   e Zd Zdd Zdee fddZdee dee fddZd	S )
r2   c                 C   sJ   t   | jj}|d u r|  }|d u rtd t  t|| jjd d S )Nz!No token name provided. Aborting.)rU   )	r   rV   r'   rZ   _select_token_nameprintexitr   rU   )rQ   rZ   r(   r(   r)   rX      s   zAuthSwitchCommand.runreturnc                 C   s   t t  }|std d S tr| |S td t|dD ]\}}t| d|  q 	 z,t	d}|
 dkr=W d S t|d }d|  krOt|k rVn n|| W S td	 W n tyh   td
 Y nw q/)Nz+No stored tokens found. Please login first.zAvailable stored tokens:r   z. Tz=Enter the number of the token to switch to (or 'q' to quit): qr   z$Invalid selection. Please try again.z4Invalid input. Please enter a number or 'q' to quit.)r3   r   keysloggererror_inquirer_py_available_select_token_name_tuir\   	enumerateinputlowerintlen
ValueError)rQ   token_namesirZ   choiceindexr(   r(   r)   r[      s,   


z$AuthSwitchCommand._select_token_namerk   c                 C   sD   dd |D }zt jd|d d W S  ty!   td Y d S w )Nc                 S   s   g | ]}t ||d qS )r7   r   ).0rZ   r(   r(   r)   
<listcomp>   s    z<AuthSwitchCommand._select_token_name_tui.<locals>.<listcomp>zSelect a token to switch to:)messager8   defaultzToken selection cancelled.)r   selectexecuteKeyboardInterruptra   info)rQ   rk   r8   r(   r(   r)   rd      s   
z(AuthSwitchCommand._select_token_name_tuiN)	rJ   rK   rL   rX   r   r<   r[   r   rd   r(   r(   r(   r)   r2      s    r2   c                   @   rN   )r4   c                 C   s   t   t  d S r$   )r   rV   r   rW   r(   r(   r)   rX      s   
zAuthListCommand.runNrY   r(   r(   r(   r)   r4      rS   r4   c                   @   rN   )r.   c              
   C   s   t  }|d u rtd t  z3| j|}t|d  dd |d D }|r1ttdd| tdkr?td	t  W d S W d S  t	ye } zt| tt
|jj td
 W Y d }~d S d }~ww )NNot logged inr7   c                 S   s   g | ]}|d  qS ro   r(   )rp   orgr(   r(   r)   rq      s    z%WhoamiCommand.run.<locals>.<listcomp>orgszorgs: ,zhttps://huggingface.coz(Authenticated through private endpoint: r   )r   r\   r]   rP   r-   r   boldjoinr   r   redresponsetext)rQ   rT   rw   rz   er(   r(   r)   rX      s&   zWhoamiCommand.runNrY   r(   r(   r(   r)   r.      rS   r.   c                   @   rN   )r9   c           
   
   C   s$  t  }|d u rtd td ztddgd}tt|  W n t	y1   td Y nw ztddgd}tt|  W n t	yW   tt
d Y nw td	 | j|d
 }| jjd urn| jjn|}| d| jj }| jjtvrtd td | jjtv rt| jj | }n|}tdt|  | jjstd }|d	ks|dks|dkstd t  z| jj||| jj| jjd}W n# ty }	 zt|	 tt
|	jj td W Y d }	~	nd }	~	ww td tdt|  td td|  td	 d S )Nrx   r   gitz	--versionzutf-8z9Looks like you do not have git installed, please install.zgit-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). r7   /zInvalid repo --typezYou are about to create zProceed? [Y/n] yyesAbort)repo_idrT   	repo_type	space_sdkz
Your repo now lives at:z  zK
You can clone it locally with the command below, and commit/push as usual.z
  git clone )r   r\   r]   
subprocesscheck_outputdecoder   graystripFileNotFoundErrorr~   rP   r-   r'   organizationr7   r    r   r	   r|   r   rf   rg   create_repor   r   r   r   )
rQ   rT   stdoutuser	namespacer   prefixed_repo_idrm   urlr   r(   r(   r)   rX      sl   
zRepoCreateCommand.runNrY   r(   r(   r(   r)   r9      rS   r9   )2__doc__r   argparser   typingr   r   requests.exceptionsr   huggingface_hub.commandsr   huggingface_hub.constantsr   r   r	   r
   huggingface_hub.hf_apir   _loginr   r   r   r   r   r   r   r   utilsr   r   r   
_cli_utilsr   
get_loggerrJ   ra   
InquirerPyr   InquirerPy.base.controlr   rc   ImportErrorr   rO   r%   r/   r2   r4   r.   r9   r(   r(   r(   r)   <module>   s6   (

H	3