a d=ic$@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 q/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tensorboard/uploader/server_info.py_server_info_requestJs r c Csd|}t|}|}td|z tj||tddtjid}Wn2tj yv}zt d|WYd}~n d}~00|j st d|j |j |jfztj|jWStjy}zt d ||jfWYd}~n d}~00dS) 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_infoZs2  " r!cCs^t}tj|j_||j_|j}d}||vr6d|}q$d||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&|drt|jjSttjfSdS)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&N)HasField frozensetr&r'scalars_metadataZ PLUGIN_NAME)r,r r r r's  r'cCs~|dr|j}nt}|js&t|_|js2t|_|js>t |_|j sJt |_ |j sVt |_ |jsbt|_|jsnt|_|jszt|_|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,     +(