
                          ,   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$                  e	j"                  j&                         G d de	j(                                      Zee_        y)z,List command for Colab Enterprise Schedules.    )
list_pager)util)	schedules)base)endpoint_util)flagszb
        List your project's Colab Enterprise notebook execution schedules in a given region.
    zl
        To list your schedules in region `us-central1`, run:

        $ {command} --region=us-central1
    )DESCRIPTIONEXAMPLESc                   &    e Zd ZdZed        Zd Zy)Listz8List your Colab Enterprise notebook execution schedules.c                 d    t        j                  |        | j                  j                  d       y)z Register flags for this command.zu
        table(name.segment(-1):label=ID,
        displayName,
        state,
        cron,
        nextRunTime)
    N)r   AddListSchedulesFlagsdisplay_info	AddFormat)parsers    #lib/surface/colab/schedules/list.pyArgsz	List.Args+   s+     
'
!! # 	    c           
         | j                         }t        j                  | j                               }|j                  j                  j                         }|j                         d   }t        j                  d|      5  t        j                  |      }|j                  }t        j                  |t        j                  ||      d|j                  d|j                   d       cddd       S # 1 sw Y   yxY w)	z9This is what gets called when the user runs this command.locationsIdBETA)versionregionr   pageSizec                 .    t        j                  |        S )N)schedules_utilFilterWorkbenchSchedule)schedules    r   <lambda>zList.Run.<locals>.<lambda>N   s    )O)O* &r   )servicerequestfieldlimitbatch_size_attribute
batch_size	predicateN)ReleaseTrackr   GetMessagesCONCEPTSr   ParseAsDictr   AiplatformEndpointOverrides	GetClientprojects_locations_schedulesr   YieldFromListr   CreateScheduleListRequestr#   	page_size)selfargsrelease_trackmessages
region_refr   
api_clientschedules_services           r   RunzList.Run7   s    %%'M 1 1 34H%%++-J /F		2	2v
 >>-0j

1
1  %%# ::H 

)^^
 
 
s   A&C22C;N)__name__
__module____qualname____doc__staticmethodr   r9    r   r   r   r   &   s     A		 		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'   GAr   ListCommandr   detailed_helpr?   r   r   <module>rL      s     3 ' 8 P ( 7 >		 D%%(($*;*;*@*@A)4 ) B )V $ r   