File: //usr/lib/google-cloud-sdk/lib/surface/workbench/schedules/__pycache__/pause.cpython-312.pyc
�
���
� � � d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
dd lmZ d
d d �Zej" ej$ ej& j( � G d
� dej* � � � Zee_ y)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 �Z � e Zd ZdZedej fd�� Zdej fd�Z
y)�PausezPause a schedule.�parserc �. � t j | � y)z^Register flags for this command.
Args:
parser: argparse parser for the command
N)r �AddPauseScheduleFlags)r s �(lib/surface/workbench/schedules/pause.py�Argsz
Pause.Args- s � �
� � ��'� �argsc �, � | 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� |cddd� S # 1 sw Y yxY w)z�This 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�regionzPaused 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 �Runz Pause.Run6 s� � � �%�%�'�M�� � �� 1� 1� 3�4�H��=�=�)�)�/�/�1�L�
�
�
�
"�=�
1�F� � 2� 2��v�
� �>�>�-�0�j�
�
1�
1� � �4�4�
��+�� '�,�,�
�
3�
3��H� ��l�
�j�j���)�*�
�!
�
�
�s
�A>D
�
DN) �__name__�
__module__� __qualname__�__doc__� staticmethodr �ArgumentInterceptorr r � Namespacer0 � r r r r ( s: � � ��(�#�7�7� (� �(� �'�1�1� r r N)r4 �'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_helpr8 r r �<module>rD s� �� -� 8� P� (� 4� 5� 7� >� #��� �� �������D�%�%�*�*�+�*�D� � � *� ,� �*�X %�� r