o e$@sdZddlmZddlZddlmZddlmZddlm Z ddl m Z dZ d Zd Zd Zd Zd Zd ZdZdZddZddZddZddZddZddZGdddeZdS)z=Initial server communication to determine session parameters.)messageN)logging)version)metadata)server_info_pb2 iiiii<ii@cCs$t}tj|_||jjdd<|S)zGenerates a ServerInfoRequest Args: upload_plugins: List of plugin names requested by the user and to be verified by the server. Returns: A `server_info_pb2.ServerInfoRequest` message. N)rZServerInfoRequestrVERSIONZplugin_specificationupload_plugins)r requestr PD:\Projects\ConvertPro\env\Lib\site-packages\tensorboard/uploader/server_info.py_server_info_requestJs r c Csd|}t|}|}td|ztj||tddtjid}Wntj y5}zt d|d}~ww|j sEt d|j |j |jfztj|jWStjyc}z t d ||jfd}~ww) aFetches server info from a remote server. Args: origin: The server with which to communicate. Should be a string like "https://tensorboard.dev", including protocol, host, and (if needed) port. upload_plugins: List of plugins names requested by the user and to be verified by the server. Returns: A `server_info_pb2.ServerInfoResponse` message. Raises: CommunicationError: Upon failure to connect to or successfully communicate with the remote server. z%s/api/uploaderzRequested server info: <%r>z User-Agentztensorboard/%s)datatimeoutheadersz Failed to connect to backend: %sNz&Non-OK status from backend (%d %s): %rz&Corrupt response from backend (%s): %r)r ZSerializeToStringrinforequestspost_REQUEST_TIMEOUT_SECONDSrrRequestExceptionCommunicationErrorok status_codereasoncontentrServerInfoResponseZ FromStringr DecodeError)originr endpointZserver_info_requestZ post_bodyresponseer r r fetch_server_infoZs:     r!cCsdt}tj|j_||j_|j}d}||vrd|}||vsd||f|_||_ ||j j dd<|S)aManually creates server info given a frontend and backend. Args: frontend_origin: The origin of the TensorBoard.dev frontend, like "https://tensorboard.dev" or "http://localhost:8000". api_endpoint: As to `server_info_pb2.ApiServer.endpoint`. upload_plugins: List of plugin names requested by the user and to be verified by the server. Returns: A `server_info_pb2.ServerInfoResponse` message. z{{EID}}z{%s}z%s/experiment/%s/N) rrZ VERDICT_OK compatibilityZverdictZ api_serverr url_formattemplateid_placeholderplugin_controlallowed_plugins)Zfrontend_originZ api_endpointr resultr# placeholderr r r create_server_infos r*cCs|j}|j|j|S)aFormats a URL that will resolve to the provided experiment. Args: server_info: A `server_info_pb2.ServerInfoResponse` message. experiment_id: A string; the ID of the experiment to link to. Returns: A URL resolving to the given experiment, as a string. )r#r$replacer%) server_infoZ experiment_idr#r r r experiment_urls r-cCs"|dr t|jjSttjfS)a-Determines which plugins may upload data. This pulls from the `plugin_control` on the `server_info` when that submessage is set, else falls back to a default. Args: server_info: A `server_info_pb2.ServerInfoResponse` message. Returns: A `frozenset` of plugin names. r&)HasField frozensetr&r'scalars_metadataZ PLUGIN_NAME)r,r r r r's  r'cCs~|dr |j}nt}|jst|_|jst|_|jst |_|j s%t |_ |j s+t |_ |js1t|_|js7t|_|js=t|_|S)zReturns UploadLimits, from server_info if possible, otherwise from defaults. Args: server_info: A `server_info_pb2.ServerInfoResponse` message. Returns: An instance of UploadLimits. upload_limits)r.r1rZ UploadLimitsZmax_scalar_request_size _DEFAULT_MAX_SCALAR_REQUEST_SIZEZmax_tensor_request_size _DEFAULT_MAX_TENSOR_REQUEST_SIZEZmax_blob_request_size_DEFAULT_MAX_BLOB_REQUEST_SIZEZmin_scalar_request_interval$_DEFAULT_MIN_SCALAR_REQUEST_INTERVALZmin_tensor_request_interval$_DEFAULT_MIN_TENSOR_REQUEST_INTERVALZmin_blob_request_interval"_DEFAULT_MIN_BLOB_REQUEST_INTERVALZ max_blob_size_DEFAULT_MAX_BLOB_SIZEZmax_tensor_point_size_DEFAULT_MAX_TENSOR_POINT_SIZE)r,r1r r r r1s. r1c@seZdZdZdS)rz3Raised upon failure to communicate with the server.N)__name__ __module__ __qualname____doc__r r r r rsr)r=Zgoogle.protobufrrZabslrZ tensorboardrZtensorboard.plugins.scalarrr0Ztensorboard.uploader.protorrr5r6r7r2r3r4r8r9r r!r*r-r'r1 RuntimeErrorrr r r r s.     +(