
    v                     D   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                  R"                  \
R                  R$                  5       " S S\
R&                  5      5       r\\l        g)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                   .    \ rS rSrSr\S 5       rS rSrg)Upgrade(   z Request for upgrading instances.c                 x    [         R                  " U R                  5       5      n[        R                  " X!5        g)zUpgrade flags for this command.N)r   ApiVersionSelectorReleaseTrackr	   AddUpgradeInstanceFlags)clsparserapi_versions      *lib/surface/notebooks/instances/upgrade.pyArgsUpgrade.Args-   s+     ))#*:*:*<=K	!!+6    c                 <   U R                  5       n[        R                  " U5      n[        R                  " U5      nUR                  nUR                  [        R                  " X5      5      n[        R                  " UUUU[        R                  R                  S9$ )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   RunUpgrade.Run3   s    %%'M^^M*F.H:: ((224BDI""$22::< <r    N)	__name__
__module____qualname____firstlineno____doc__classmethodr   r+   __static_attributes__r-   r   r   r   r   (   s     )7 7
<r   r   N)r2   
__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   