
                             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                          G d de
j"                               Zee_        y)z''notebooks instances rollback' command.    )absolute_import)division)unicode_literals)	instances)util)base)flagsz:
        Request for rolling back notebook instances.
    z
    To rollback an instance, run:

        $ {command} example-instance target-snapshot=projects/example-project/global/snapshots/aorlbjvpavvf --location=us-central1-a
    )DESCRIPTIONEXAMPLESc                   &    e Zd ZdZed        Zd Zy)Rollbackz#Request for rolling back instances.c                 v    t        j                  | j                               }t        j                  ||       y)zUpgrade flags for this command.N)r   ApiVersionSelectorReleaseTrackr	   AddRollbackInstanceFlags)clsparserapi_versions      +lib/surface/notebooks/instances/rollback.pyArgszRollback.Args,   s-     ))#*:*:*<=K	"";7    c                 `   | j                         }t        j                  |      }t        j                  |      }|j                  }|j                  d      rW|j                  t        j                  ||            }t        j                  ||||t        j                  j                        S y )Ntarget_snapshot)operation_type)r   r   	GetClientGetMessagesprojects_locations_instancesIsSpecifiedr   instance_utilCreateInstanceRollbackRequest	HandleLROOperationTypeROLLBACK)selfargsrelease_trackclientmessagesinstance_service	operations          r   RunzRollback.Run2   s    %%'M^^M*F.H::)*"++

5
5dH
EGi$$



&44==? ? +r   N)__name__
__module____qualname____doc__classmethodr   r+    r   r   r   r   (   s    +8 8
?r   r   N)r/   
__future__r   r   r    googlecloudsdk.api_lib.notebooksr   r   r   googlecloudsdk.callioper   $googlecloudsdk.command_lib.notebooksr	   DETAILED_HELPReleaseTracksr   GACommandr   detailed_helpr1   r   r   <module>r;      sr    . &  ' G 1 ( 6		 D%%(()?t|| ? *?2 ' r   