
    K                        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  SSKJr  SSS.r\R&                  " \R(                  \R*                  \R,                  5       " S S\R.                  5      5       rg)z)Command to update a new Assured Workload.    )absolute_import)division)unicode_literals)endpoint_util)message_util)	workloads)base)ReleaseTrack)flags)logz-Update a given Assured Workloads environment.a           To update a given Assured Workloads environment in the us-central1
        region, belonging to an organization with ID 123, with workload ID 456
        and an etag of 789 with a new display name of 'Test-Workload-2' and a
        new set of labels (including any required existing labels) of
        (key = 'ExistingLabelKey1', value = 'ExistingLabelValue1') and
        (key = 'NewLabelKey2', value = 'NewLabelValue2'), run:

          $ {command} organizations/123/locations/us-central1/workloads/456 --display-name=Test-Workload-2 --labels=ExistingLabelKey1=ExistingLabelValue1,NewLabelKey2=NewLabelValue2 --etag=789
        )DESCRIPTIONEXAMPLESc                   8    \ rS rSrSr\r\S 5       rS r	S r
Srg)Update/   z&Update Assured Workloads environments.c                 0    [         R                  " U 5        g )N)r   AddUpdateWorkloadFlags)parsers    'lib/surface/assured/workloads/update.pyArgsUpdate.Args5   s    	  (    c           	         UR                   R                  R                  5       nUR                  5       R	                  5       nUR                  5       n[        R                  " U R                  5       US9   [        R                  " UR                  UR                  UR                  5      n[        R                  " UR                  UR                  UR                  UR                  U R                  5       S9n[         R"                  " U R                  5       S9nUR%                  XdUS9U l        U R&                  sSSS5        $ ! , (       d  f       g= f)zRun the update command.)release_trackregion)display_namelabelsetagviolation_notifications_enabledr   )r   )workloadnameupdate_maskN)CONCEPTSr    ParseParentNameRelativeNamer   +AssuredWorkloadsEndpointOverridesFromRegionr
   r   CreateUpdateMaskr   r   r   CreateAssuredWorkloadr   apisWorkloadsClientr   updated_resource)selfargsworkload_resourcer   workload_namer"   r    clients           r   Run
Update.Run9   s   ..446%%',,.F%224M		B	B'')&
: 11


T[[$*N*Nk 33((yy*.*N*N))+h ##$2C2C2EFf$mm[ , Jd""
: 
: 
:s   6CE
Ec                 V    [         R                  " U R                  R                  SS9  g )NzAssured Workloads environment)kind)r   UpdatedResourcer-   r!   )r.   resources_were_displayeds     r   EpilogUpdate.EpilogO   s!    --22<>r   )r-   N)__name__
__module____qualname____firstlineno____doc___DETAILED_HELPdetailed_helpstaticmethodr   r3   r9   __static_attributes__ r   r   r   r   /   s&    . -) )#,>r   r   N)r?   
__future__r   r   r   googlecloudsdk.api_lib.assuredr   r   r   r+   googlecloudsdk.callioper	   googlecloudsdk.calliope.baser
   "googlecloudsdk.command_lib.assuredr   googlecloudsdk.corer   r@   ReleaseTracksGABETAALPHAUpdateCommandr   rD   r   r   <module>rP      sz    0 &  ' 8 7 < ( 5 4 # 	8			$ LOO\%6%68J8JK!>T !> L!>r   