
    5pj                         S r SSKJr  SSKJr  SSKJrJrJrJ	r	J
r
JrJr  SSKJr   " S S	\5      rSS
 jr " S S\5      r " S S\5      r " S S\5      r " S S\5      rg)u.   ``ms secret`` group — manage Studio secrets.    )annotations   )RepoType   )
CLICommand
SubParsersadd_repo_type_arginfomake_apirender_tablesuccess)add_subcmd_token_endpointc                  6    \ rS rSrSr\SS j5       rSS jrSrg)	SecretCommand
   z4Top-level dispatcher for the ``secret`` subcommands.c                    U R                  SSS9nUR                  SSS9nSUl        [        R	                  U5        [
        R	                  U5        [        R	                  U5        [        R	                  U5        UR                  [        S9  g )	Nsecretz#Manage repository secrets (studio).helpsecret_actionACTION)destmetavarT)_command)

add_parseradd_subparsersrequired_SecretListregister
_SecretAdd_SecretUpdate_SecretDeleteset_defaultsr   )
subparsersparsersubs      U/var/www/html/pdf-tiff/venv/lib/python3.13/site-packages/modelscope_hub/cli/secret.pyr   SecretCommand.register   s}    &&x6[&\##(#KS!C s#s#]3    c                    [        U R                  SS 5      nUc  [        S5      eU" U R                  5      R                  5         g )N_secret_leafz/No secret action given. See `ms secret --help`.)getattrargs
SystemExitexecute)selfleafs     r'   r/   SecretCommand.execute   s9    tyy.$7<NOOTYY!r)    Nr$   r   returnNoner5   r6   )	__name__
__module____qualname____firstlineno____doc__staticmethodr   r/   __static_attributes__r3   r)   r'   r   r   
   s    >
4 
4"r)   r   c                ~    [        U [        R                  R                  /[        R                  R                  SS9  g )NF)choicesdefaultr   )r	   r   STUDIOvalue)r%   s    r'   _add_studio_repo_typerD   !   s-    &&'%%	r)   c                  2    \ rS rSr\SS j5       rSS jrSrg)r   *   c                    U R                  SSS9nUR                  S5        [        U5        [        U5        UR	                  [
        [        S9  g )NlistzList secrets of a studio space.r   repo_idr   r+   )r   add_argumentrD   r   r#   r   r   r$   ps     r'   r   _SecretList.register+   sD    !!&/P!Q	y!a !!$	KHr)   c                   [        U R                  5      nUR                  U R                  R                  U R                  R                  5      nU(       d  [        S5        g U Vs/ s H  nUR                  S5      =(       d    UR                  S5      =(       d    SUR                  S5      =(       d    SUR                  S5      =(       d    UR                  S5      =(       d    S4PM     nn[        [        U/ SQS	95        g s  snf )
Nz(no secrets)keyname-description
created_at
updated_at)rP   rS   rU   )headers)r   r-   list_secretsrI   	repo_typer
   getr   )r0   apisecretssrowss        r'   r/   _SecretList.execute3   s    tyy!""499#4#4dii6I6IJ  
 	 e4f4m$+l#Aquu\':Ac
  	 
 	\$(LMN
s   'BDr3   Nr4   r7   r8   r9   r:   r;   r=   r   r/   r>   r3   r)   r'   r   r   *   s    I IOr)   r   c                  2    \ rS rSr\SS j5       rSS jrSrg)r    D   c                    U R                  SSS9nUR                  S5        UR                  S5        UR                  S5        [        U5        [        U5        UR	                  [
        [        S9  g )NaddzAdd a new secret.r   rI   rP   rC   rJ   )r   rK   rD   r   r#   r   r    rL   s     r'   r   _SecretAdd.registerE   s\    !!%.A!B	y!	u	wa !!$	JGr)   c                l   [        U R                  5      nUR                  U R                  R                  U R                  R                  U R                  R
                  U R                  R                  5        [        SU R                  R                  < SU R                  R                   S35        g )NzAdded secret z to .)r   r-   
add_secretrI   rP   rC   rX   r   r0   rZ   s     r'   r/   _SecretAdd.executeO   sq    tyy!tyy(($))--$))J]J]^-		0TYY5F5F4GqIJr)   r3   Nr4   r7   r_   r3   r)   r'   r    r    D   s    H HKr)   r    c                  2    \ rS rSr\SS j5       rSS jrSrg)r!   U   c                    U R                  SSS9nUR                  S5        UR                  S5        UR                  S5        [        U5        [        U5        UR	                  [
        [        S9  g )NupdatezUpdate an existing secret.r   rI   rP   rC   rJ   )r   rK   rD   r   r#   r   r!   rL   s     r'   r   _SecretUpdate.registerV   s\    !!(1M!N	y!	u	wa !!$	MJr)   c                l   [        U R                  5      nUR                  U R                  R                  U R                  R                  U R                  R
                  U R                  R                  5        [        SU R                  R                  < SU R                  R                   S35        g )NzUpdated secret z on rf   )r   r-   update_secretrI   rP   rC   rX   r   rh   s     r'   r/   _SecretUpdate.execute`   ss    tyy!$))++TYY]]DIIOOTYYM`M`a/$))--!2$tyy7H7H6IKLr)   r3   Nr4   r7   r_   r3   r)   r'   r!   r!   U   s    K KMr)   r!   c                  2    \ rS rSr\SS j5       rSS jrSrg)r"   f   c                    U R                  SSS9nUR                  S5        UR                  S5        [        U5        UR                  SSSS	9  [        U5        UR	                  [
        [        S
9  g )NdeletezDelete a secret.r   rI   rP   z--yesz-y
store_true)actionrJ   )r   rK   rD   r   r#   r   r"   rL   s     r'   r   _SecretDelete.registerg   sb    !!(1C!D	y!	ua 	w\:!!$	MJr)   c                J   U R                   R                  (       di  [        SU R                   R                  < SU R                   R                   S35      R                  5       R                  5       nUS;  a  [        S5        g [        U R                   5      nUR                  U R                   R                  U R                   R                  U R                   R                  5        [        SU R                   R                  < SU R                   R                   S35        g )NzDelete secret z from z? [y/N] )yyeszAborted.zDeleted secret rf   )r-   r{   inputrP   rI   striplowerr
   r   delete_secretrX   r   )r0   answerrZ   s      r'   r/   _SecretDelete.executeq   s    yy}}^DIIMM+<F499CTCTBUU]^_eegmmoF\)Z tyy!$))++TYY]]DII<O<OP/$))--!2&9J9J8K1MNr)   r3   Nr4   r7   r_   r3   r)   r'   r"   r"   f   s    K KOr)   r"   Nr7   )r<   
__future__r   	constantsr   baser   r   r	   r
   r   r   r   compatr   r   rD   r   r    r!   r"   r3   r)   r'   <module>r      sc    4 "   b b b -"J ".O* O4K K"MJ M"OJ Or)   