
    	                        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diZ ej"                  ej$                  j&                         G d dej(                               Z ej"                  ej$                  j,                  ej$                  j.                         G d de             Zy)z)ai-platform versions set-default command.    )absolute_import)division)unicode_literals)versions_api)base)endpoint_util)flags)region_util)versions_utilc                     t        j                  dd      j                  |        t        j                  d      j                  |        t         j                  j                  |        y )NFT)
positionalrequired)include_global)r	   GetModelNameAddToParserGetRegionArgVERSION_NAMEparsers    /lib/surface/ai_platform/versions/set_default.py_AddSetDefaultArgsr      sJ    5AA&ID)55f=  (    c                    t        j                  |       }t        j                  |      5  t	        j
                         }t        j                  || j                  | j                        cd d d        S # 1 sw Y   y xY w)N)region)model)
r
   	GetRegionr   MlEndpointOverridesr   VersionsClientr   
SetDefaultversionr   )argsr   clients      r   _Runr#   #   sT      &&((7((*F##FDLL

K 877s   A A66A?DESCRIPTIONzSets an existing AI Platform version as the default for its model.

*{command}* sets an existing AI Platform version as the default for its
model.  Only one version may be the default for a given version.
c                   *    e Zd ZdZeZed        Zd Zy)r   BSets an existing AI Platform version as the default for its model.c                     t        |        y Nr   r   s    r   ArgszSetDefault.Args;   
    vr   c                     t        |      S r(   r#   selfr!   s     r   RunzSetDefault.Run?       :r   N)	__name__
__module____qualname____doc___DETAILED_HELPdetailed_helpstaticmethodr*   r0    r   r   r   r   5   s!    J - r   r   c                   &    e Zd ZdZed        Zd Zy)SetDefaultBetar&   c                     t        |        y r(   r)   r   s    r   r*   zSetDefaultBeta.ArgsG   r+   r   c                     t        |      S r(   r-   r.   s     r   r0   zSetDefaultBeta.RunK   r1   r   N)r2   r3   r4   r5   r8   r*   r0   r9   r   r   r;   r;   C   s    J r   r;   N)r5   
__future__r   r   r    googlecloudsdk.api_lib.ml_enginer   googlecloudsdk.callioper   $googlecloudsdk.command_lib.ml_enginer   r	   r
   r   r   r#   r6   ReleaseTracksReleaseTrackGADescribeCommandr   ALPHABETAr;   r9   r   r   <module>rH      s    0 &  ' 9 ( > 6 < >)L 	 D%%(()
%% 
 *
 D%%++T->->-C-CDZ  Er   