
                            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 Zd Zd Z ej$                  ej&                  j(                         G d dej*                               Z ej$                  ej&                  j.                  ej&                  j0                         G d dej*                               Zy)z*Command to list Tensorboards in Vertex AI.    )absolute_import)division)unicode_literals)client)base)	constants)endpoint_util)flags)	resourcesc                     t         j                  j                  | j                  t        j
                  t        j                  t        j                           }|j                         S N)api_version)	r   REGISTRYParseRelativeNamenamer   TENSORBOARDS_COLLECTIONAI_PLATFORM_API_VERSION
GA_VERSIONSelfLinktensorboardrefs     #lib/surface/ai/tensorboards/list.py	_GetUriGar      sR    ,,''33I4H4HI 	- 	K# 
    c                     t         j                  j                  | j                  t        j
                  t        j                  t        j                           }|j                         S r   )	r   r   r   r   r   r   r   BETA_VERSIONr   r   s     r   _GetUriBetar   %   sR    ,,''33I4J4JK 	- 	M# 
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!   )limit	page_size
region_refsort_by)CONCEPTSr"   ParseAsDictr	   AiplatformEndpointOverridesr   TensorboardsClientListr#   r$   r&   )argsr!   r%   r"   s       r   _Runr.   -   s    }}##))+*}-&00f&$$W5::jj..	 ; & & &s   AB  B)c                   .    e Zd ZdZddiZed        Zd Zy)ListGa7Lists the Tensorboards of the given project and region.EXAMPLESE          To list Tensorboards:

              $ {command}
          c                 n    t        j                  | d       | j                  j                  t               y Nzto list Tensorboards)r
   AddRegionResourceArgdisplay_info
AddUriFuncr   parsers    r   ArgszListGa.ArgsF   s'    	v'=>
""9-r   c                 6    t        |t        j                        S N)r.   r   r   selfr-   s     r   Runz
ListGa.RunK   s    i**++r   N__name__
__module____qualname____doc__detailed_helpstaticmethodr;   r@    r   r   r0   r0   9   s/    ? - . .,r   r0   c                   .    e Zd ZdZddiZed        Zd Zy)ListBetar1   r2   r3   c                 n    t        j                  | d       | j                  j                  t               y r5   )r
   r6   r7   r8   r   r9   s    r   r;   zListBeta.Args\   s'    	v'=>
"";/r   c                 6    t        |t        j                        S r=   )r.   r   r   r>   s     r   r@   zListBeta.Runa   s    i,,--r   NrA   rH   r   r   rJ   rJ   O   s/    ? - 0 0.r   rJ   N)rE   
__future__r   r   r   &googlecloudsdk.api_lib.ai.tensorboardsr   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   r	   r
   googlecloudsdk.corer   r   r   r.   ReleaseTracksReleaseTrackGAListCommandr0   BETAALPHArJ   rH   r   r   <module>rX      s    1 &  ' 9 ( 3 7 / )	 D%%((),T , *,* D%%**D,=,=,C,CD.t . E.r   