U
    Vc2,                     @   s@   d dl Z d dlmZ G dd deZG dd deZdd ZdS )	    N)write_service_pb2c                   @   s   e Zd ZdZdd ZdS )TensorBoardWriterServiceStub/Service for writing data to TensorBoard.dev.
  c                 C   s  |j dtjjtjjd| _|j dtjjtjjd| _	|j dtj
jtjjd| _|j dtjjtjjd| _|j dtjjtjjd| _|j dtjjtjjd| _|j dtjjtjjd| _|j d	tjjtjjd| _|jd
tjjtjjd| _|j dtj jtj!jd| _"dS )z;Constructor.

    Args:
      channel: A grpc.Channel.
    z>/tensorboard.service.TensorBoardWriterService/CreateExperiment)Zrequest_serializerZresponse_deserializerz>/tensorboard.service.TensorBoardWriterService/UpdateExperimentz>/tensorboard.service.TensorBoardWriterService/DeleteExperimentz7/tensorboard.service.TensorBoardWriterService/PurgeDataz9/tensorboard.service.TensorBoardWriterService/WriteScalarz9/tensorboard.service.TensorBoardWriterService/WriteTensorzE/tensorboard.service.TensorBoardWriterService/GetOrCreateBlobSequencez=/tensorboard.service.TensorBoardWriterService/GetBlobMetadataz7/tensorboard.service.TensorBoardWriterService/WriteBlobz;/tensorboard.service.TensorBoardWriterService/DeleteOwnUserN)#Zunary_unary:tensorboard_dot_uploader_dot_proto_dot_write__service__pb2CreateExperimentRequestSerializeToStringCreateExperimentResponse
FromStringCreateExperimentUpdateExperimentRequestUpdateExperimentResponseUpdateExperimentDeleteExperimentRequestDeleteExperimentResponseDeleteExperimentPurgeDataRequestPurgeDataResponse	PurgeDataWriteScalarRequestWriteScalarResponseWriteScalarWriteTensorRequestWriteTensorResponseWriteTensorGetOrCreateBlobSequenceRequestGetOrCreateBlobSequenceResponseGetOrCreateBlobSequenceGetBlobMetadataRequestGetBlobMetadataResponseGetBlobMetadataZstream_streamWriteBlobRequestWriteBlobResponse	WriteBlobDeleteOwnUserRequestDeleteOwnUserResponseDeleteOwnUser)selfZchannel r'   U/tmp/pip-unpacked-wheel-g8kmtpbc/tensorboard/uploader/proto/write_service_pb2_grpc.py__init__   sd    z%TensorBoardWriterServiceStub.__init__N)__name__
__module____qualname____doc__r)   r'   r'   r'   r(   r      s   r   c                   @   s`   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd ZdS ) TensorBoardWriterServiceServicerr   c                 C   s$   | tjj |d tddS )zERequest for a new location to write TensorBoard readable events.
    Method not implemented!NZset_codegrpcZ
StatusCodeZUNIMPLEMENTEDZset_detailsNotImplementedErrorr&   requestcontextr'   r'   r(   r
   I   s    
z1TensorBoardWriterServiceServicer.CreateExperimentc                 C   s$   | tjj |d tddS )z>Request to mutate metadata associated with an experiment.
    r/   Nr0   r3   r'   r'   r(   r   P   s    
z1TensorBoardWriterServiceServicer.UpdateExperimentc                 C   s$   | tjj |d tddS )zRequest that an experiment be deleted, along with all tags and scalars
    that it contains. This call may only be made by the original owner of the
    experiment.
    r/   Nr0   r3   r'   r'   r(   r   W   s    
z1TensorBoardWriterServiceServicer.DeleteExperimentc                 C   s$   | tjj |d tddS )z`Request that unreachable data be purged. Used only for testing;
    disabled in production.
    r/   Nr0   r3   r'   r'   r(   r   `   s    
z*TensorBoardWriterServiceServicer.PurgeDatac                 C   s$   | tjj |d tddS )zARequest additional scalar data be stored in TensorBoard.dev.
    r/   Nr0   r3   r'   r'   r(   r   h   s    
z,TensorBoardWriterServiceServicer.WriteScalarc                 C   s$   | tjj |d tddS )zARequest additional tensor data be stored in TensorBoard.dev.
    r/   Nr0   r3   r'   r'   r(   r   o   s    
z,TensorBoardWriterServiceServicer.WriteTensorc                 C   s$   | tjj |d tddS )zyRequest to obtain a specific BlobSequence entry, creating it if needed,
    to be subsequently populated with blobs.
    r/   Nr0   r3   r'   r'   r(   r   v   s    
z8TensorBoardWriterServiceServicer.GetOrCreateBlobSequencec                 C   s$   | tjj |d tddS )zoRequest the current status of blob data being stored in TensorBoard.dev,
    to support resumable uploads.
    r/   Nr0   r3   r'   r'   r(   r   ~   s    
z0TensorBoardWriterServiceServicer.GetBlobMetadatac                 C   s$   | tjj |d tddS )z?Request additional blob data be stored in TensorBoard.dev.
    r/   Nr0   )r&   Zrequest_iteratorr5   r'   r'   r(   r"      s    
z*TensorBoardWriterServiceServicer.WriteBlobc                 C   s$   | tjj |d tddS )zlRequest that the calling user and all their data be permanently deleted.
    Used for testing purposes.
    r/   Nr0   r3   r'   r'   r(   r%      s    
z.TensorBoardWriterServiceServicer.DeleteOwnUserN)r*   r+   r,   r-   r
   r   r   r   r   r   r   r   r"   r%   r'   r'   r'   r(   r.   E   s   	r.   c                 C   s  t j| jtjjtjjdt j| jtj	jtj
jdt j| jtjjtjjdt j| jtjjtjjdt j| jtjjtjjdt j| jtjjtjjdt j| jtjjtjjdt j| jtjjtjjdt j| jtjjtj jdt j| j!tj"jtj#jdd
}t $d|}|%|f d S )N)Zrequest_deserializerZresponse_serializer)
r
   r   r   r   r   r   r   r   r"   r%   z,tensorboard.service.TensorBoardWriterService)&r1   Zunary_unary_rpc_method_handlerr
   r   r   r	   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   Z stream_stream_rpc_method_handlerr"   r    r!   r%   r#   r$   Zmethod_handlers_generic_handlerZadd_generic_rpc_handlers)ZservicerserverZrpc_method_handlersZgeneric_handlerr'   r'   r(   .add_TensorBoardWriterServiceServicer_to_server   sp    4 r7   )r1   Ztensorboard.uploader.protor   r   objectr   r.   r7   r'   r'   r'   r(   <module>   s   >Q