
    e                        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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 ej4                  ej6                  j8                         G d dej:                               Z ej4                  ej6                  j>                  ej6                  j@                         G d dej:                               Z!y)z#Vertex AI endpoints create command.    )absolute_import)division)unicode_literals)encoding)
operations)client)base)	constants)endpoint_util)flags)operations_util)tensorboards_util)
validation)labels_util)logc                    t        j                  | d       t        j                  d      j                  |        t        j                  d      j                  |        t        j
                  | d       t        j                  |        y )Nzto create a Tensorboardtensorboard)r   AddRegionResourceArgGetDisplayNameArgAddToParserGetDescriptionArgAddKmsKeyResourceArgr   AddCreateLabelsFlagsparsers    %lib/surface/ai/tensorboards/create.py_AddArgsr   #   sc    V%>?-(44V<-(44V<V]3""6*    c                    t        j                  | j                         | j                  j                  j                         }|j                         d   | _        t        j                  || j                        5  t        j                  |      }t        j                         }|j                  ||       }t        j                  ||t!        j"                  |j$                              }|Jt'        j(                  |      }d|v r1t*        j,                  j/                  dj1                  |d                |cddd       S # 1 sw Y   yxY w)#Create a new Vertex AI Tensorboard.locationsId)region)versionNnamez"Created Vertex AI Tensorboard: {}.)r   ValidateDisplayNamedisplay_nameCONCEPTSr"   ParseAsDictr   AiplatformEndpointOverridesr   TensorboardsClientr   OperationsClientCreater   WaitForOpMayber   ParseTensorboardOperationr$   r   MessageToPyValuer   statusPrintformat)argsr#   
region_reftensorboards_clientoperation_clientopresponse_msgresponses           r   _Runr;   +   s     !2!23}}##))+*!!#M2$+00M 33GD!224		#	#J	5B"11"33BGG<>L **<8h	8	

199(6:JK	M NMMs   =B>EEc                   .    e Zd ZdZddiZed        Zd Zy)CreateGar    EXAMPLESm            To create a Tensorboard with the display name `my tensorboard`:

              $ {command} --display-name="my tensorboard"

          You may also provide a description:

              $ {command} --description="my description"

          You may also provide labels:

              $ {command} --labels="label1=value1" --labels="label2=value2"
          c                     t        |        y Nr   r   s    r   ArgszCreateGa.ArgsU   
    Vr   c                 6    t        |t        j                        S rA   )r;   r
   
GA_VERSIONselfr4   s     r   RunzCreateGa.RunY   s    i**++r   N__name__
__module____qualname____doc__detailed_helpstaticmethodrC   rI    r   r   r=   r=   @   s/    + -"  ,r   r=   c                   .    e Zd ZdZddiZed        Zd Zy)
CreateBetar    r>   r?   c                     t        |        y rA   rB   r   s    r   rC   zCreateBeta.Argsr   rD   r   c                 6    t        |t        j                        S rA   )r;   r
   BETA_VERSIONrG   s     r   rI   zCreateBeta.Runv   s    i,,--r   NrJ   rQ   r   r   rS   rS   ]   s/    + -"  .r   rS   N)"rN   
__future__r   r   r   apitools.base.pyr   googlecloudsdk.api_lib.air   &googlecloudsdk.api_lib.ai.tensorboardsr   googlecloudsdk.callioper	   googlecloudsdk.command_lib.air
   r   r   r   r   r   $googlecloudsdk.command_lib.util.argsr   googlecloudsdk.corer   r   r;   ReleaseTracksReleaseTrackGACreateCommandr=   BETAALPHArS   rQ   r   r   <module>re      s    * &  ' % 0 9 ( 3 7 / 9 ; 4 < #+* D%%((),t!! , *,8 D%%**D,=,=,C,CD.## . E.r   