File: //lib/google-cloud-sdk/lib/surface/dataflow/jobs/__pycache__/drain.cpython-312.pyc
�
��� � �, � d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z dd l
mZ e
j e
j j" e
j j$ � e
j& G d
� d e
j( � � � Zy )
z6Implementation of gcloud dataflow jobs drain command.
� )�absolute_import)�division)�unicode_literals)�apis)�
exceptions)�base)� job_utils)�logc �& � e Zd ZdZed� � Zd� Zy)�Draina, Drains all jobs that match the command line arguments.
Once Drain is triggered, the pipeline will stop accepting new inputs.
The input watermark will be advanced to infinity. Elements already in the
pipeline will continue to be processed. Drained jobs can safely be
cancelled.
c �2 � t j | d�� y)z Register flags for this command.�+)�nargsN)r �ArgsForJobRefs)�parsers �"lib/surface/dataflow/jobs/drain.py�Argsz
Drain.Args) s � � � � �V�3�/� c � � t j |� D ]{ } t j j |j
|j |j �� t j j dj |j
� � �} y# t j $ rb}t j j dj |j
|j j |j � � Y d}~��d}~ww xY w)z�This is what gets called when the user runs this command.
Args:
args: all the arguments that were provided to this command invocation.
)�
project_id� region_idzStarted draining job [{0}]z�Failed to drain job [{0}]: {1} Ensure that you have permission to access the job and that the `--region` flag, {2}, matches the job's region.N)r �ExtractJobRefsr �Jobsr �jobId� projectId�locationr
�status�Print�formatr �
HttpException�payload�status_message)�self�args�job_ref�errors r �Runz Drain.Run. s� � � �+�+�D�1��
� � � ��� �M�M� �(�(� �&�&� � (� �
�
���5�<�<�W�]�]�K�L�
2�� �
%�
%�
� �
�
���%� �f��
�
�u�}�}� ;� ;�W�=M�=M��
�
��
�s �A8B�D
�(AD�D
N)�__name__�
__module__� __qualname__�__doc__� staticmethodr r' � r r r r s � �� �0� �0�
r r N)r+ �
__future__r r r � googlecloudsdk.api_lib.dataflowr �googlecloudsdk.api_lib.utilr �googlecloudsdk.callioper �#googlecloudsdk.command_lib.dataflowr �googlecloudsdk.corer
�
ReleaseTracks� ReleaseTrack�BETA�GA�DefaultUniverseOnly�Commandr r- r r �<module>r: sy �� � '� � '� 0� 2� (� 9� #� ����D�%�%�*�*�D�,=�,=�,@�,@�A����$
�D�L�L� $
� � B�$
r