
                         8   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
lmZ ddlmZ d Zd Zd Z ej(                  ej*                  j,                  ej*                  j.                         G d dej0                               Zy)z.Command to list Tensorboard runs in Vertex AI.    )absolute_import)division)unicode_literals)client)base)	constants)endpoint_util)flags)concept_parsers)	resourcesc                     t         j                  j                  | j                  t        j
                  t        j                  t        j                           }|j                         S )N)api_version)	r   REGISTRYParseRelativeNamenamer   TENSORBOARD_RUNS_COLLECTIONAI_PLATFORM_API_VERSIONBETA_VERSIONSelfLink)tensorboard_exprefs     'lib/surface/ai/tensorboard_runs/list.py_GetUriBetar      sR    ,,++33I4J4JK 	- 	M# 
    c                     t         j                  j                  dt        j                         dd      j                  |        y Nz--tensorboard-experiment-idzTo list Tensorboard runsT)required)r   ConceptParserForResourcer
   $GetTensorboardExperimentResourceSpecAddToParserparsers    r   _AddArgsr$   &   s<    ++#002 	 ,  ![(r   c                 X   | j                   j                  j                         }|j                         d   }t	        j
                  ||      5  t        j                  |      j                  || j                  | j                  | j                        cd d d        S # 1 sw Y   y xY w)NlocationsId)versionregion)r'   )tensorboard_exp_reflimit	page_sizesort_by)CONCEPTStensorboard_experiment_idParseAsDictr	   AiplatformEndpointOverridesr   TensorboardRunsClientListr*   r+   r,   )argsr'   r)   r(   s       r   _Runr5   .   s    ??EEG%%'6&00f&''8==/jj..	 > & & &s   AB  B)c                   .    e Zd ZdZddiZed        Zd Zy)ListBetazSList the Tensorboard runs of the given project, region, and Tensorboard experiment.EXAMPLESz          To list Tensorboard Runs in Tensorboard `12345` and Tensorboard Experiment `my-tensorboard-experiment`:

              $ {command} --tensorboard-id=12345 --tensorboard-experiment-id=my-tensorboard-experiment
          c                     t         j                  j                  dt        j                         dd      j                  |        | j                  j                  t               y r   )	r   r   r   r
   r    r!   display_info
AddUriFuncr   r"   s    r   ArgszListBeta.ArgsG   sP    !!--%224"	 .  #{6*
"";/r   c                 6    t        |t        j                        S )N)r5   r   r   )selfr4   s     r   RunzListBeta.RunP   s    i,,--r   N)__name__
__module____qualname____doc__detailed_helpstaticmethodr<   r?    r   r   r7   r7   :   s/    [ - 0 0.r   r7   N)rC   
__future__r   r   r   *googlecloudsdk.api_lib.ai.tensorboard_runsr   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   r	   r
   (googlecloudsdk.command_lib.util.conceptsr   googlecloudsdk.corer   r   r$   r5   ReleaseTracksReleaseTrackBETAALPHAListCommandr7   rF   r   r   <module>rR      s    5 &  ' = ( 3 7 / D ))	 D%%**D,=,=,C,CD.t . E.r   