
                            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
 rS rSS0r\R                   " \R"                  R$                  5       " S S\R&                  5      5       r\R                   " \R"                  R*                  \R"                  R,                  5       " S S\R&                  5      5       rg)z$ai-platform operations wait command.    )absolute_import)division)unicode_literals)
operations)base)endpoint_util)flags)operations_utilc                     [         R                  R                  U 5        [         R                  " 5       R                  U 5        g N)r	   OPERATION_NAMEAddToParserGetRegionArgparsers    *lib/surface/ai_platform/operations/wait.py_AddWaitArgsr      s-    ""6*""6*    c                     [         R                  " U R                  S9   [        R                  " 5       n[
        R                  " XR                  5      sS S S 5        $ ! , (       d  f       g = f)N)region)r   MlEndpointOverridesr   r   OperationsClientr
   Wait	operation)argsclients     r   _Runr   !   s@    ((<((*F7 =<<s   5A
A,DESCRIPTIONa          Wait for an AI Platform operation to complete.

        Given an operation ID, this command polls the operation and blocks
        until it completes. At completion, the operation message is printed
        (which includes the operation response).
    c                   2    \ rS rSrSr\r\S 5       rS r	Sr
g)r   2   .Wait for an AI Platform operation to complete.c                     [        U 5        g r   r   r   s    r   Args	Wait.Args8   
    r   c                     [        U5      $ r   r   selfr   s     r   RunWait.Run<       :r    N__name__
__module____qualname____firstlineno____doc___DETAILED_HELPdetailed_helpstaticmethodr$   r+   __static_attributes__r.   r   r   r   r   2   !    6 - r   r   c                   2    \ rS rSrSr\r\S 5       rS r	Sr
g)WaitBeta@   r!   c                     [        U 5        g r   r#   r   s    r   r$   WaitBeta.ArgsF   r&   r   c                     [        U5      $ r   r(   r)   s     r   r+   WaitBeta.RunJ   r-   r   r.   Nr/   r.   r   r   r;   r;   @   r9   r   r;   N)r4   
__future__r   r   r    googlecloudsdk.api_lib.ml_enginer   googlecloudsdk.callioper   $googlecloudsdk.command_lib.ml_enginer   r	   r
   r   r   r5   ReleaseTracksReleaseTrackGACreateCommandr   ALPHABETAr;   r.   r   r   <module>rK      s    + &  ' 7 ( > 6 @+
8 		 D%%(()
4 
 *
 D%%++T->->-C-CD
t!! 
 E
r   