
    	                     ,   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	d
Zej                   ej                   ej"                  j$                  ej"                  j&                         G d dej(                                      Zee_        y)z.Resume command for Colab Enterprise Schedules.    )util)	schedules)base)endpoint_util)flags)logzD
        Resume a Colab Enterprise notebook execution schedule.
    z
        To resume a paused schedule with id `my-schedule`, in region `us-central1`, run:

         $ {command} my-schedule --region=us-central1
    )DESCRIPTIONEXAMPLESc                   &    e Zd ZdZed        Zd Zy)ResumezResume a schedule.c                 .    t        j                  |        y)z Register flags for this command.N)r   AddResumeScheduleFlags)parsers    %lib/surface/colab/schedules/resume.pyArgszResume.Args+   s     
  (    c                 ,   | j                         }t        j                  | j                               }|j                  j                  j                         }|j                         d   }t        j                  d|      5  t        j                  |      }|j                  }t        j                  |||       |j                  t        j                  ||            }t        j                   j#                  d       |cddd       S # 1 sw Y   yxY w)z9This is what gets called when the user runs this command.locationsIdBETA)versionregionzResumed schedule.N)ReleaseTrackr   GetMessagesCONCEPTSscheduleParseAsDictr   AiplatformEndpointOverrides	GetClientprojects_locations_schedulesschedules_utilValidateAndGetColabScheduler   CreateScheduleResumeRequestr   statusPrint)	selfargsrelease_trackmessagesschedule_refr   
api_clientschedules_serviceapi_responses	            r   Runz
Resume.Run0   s    %%'M 1 1 34H==))//1L  "=1F		2	2v
 >>-0j

1
1  00
+ '--

4
4Hl
 
jj*+!
 
 
s   A>D

DN)__name__
__module____qualname____doc__staticmethodr   r.    r   r   r   r   &   s     ) )r   r   N)r2   'googlecloudsdk.api_lib.colab_enterpriser   (googlecloudsdk.api_lib.notebook_executorr   r!   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   ,googlecloudsdk.command_lib.notebook_executorr   googlecloudsdk.corer   _DETAILED_HELPDefaultUniverseOnlyReleaseTracksr   GAr   UpdateCommandr   detailed_helpr4   r   r   <module>rA      s     5 8 P ( 7 > #		 D%%(($*;*;*@*@AT  B @ & r   