
                            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d Zd Z ej*                  ej,                  j.                         G d dej0                               Z ej*                  ej,                  j4                  ej,                  j6                         G d dej0                               Zy)z ai-platform models list command.    )absolute_import)division)unicode_literals)models)base)endpoint_util)flags)models_util)region_util)	resourcesz	ml.modelszf
        table(
            name.basename(),
            defaultVersion.name.basename()
        )
    c                     t         j                  j                  | j                  t        j
                        }|j                         S N)r   REGISTRYParseRelativeNamenamer
   MODELS_COLLECTIONSelfLink)modelrefs     &lib/surface/ai_platform/models/list.py_GetUrir   '   s3    ,,jj+//	1#	    c                     | j                   j                  t               | j                   j                  t               t        j                  d      j                  |        y )NT)include_global)display_info	AddFormat_DEFAULT_FORMAT
AddUriFuncr   r	   GetRegionArgAddToParserparsers    r   _AddListArgsr#   -   sC    0  )D)55f=r   c                     t        j                  |       }t        j                  |      5  t	        j
                  t        j                               cd d d        S # 1 sw Y   y xY w)N)region)r   	GetRegionr   MlEndpointOverridesr
   Listr   ModelsClient)argsr%   s     r   _Runr+   3   sC      &&((7F//12 877s   'AA&c                   &    e Zd ZdZed        Zd Zy)r(   !List existing AI Platform models.c                     t        |        y r   r#   r!   s    r   Argsz	List.Args=   
    r   c                     t        |      S r   r+   selfr*   s     r   RunzList.RunA       :r   N__name__
__module____qualname____doc__staticmethodr0   r6    r   r   r(   r(   9       ) r   r(   c                   &    e Zd ZdZed        Zd Zy)ListBetar-   c                     t        |        y r   r/   r!   s    r   r0   zListBeta.ArgsI   r1   r   c                     t        |      S r   r3   r4   s     r   r6   zListBeta.RunM   r7   r   Nr8   r>   r   r   rA   rA   E   r?   r   rA   N)r<   
__future__r   r   r    googlecloudsdk.api_lib.ml_enginer   googlecloudsdk.callioper   $googlecloudsdk.command_lib.ml_enginer   r	   r
   r   googlecloudsdk.corer   _COLLECTIONr   r   r#   r+   ReleaseTracksReleaseTrackGAListCommandr(   BETAALPHArA   r>   r   r   <module>rP      s    ' &  ' 3 ( > 6 < < ) >3 D%%(()4  * D%%**D,=,=,C,CDt  Er   