
    W                         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lm
Z
 dd	lmZ d
diZd Z G d dej                        Zy)z%ai-platform jobs stream-logs command.    )absolute_import)division)unicode_literals)base)flags)	jobs_util)	log_utils)
propertiesEXAMPLESzu          To show the logs from running the AI Platform job ``my-job'', run:

            $ {command} my-job
        c                     t         j                  j                  |        t         j                  j                  |        t         j                  j                  |        t         j
                  j                  |        y N)r   JOB_NAMEAddToParserPOLLING_INTERVALALLOW_MULTILINE_LOGS	TASK_NAMEparsers    +lib/surface/ai_platform/jobs/stream_logs.py_AddStreamLogsArgsr   &   sN    ..V$$$V,((0//f%    c                   *    e Zd ZdZeZed        Zd Zy)
StreamLogsz)Show logs from a running AI Platform job.c                 l    t        |        | j                  j                  t        j                         y r   )r   display_info	AddFormatr	   
LOG_FORMATr   s    r   ArgszStreamLogs.Args2   s$    v
!!)"6"67r   c                     t        j                  |j                  |j                  t        j
                  j                  j                  j                         |j                        S )zRun the stream-logs command.)
r   r   job	task_namer
   VALUES	ml_enginepolling_intervalGetIntallow_multiline_logs)selfargss     r   RunzStreamLogs.Run7   sI    $..##44;;=!!# #r   N)	__name__
__module____qualname____doc__DETAILED_HELPdetailed_helpstaticmethodr   r)    r   r   r   r   -   s!    1-8 8#r   r   N)r-   
__future__r   r   r   googlecloudsdk.callioper   $googlecloudsdk.command_lib.ml_enginer   r   r	   googlecloudsdk.corer
   r.   r   Commandr   r1   r   r   <module>r7      sH    , &  ' ( 6 : : * 	&# #r   