
                         D   d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z
 ddlmZ d	d
dZ e
j                  e
j                  j                   e
j                  j"                  e
j                  j$                         G d de
j&                               Zee_        y)z%'notebooks instances update' command.    )absolute_import)division)unicode_literals)	instances)util)base)flagsz6
        Request for updating notebook instances.
    a  
    To update machine type for an instance, run:

        $ {command} example-instance --machine-type=n1-standard-8 --location=us-central1-a

    To update labels for an instance, run:

        $ {command} example-instance --labels=k1=v1,k2=v2 --location=us-central1-a

    To update labels and accelerator cores, run:

        $ {command} example-instance --labels=k1=v1,k2=v2 --accelerator-core-count=2 --location=us-central1-a
    )DESCRIPTIONEXAMPLESc                   &    e Zd ZdZed        Zd Zy)UpdatezRequest for updating instances.c                 v    t        j                  | j                               }t        j                  ||       y)z Register flags for this command.N)r   ApiVersionSelectorReleaseTrackr	   AddUpdateInstanceFlags)clsparserapi_versions      )lib/surface/notebooks/instances/update.pyArgszUpdate.Args5   s-     ))#*:*:*<=K	  f5    c                 $   | j                         }t        j                  |      }t        j                  |      }|j                  }|j                  d      s|j                  d      rW|j                  t        j                  ||            }t        j                  ||||t        j                  j                         |j                  d      rW|j                  t        j                  ||            }t        j                  ||||t        j                  j                         |j                  d      rX|j                  t        j                  ||            }t        j                  ||||t        j                  j                         y y )Naccelerator_typeaccelerator_core_count)operation_typelabelsmachine_type)r   r   	GetClientGetMessagesprojects_locations_instancesIsSpecifiedSetAcceleratorinstance_utilCreateSetAcceleratorRequest	HandleLROOperationTypeUPDATE	SetLabelsCreateSetLabelsRequestSetMachineTypeCreateSetMachineTypeRequest)selfargsrelease_trackclientmessagesinstance_service	operations          r   Runz
Update.Run;   s\   %%'M^^M*F.H::*+t/?/? 0""11

3
3D(
CEi



&44;;= !",,

.
.tX
>@i



&44;;= '"11

3
3D(
CEi



&44;;= (r   N)__name__
__module____qualname____doc__classmethodr   r3    r   r   r   r   0   s     (6 6
 =r   r   N)r7   
__future__r   r   r    googlecloudsdk.api_lib.notebooksr   r#   r   googlecloudsdk.callioper   $googlecloudsdk.command_lib.notebooksr	   DETAILED_HELPReleaseTracksr   ALPHABETAGACommandr   detailed_helpr9   r   r   <module>rE      s    , &  ' G 1 ( 6		, D%%++T->->-C-C%%((*)=T\\ )=*)=X % r   