
                             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
S.r\R                  " \	R                   \	R"                  \	R$                  5       " S S\R&                  5      5       rg)z)Command to update a new Assured Workload.    )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.Args2   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.Run6   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.EpilogL   s!    --22<>r   )r*   N)__name__
__module____qualname____firstlineno____doc___DETAILED_HELPdetailed_helpstaticmethodr   r0   r6   __static_attributes__ r   r   r   r   ,   s&    . -) )#,>r   r   N)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   rA   r   r   <module>rL      ss    0 9 7 < ( 5 4 # 	8			$ LOO\%6%68J8JK!>T !> L!>r   