
                         ,   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	S
.r\	R                  \	R                   " \	R"                  R$                  \	R"                  R&                  5       " S S\	R(                  5      5       5       r\\l        g)z-List command for Colab Enterprise Executions.    )
list_pager)util)
executions)base)endpoint_util)flagszY
        List your project's Colab Enterprise notebook executions in a given region.
    zm
        To list your executions in region `us-central1`, run:

        $ {command} --region=us-central1
    )DESCRIPTIONEXAMPLESc                   .    \ rS rSrSr\S 5       rS rSrg)List&   z3List your Colab Enterprise notebook execution jobs.c                 f    [         R                  " U 5        U R                  R                  S5        g)z Register flags for this command.z
        table(name.segment(-1):label=ID,
        displayName,
        name.segment(-3):label=REGION,
        gcsOutputUri,
        jobState)
    N)r   AddListExecutionsFlagsdisplay_info	AddFormat)parsers    $lib/surface/colab/executions/list.pyArgs	List.Args+   s+     
  (
!! # 	    c                    U R                  5       n[        R                  " U R                  5       5      nUR                  R                  R                  5       nUR                  5       S   n[        R                  " SUS9   [        R                  " U5      nUR                  n[        R                  " U[        R                  " X5      SUR                  SUR                   S S9sSSS5        $ ! , (       d  f       g= f)	z9This is what gets called when the user runs this command.locationsIdBETA)versionregionnotebookExecutionJobspageSizec                 8    [         R                  " U 5      (       + $ )N)executions_utilIsWorkbenchExecution)	executions    r   <lambda>List.Run.<locals>.<lambda>J   s    /*N*N+ 'r   )servicerequestfieldlimitbatch_size_attribute
batch_size	predicateN)ReleaseTrackr   GetMessagesCONCEPTSr   ParseAsDictr   AiplatformEndpointOverrides	GetClient(projects_locations_notebookExecutionJobsr   YieldFromListr   CreateExecutionListRequestr'   	page_size)selfargsrelease_trackmessages
region_refr   
api_clientexecutions_services           r   RunList.Run7   s    %%'M 1 1 34H%%++-J /F		2	2v
 >>-0j%NN%%$!<<TL'

)^^

 
 
s   A%C00
C> N)	__name__
__module____qualname____firstlineno____doc__staticmethodr   r=   __static_attributes__r?   r   r   r   r   &   s     <		 		r   r   N)rD   apitools.base.pyr   'googlecloudsdk.api_lib.colab_enterpriser   (googlecloudsdk.api_lib.notebook_executorr   r   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   ,googlecloudsdk.command_lib.notebook_executorr   _DETAILED_HELPDefaultUniverseOnlyReleaseTracksr+   GAr   ListCommandr   detailed_helpr?   r   r   <module>rS      s     4 ' 8 R ( 7 >		 D%%(($*;*;*@*@A%4 % B %P $ r   