
                             S r SSKJr  SSKJr  SSKJr  SSKJr  SSKJr  SSKJ	r	  SSK
Jr  SS	K
Jr  \R                  4S
 jr " S S\5      rg)z6Utilities for AI Platform Tensorboard time series API.    )absolute_import)division)unicode_literals)
list_pager)apis)common_args)	constants)errorsc                 n    [         R                  " [        R                  [        R                  U    5      $ N)r   GetMessagesModuler	   AI_PLATFORM_API_NAMEAI_PLATFORM_API_VERSION)versions    ?lib/googlecloudsdk/api_lib/ai/tensorboard_time_series/client.pyr   r      s-    				 > > ) A A' J
L L    c                   p    \ rS rSrSrSS\R                  4S jrS rS r	SS jr
S rS	 rS
 rS rS rSrg)TensorboardTimeSeriesClient!   zFHigh-level client for the AI Platform Tensorboard time series surface.Nc                    U=(       d6    [         R                  " [        R                  [        R                  U   5      U l        U=(       d    U R
                  R                  U l        U R
                  R                  U l	        X0l
        g r   )r   GetClientInstancer	   r   r   clientMESSAGES_MODULEmessages;projects_locations_tensorboards_experiments_runs_timeSeries_service_version)selfr   r   r   s       r   __init__$TensorboardTimeSeriesClient.__init__$   sb      4D22&&))'24DK ; ; ;DMKK[[DMMr   c                 $    U R                  X5      $ r   )
CreateBeta)r   tensorboard_run_refargss      r   Create"TensorboardTimeSeriesClient.Create/   s    ??.55r   c                    UR                   S:X  a+  U R                  R                  R                  R                  nOeUR                   S:X  a+  U R                  R                  R                  R
                  nO*U R                  R                  R                  R                  nUR                  c  SnOUR                  nU R                  R                  UR                  5       U R                  R                  UR                  UR                  UUR                  [        USS9S9S9nU R                  R                  U5      $ )z%Create a new Tensorboard time series.scalarzblob-sequence utf8encoding)displayNamedescription	valueType
pluginName
pluginData)parent1googleCloudAiplatformV1beta1TensorboardTimeSeries)typer   1GoogleCloudAiplatformV1beta1TensorboardTimeSeriesValueTypeValueValuesEnumSCALARBLOB_SEQUENCETENSORplugin_dataMAiplatformProjectsLocationsTensorboardsExperimentsRunsTimeSeriesCreateRequestRelativeNamedisplay_namer.   plugin_namebytesr   r%   )r   r#   r$   
value_typer:   requests         r   r"   &TensorboardTimeSeriesClient.CreateBeta2   s   yyH
--
I
I##FF  
o	%
--
I
I##MM 
 --
I
I##FF  k$$kmmii"//1:>--	:	:))(( ''[6: 
; 
< j =G ==((r   c           	          U R                   R                  UR                  5       [        R                  " U5      S9n[
        R                  " U R                  USSUUS9$ )N)r2   orderBytensorboardTimeSeriespageSize)fieldbatch_size_attribute
batch_sizelimit)r   KAiplatformProjectsLocationsTensorboardsExperimentsRunsTimeSeriesListRequestr<   r   ParseSortByArgr   YieldFromListr   )r   r#   rJ   	page_sizesort_byrA   s         r   List TensorboardTimeSeriesClient.ListQ   sa    mmgg"//1**73 h 5G ##%' r   c                     U R                   R                  UR                  5       S9nU R                  R	                  U5      $ N)name)r   JAiplatformProjectsLocationsTensorboardsExperimentsRunsTimeSeriesGetRequestr<   r   Getr   tensorboard_time_series_refrA   s      r   rV   TensorboardTimeSeriesClient.Get]   s=    mmff(557 g 9G==W%%r   c                     U R                   R                  UR                  5       S9nU R                  R	                  U5      $ rS   )r   MAiplatformProjectsLocationsTensorboardsExperimentsRunsTimeSeriesDeleteRequestr<   r   DeleterW   s      r   r\   "TensorboardTimeSeriesClient.Deleteb   s=    mmii(557 j 9G==((r   c                 $    U R                  X5      $ r   )	PatchBeta)r   rX   r$   s      r   Patch!TensorboardTimeSeriesClient.Patchg   s    >>5<<r   c                    U R                   R                  5       n/ nUR                  b"  UR                  Ul        UR	                  S5        UR
                  b"  UR
                  Ul        UR	                  S5        UR                  b"  UR                  Ul        UR	                  S5        UR                  b*  [        UR                  SS9Ul
        UR	                  S5        U(       d  [        R                  " S5      eU R                   R                  UR                  5       USR                  U5      S	9nU R                   R#                  U5      $ )
z!Update a Tensorboard time series.r=   r.   r>   r*   r+   r:   zNo updates requested.,)rT   r3   
updateMask)r   r5   r=   r-   appendr.   r>   r0   r:   r?   r1   r
   NoFieldsSpecifiedErrorLAiplatformProjectsLocationsTensorboardsExperimentsRunsTimeSeriesPatchRequestr<   joinr   r`   )r   rX   r$   tensorboard_time_seriesupdate_maskrA   s         r   r_   %TensorboardTimeSeriesClient.PatchBetaj   s-    	GGI K$,0,=,=)(#,0,<,<)'#+/+;+;('#+0


V,-('))*ABBmmhh(557:Q88K( i *G ==w''r   c                     U R                   R                  UR                  5       UUS9nU R                  R	                  U5      $ )N)rE   maxDataPointsfilter)r   KAiplatformProjectsLocationsTensorboardsExperimentsRunsTimeSeriesReadRequestr<   r   Read)r   rX   max_data_pointsdata_filterrA   s        r   rp    TensorboardTimeSeriesClient.Read   sE    mmgg9FFH% h G ==g&&r   )r   r   r   r   )i  2   N)__name__
__module____qualname____firstlineno____doc__r	   BETA_VERSIONr   r%   r"   rP   rV   r\   r`   r_   rp   __static_attributes__ r   r   r   r   !   sD    N  --	6)>
&
)
=(@'r   r   N)ry   
__future__r   r   r   apitools.base.pyr   googlecloudsdk.api_lib.utilr   r   googlecloudsdk.command_lib.air	   r
   rz   r   objectr   r|   r   r   <module>r      sA    = &  ' ' , 3 3 0 (44 L
n'& n'r   