
    v                     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 register' command.    )absolute_import)division)unicode_literals)	instances)util)base)flagsz7
        Request for upgrading notebook instances.
    ze
    To upgrade an instance, run:

        $ {command} example-instance --location=us-central1-a
    )DESCRIPTIONEXAMPLESc                   &    e Zd ZdZed        Zd Zy)Upgradez Request for upgrading instances.c                 v    t        j                  | j                               }t        j                  ||       y)zUpgrade flags for this command.N)r   ApiVersionSelectorReleaseTrackr	   AddUpgradeInstanceFlags)clsparserapi_versions      *lib/surface/notebooks/instances/upgrade.pyArgszUpgrade.Args-   s-     ))#*:*:*<=K	!!+v6    c                 <   | j                         }t        j                  |      }t        j                  |      }|j                  }|j                  t        j                  ||            }t        j                  ||||t        j                  j                        S )N)operation_type)r   r   	GetClientGetMessagesprojects_locations_instancesr   instance_utilCreateInstanceUpgradeRequest	HandleLROOperationTypeUPGRADE)selfargsrelease_trackclientmessagesinstance_service	operations          r   RunzUpgrade.Run3   s    %%'M^^M*F.H:: ((224BDI""$22::< <r   N)__name__
__module____qualname____doc__classmethodr   r)    r   r   r   r   (   s     )7 7
<r   r   N)r-   
__future__r   r   r    googlecloudsdk.api_lib.notebooksr   r   r   googlecloudsdk.callioper   $googlecloudsdk.command_lib.notebooksr	   DETAILED_HELPReleaseTracksr   ALPHABETAGACommandr   detailed_helpr/   r   r   <module>r;      s    . &  ' G 1 ( 6		 D%%++T->->-C-C%%((*<dll <*<0 & r   