
                            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	d
Ze	j                   e	j                   e	j"                  j$                         G d de	j&                                      Zee_        y)z&List command for Workbench Executions.    )
list_pager)util)
executions)base)endpoint_util)flagszI
        List your project's Workbench executions in a given region.
    zm
        To list your executions in region `us-central1`, run:

        $ {command} --region=us-central1
    )DESCRIPTIONEXAMPLESc                   &    e Zd ZdZed        Zd Zy)Listz"List your notebook execution jobs.c                 h    t        j                  | d       | j                  j                  d       y)z_Register flags for this command.

    Args:
      parser: argparse parser to add flags to.
    T)for_workbenchz
        table(name.segment(-1):label=ID,
        displayName,
        name.segment(-3):label=REGION,
        gcsOutputUri,
        jobState)
    N)r   AddListExecutionsFlagsdisplay_info	AddFormat)parsers    (lib/surface/workbench/executions/list.pyArgsz	List.Args+   s-     
  t<
!! # 	    c           
         | j                         }t        j                  | j                               }|j                  j                  j                         }|j                         d   }t        j                  d|      5  t        j                  |      }|j                  }t        j                  t        j                  |t        j                  ||      d|j                   d|j"                        cddd       S # 1 sw Y   yxY w)zThis is what gets called when the user runs this command.

    Args:
      args: The arguments that the command was run with.

    Returns:
      A list of Workbench executions in the specified region.
    locationsIdBETA)versionregionnotebookExecutionJobspageSize)	predicateservicerequestfieldlimitbatch_size_attribute
batch_sizeN)ReleaseTrackr   GetMessagesCONCEPTSr   ParseAsDictr   AiplatformEndpointOverrides	GetClient(projects_locations_notebookExecutionJobsr   YieldFromListexecutions_utilIsWorkbenchExecutionCreateExecutionListRequestr!   	page_size)selfargsrelease_trackmessages
region_refr   
api_clientexecutions_services           r   RunzList.Run;   s     %%'M 1 1 34H%%++-J /F		2	2v
 >>-0j

=
=  %% $88$!<<H (

)^^
 
 
s   A3C??DN)__name__
__module____qualname____doc__staticmethodr   r8    r   r   r   r   &   s     +	 	!r   r   N)r<   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$   r   ListCommandr   detailed_helpr>   r   r   <module>rJ      s     - ' 8 R ( 7 >		 D%%**+44 4 , 4l $ r   