
    
                        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	KJr  S
SS.r\R"                  \R$                  " \R&                  R(                  5       " S S\R*                  5      5       5       r\\l        g)z&Pause command for Workbench Schedules.    )util)	schedules)base)parser_arguments)parser_extensions)endpoint_util)flags)logz<
        Pause a Workbench notebook execution schedule.
    z
        To pause a schedule with id `my-schedule`, in region `us-central1`, run:

         $ {command} my-schedule --region=us-central1
    )DESCRIPTIONEXAMPLESc                   f    \ rS rSrSr\S\R                  4S j5       rS\	R                  4S jrSrg)	Pause(   zPause a schedule.parserc                 0    [         R                  " U 5        g)zRRegister flags for this command.

Args:
  parser: argparse parser for the command
N)r	   AddPauseScheduleFlags)r   s    (lib/surface/workbench/schedules/pause.pyArgs
Pause.Args-   s     
'    argsc                 6   U R                  5       n[        R                  " U R                  5       5      nUR                  R                  R                  5       nUR                  5       S   n[        R                  " SUS9   [        R                  " U5      nUR                  n[        R                  " XU5        UR                  [        R                  " X5      5      n[        R                   R#                  S5        UsSSS5        $ ! , (       d  f       g= f)zThis is what gets called when the user runs this command.

Args:
  args: the arguments that this command was invoked
    with.

Returns:
  the API response for pausing the schedule.
locationsIdBETA)versionregionzPaused schedule.N)ReleaseTrackr   GetMessagesCONCEPTSscheduleParseAsDictr   AiplatformEndpointOverrides	GetClientprojects_locations_schedulesschedules_utilValidateAndGetWorkbenchScheduler   CreateSchedulePauseRequestr
   statusPrint)	selfr   release_trackmessagesschedule_refr   
api_clientschedules_serviceapi_responses	            r   Run	Pause.Run6   s     %%'M 1 1 34H==))//1L  "=1F		2	2v
 >>-0j

1
1  44
+ ',,

3
3l
 
jj)*!
 
 
s   A?D


D N)__name__
__module____qualname____firstlineno____doc__staticmethodr   ArgumentInterceptorr   r   	Namespacer2   __static_attributes__r4   r   r   r   r   (   s:     (#77 ( ('11 r   r   N)r9   'googlecloudsdk.api_lib.colab_enterpriser   (googlecloudsdk.api_lib.notebook_executorr   r&   googlecloudsdk.callioper   r   r   googlecloudsdk.command_lib.air   ,googlecloudsdk.command_lib.notebook_executorr	   googlecloudsdk.corer
   _DETAILED_HELPDefaultUniverseOnlyReleaseTracksr   r   UpdateCommandr   detailed_helpr4   r   r   <module>rI      s     - 8 P ( 4 5 7 > #		 D%%**+*D * , *X % r   