
    	                         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
dZ e
j                  e
j                  j                          G d de
j"                               Zee_        y)z&'notebooks runtimes diagnose' command.    )absolute_import)division)unicode_literals)runtimes)util)base)flagsz5
        Request for diagnose notebook runtimes.
    a  
    To diagnose an runtime, run:

        $ {command} example-runtime --location=us-central1 --gcs-bucket=gs://example-bucket

    To diagnose an runtime with a relative path:

        $ {command} example-runtime --location=us-central1 --gcs-bucket=gs://example-bucket --relative-path=logs

    To diagnose an runtime, with packet capture:

        $ {command} example-runtime --location=us-central1 --gcs-bucket=gs://example-bucket --enable-packet-capture
    )DESCRIPTIONEXAMPLESc                   &    e Zd ZdZed        Zd Zy)DiagnosezRequest for diagnose runtimes.c                 v    t        j                  | j                               }t        j                  ||       y)z Register flags for this command.N)r   ApiVersionSelectorReleaseTrackr	   AddDiagnoseRuntimeFlags)clsparserapi_versions      *lib/surface/notebooks/runtimes/diagnose.pyArgszDiagnose.Args4   s-     ))#*:*:*<=K	!!+v6    c                 <   | j                         }t        j                  |      }t        j                  |      }|j                  }|j                  t        j                  ||            }t        j                  ||||t        j                  j                        S )z9This is what gets called when the user runs this command.)operation_type)r   r   	GetClientGetMessagesprojects_locations_runtimesr   runtime_utilCreateRuntimeDiagnoseRequest	HandleLROOperationTypeUPDATE)selfargsrelease_trackclientmessagesruntime_service	operations          r   RunzDiagnose.Run:   s    %%'M^^M*F.H88O((11$ACI!!#1188: :r   N)__name__
__module____qualname____doc__classmethodr   r)    r   r   r   r   0   s    &7 7
:r   r   N)r-   
__future__r   r   r    googlecloudsdk.api_lib.notebooksr   r   r   googlecloudsdk.callioper   $googlecloudsdk.command_lib.notebooksr	   DETAILED_HELPReleaseTracksr   GACommandr   detailed_helpr/   r   r   <module>r9      sr    - &  ' E 1 ( 6		, D%%(():t|| : *:2 ' r   