
    f                     ^   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$                  5       " S S\R&                  5      5       r\R                   " \R"                  R*                  5       " S S\R&                  5      5       rg)zBCommand to update an application workload in the Project/Location.    )absolute_import)division)unicode_literals)utils)	workloads)base)flagsz{description}a           To update the Workload `my-workload` with a new environment
        `prod` in the Application `my-app` in location `us-east1`,
        run:

          $ {command} my-workload --environment-type=TEST --application=my-app --location=us-east1
        )DESCRIPTIONEXAMPLESc                   2    \ rS rSrSr\r\S 5       rS r	Sr
g)UpdateGA&   &Update an Apphub application workload.c                 ^    [         R                  " U [        R                  R                  S9  g Nrelease_track)r	   !AddUpdateApplicationWorkloadFlagsr   ReleaseTrackGAparsers    3lib/surface/apphub/applications/workloads/update.pyArgsUpdateGA.Args,   s!    	++d//22    c                 4   [         R                  " [        R                  R                  S9n[
        R                  " U5      n[
        R                  " U[        R                  R                  S9nUR                  UR                  5       UR                  UUS9$ zRun the create command.r   )workload_id
async_flag
attributesargs)apisWorkloadsClientr   r   r   api_lib_utilsGetApplicationWorkloadRefPopulateAttributesUpdateRelativeNameasync_selfr"   clientworkload_refr!   s        r   RunUpdateGA.Run2   s    !!0A0A0D0DEF ::4@L11D--00J == --/;;	   r    N__name__
__module____qualname____firstlineno____doc___DETAILED_HELPdetailed_helpstaticmethodr   r/   __static_attributes__r1   r   r   r   r   &   !    . - 
r   r   c                   2    \ rS rSrSr\r\S 5       rS r	Sr
g)UpdateAlphaA   r   c                 ^    [         R                  " U [        R                  R                  S9  g r   )r	   r   r   r   ALPHAr   s    r   r   UpdateAlpha.ArgsG   s!    	++d//55r   c                 4   [         R                  " [        R                  R                  S9n[
        R                  " U5      n[
        R                  " U[        R                  R                  S9nUR                  UR                  5       UR                  UUS9$ r   )r#   r$   r   r   rA   r%   r&   r'   r(   r)   r*   r+   s        r   r/   UpdateAlpha.RunM   s    !!0A0A0G0GHF ::4@L11D--33J == --/;;	   r   r1   Nr2   r1   r   r   r>   r>   A   r<   r   r>   N)r7   
__future__r   r   r   googlecloudsdk.api_lib.apphubr   r%   *googlecloudsdk.api_lib.apphub.applicationsr   r#   googlecloudsdk.callioper   !googlecloudsdk.command_lib.apphubr	   r8   ReleaseTracksr   r   UpdateCommandr   rA   r>   r1   r   r   <module>rL      s    I &  ' @ H ( 3 #	 D%%(()t!!  *4 D%%++,$$$  -r   