a d=icQ*@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z ddlmZeZdZGdd d e jZGd d d e jZd d ZddZGdddeZGdddeZGdddZddZdS)z6Provides data ingestion logic backed by a gRPC server.N) grpc_provider)ingester)data_provider_pb2) tb_loggingZTENSORBOARD_DATA_SERVER_BINARYc@s,eZdZdZddZeddZddZdS) ExistingServerDataIngesterz*Connect to an already running gRPC server.cCst||}t|||_dS)zInitializes an ingester with the given configuration. Args: address: String, as passed to `--grpc_data_provider`. channel_creds_type: `grpc_util.ChannelCredsType`, as passed to `--grpc_creds_type`. N) _make_stubrGrpcDataProvider_data_provider)selfaddresschannel_creds_typestubrq/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tensorboard/data/server_ingester.py__init__+s z#ExistingServerDataIngester.__init__cCs|jSN)r r rrr data_provider6sz(ExistingServerDataIngester.data_providercCsdSrrrrrrstart:sz ExistingServerDataIngester.startN__name__ __module__ __qualname____doc__rpropertyrrrrrrr(s   rc@s4eZdZdZdddddZeddZdd ZdS) SubprocessServerDataIngesterz(Start a new data server as a subprocess.N)samples_per_plugin extra_flagscCs:||_d|_||_||_||_|p$i|_t|p0g|_dS)aDInitializes an ingester with the given configuration. Args: server_binary: `ServerBinary` to launch. logdir: String, as passed to `--logdir`. reload_interval: Number, as passed to `--reload_interval`. channel_creds_type: `grpc_util.ChannelCredsType`, as passed to `--grpc_creds_type`. samples_per_plugin: Dict[String, Int], as parsed from `--samples_per_plugin`. extra_flags: List of extra string flags to be passed to the data server without further interpretation. N)_server_binaryr _logdir_reload_interval_channel_creds_type_samples_per_pluginlist _extra_flags)r server_binaryZlogdirZreload_intervalr rrrrrrAs z%SubprocessServerDataIngester.__init__cCs|jdurtd|jS)NzMust call `start` first)r RuntimeErrorrrrrr`s z*SubprocessServerDataIngester.data_providerc Cs||jr dStjdd}tj|jd}tj|jd}|jdkrFd}ntt |j}dd|j D}d |}|j jd tj |jd |d |d d|fdg}|j dr|d|fttjr|dttjr|d||jtd|tj|tjd}|j|_d} tdt dD]} |!durpt"|pPd#} | shd|!} t$| td| t"|} td| | pd%drt | } qtdq2| durt$d|j&d| } t'| |j(}td|j&| t)*}z|j+|d d!WnJt,j-y^}z.d"| |f} t.d#| t$| |WYd}~n d}~00td$t/0| ||_dS)%NZtensorboard_data_server_)prefixportZ startup_errorroncecSs(g|] \}}d||dkrdn|fqS)z%s=%srallr).0kvrrr ssz6SubprocessServerDataIngester.start..,z --logdir=%sz --reload=%sz--samples-per-plugin=%sz--port=0z--port-file=%sz--die-after-stdinz0.5.0a0z--error-file=%sz --verbosezSpawning data server: %r)stding{Gz?z(exited with %d; check stderr for detailsz)Polling for data server port (attempt %d)zPort file contents: %r g?zTTimed out while waiting for data server to start. It may still be running as pid %d.z localhost:%dz.Opened channel to data server at pid %d via %s)timeoutz0Failed to communicate with data server at %s: %sz%sz#Got valid response from data server)1r tempfileTemporaryDirectoryospathjoinnamer strintr"itemsr expanduserrat_least_versionappendlogger isEnabledForloggingINFODEBUGextendr$info subprocessPopenPIPEr0Z _stdin_handletimesleeprangepoll_maybe_read_filestripDataServerStartupErrorendswithpidrr!rZGetExperimentRequestZ GetExperimentgrpcZRpcErrorwarningrr)r ZtmpdirZport_file_pathZerror_file_pathreloadZsample_hint_pairsrargspopenr(imsgZport_file_contentsaddrr reqerrrrfs                    z"SubprocessServerDataIngester.startrrrrrr>s   rc Cs|z6t|}|WdWS1s*0YWn@tyv}z(|jtjkr`WYd}~dSWYd}~n d}~00dS)z5Read a file, or return `None` on ENOENT specifically.N)openreadOSErrorerrnoENOENT)r9infiler^rrrrPs , rPcCs0|\}}|dtj|||d}t|S)N)zgrpc.max_receive_message_lengthi)options)Zchannel_configrArUZsecure_channelrZ make_stub)r\r credsreZchannelrrrrs  rc@s eZdZdS)NoDataServerErrorNrrrrrrrrgsrgc@s eZdZdS)rRNrhrrrrrRsrRc@s,eZdZdZddZeddZddZdS) ServerBinaryz'Information about a data server binary.cCs"||_|durt|n||_dS)atInitializes a `ServerBinary`. Args: path: String path to executable on disk. version: PEP 396-compliant version string, or `None` if unknown or not applicable. Binaries at unknown versions are assumed to be bleeding-edge: if you bring your own binary, it's on you to make sure that it's up to date. N)_path pkg_resources parse_version_version)r r9versionrrrrs  zServerBinary.__init__cCs|jSr)rjrrrrr9szServerBinary.pathcCs|jdurdS|jt|kS)aTest whether the binary's version is at least the given one. Useful for gating features that are available in the latest data server builds from head, but not yet released to PyPI. For example, if v0.4.0 is the latest published version, you can check `at_least_version("0.5.0a0")` to include both prereleases at head and the eventual final release of v0.5.0. If this binary's version was set to `None` at construction time, this method always returns `True`. Args: required_version: PEP 396-compliant version string. Returns: Boolean. NT)rmrkrl)r required_versionrrrr@s zServerBinary.at_least_versionN)rrrrrrr9r@rrrrris  ricCstjt}|rDtd|tj|s8tdt|ft |ddStj tj t dd}tj |rtd|t |ddSz ddl}WntyYn80|}|j}td|||durtd t ||Std dS) z5Get `ServerBinary` info or raise `NoDataServerError`.zServer binary (from env): %sz:Found environment variable %s=%s, but no such file exists.N)rnserverzServer binary (from bundle): %srz+Server binary (from Python package v%s): %sz7TensorBoard data server not supported on this platform.z{TensorBoard data server not found. This mode is experimental. If building from source, pass --define=link_data_server=true.)r8environget_ENV_DATA_SERVER_BINARYrDrHr9isfilergrir:dirname__file__existstensorboard_data_server ImportErrorr% __version__)Z env_resultZ bundle_resultrxZ pkg_resultrnrrrget_server_binary s>          r{)rrbrDr8rIr6rLrUrkZtensorboard.datarrZtensorboard.data.protorZtensorboard.utilrZ get_loggerrBrsZ DataIngesterrrrPrr&rgrRrir{rrrrs.     /