
                             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\\l        g)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                   .    \ rS rSrSr\S 5       rS rSrg)Rollback(   z#Request for rolling back instances.c                 x    [         R                  " U R                  5       5      n[        R                  " X!5        g)zUpgrade flags for this command.N)r   ApiVersionSelectorReleaseTrackr	   AddRollbackInstanceFlags)clsparserapi_versions      +lib/surface/notebooks/instances/rollback.pyArgsRollback.Args,   s+     ))#*:*:*<=K	"";7    c                 j   U R                  5       n[        R                  " U5      n[        R                  " U5      nUR                  nUR                  S5      (       aU  UR                  [        R                  " X5      5      n[        R                  " UUUU[        R                  R                  S9$ g )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   RunRollback.Run2   s    %%'M^^M*F.H::)**"++

5
5d
EGi$$



&44==? ? +r    N)	__name__
__module____qualname____firstlineno____doc__classmethodr   r-   __static_attributes__r/   r   r   r   r   (   s    +8 8
?r   r   N)r4   
__future__r   r   r    googlecloudsdk.api_lib.notebooksr   r!   r   googlecloudsdk.callioper   $googlecloudsdk.command_lib.notebooksr	   DETAILED_HELPReleaseTracksr   GACommandr   detailed_helpr/   r   r   <module>r@      sp    . &  ' G 1 ( 6		 D%%(()?t|| ? *?2 ' r   