
    [                        S r SSKJr  SSKJr  SSKJr  SSKJr  SSKJr  SSK	J
r
  SSK	Jr  SS	K	Jr  SS
KJr  \R                  " \R                   R"                  \R                   R$                  5       " S S\R&                  5      5       rg)z&Command to update a SHA custom module.    )absolute_import)division)unicode_literals)clients)base)	constants)flags)parsing)
console_ioc                   .    \ rS rSrSr\S 5       rS rSrg)Update   a-  Update a Security Health Analytics custom module.

## EXAMPLES

To update an Security Health Analytics custom module with ID 123456 for
organization 123, run:

    $ {command} 123456
        --organization=organizations/123 --enablement-state="ENABLED"
        --custom-config-file=custom_config.yaml
c                    [         R                  " [        R                  R                  S9R                  U 5        [         R                  " SS9R                  U 5        [         R                  " SS9R                  U 5        [         R                  " S[        R                  R                  SS9R                  U 5        g )Nmodule_typeF)requiredTYAML)r   r   	file_type)	r	   CreateModuleIdOrNameArgr   CustomModuleTypeSHAAddToParserCreateParentFlagCreateValidateOnlyFlagCreateUpdateFlags)parsers    3lib/surface/scc/manage/custom_modules/sha/update.pyArgsUpdate.Args+   s    	!!..22k&	E*66v>	  %0<<VD	..22 k&    c                    [         R                  " U[        R                  R                  S9nUR
                  n[         R                  " UR                  5      n[         R                  " UR                  [        R                  R                  S9n[         R                  " U5      nU(       d$  [        R                  " SR                  U5      SS9  [        R                  " 5       nUR!                  UUUUUS9$ )Nr   zPAre you sure you want to update the Security Health Analytics custom module {}?
T)messagecancel_on_no)namevalidate_onlycustom_configenablement_stateupdate_mask)r
   GetModuleNameFromArgsr   r   r   r%   GetCustomConfigFromArgscustom_config_fileGetEnablementStateFromArgsr'   CreateUpdateMaskFromArgsr   PromptContinueformatr   SHACustomModuleClientr   )selfargsr$   r%   r&   r'   r(   clients           r   Run
Update.Run8   s    (()4488D &&M33D4K4KLM999+E+E+I+I 2248K%%+VD\ **,F==##)   r     N)	__name__
__module____qualname____firstlineno____doc__staticmethodr   r4   __static_attributes__r6   r    r   r   r      s     
 
 
r    r   N)r;   
__future__r   r   r   %googlecloudsdk.api_lib.scc.manage.shar   googlecloudsdk.callioper   %googlecloudsdk.command_lib.scc.manager   r	   r
   googlecloudsdk.core.consoler   ReleaseTracksReleaseTrackGAALPHACommandr   r6   r    r   <module>rH      sj    - &  ' 9 ( ; 7 9 2 D%%(($*;*;*A*AB5T\\ 5 C5r    