1Vh?jdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl m Z ddlmZej dGdd Zd Zd Zd Zd ZdZdZdZdZdZej dGddZej dGddZej dGddZej dGddZej dGddZej@dfdZ!d Z"d!Z#y)"z>Private utilities for managing multiple TensorBoard processes.N)Optional)version) tb_loggingT)frozencbeZdZUdZeed<eed<eed<eed<eed<eed<eed<eed <y ) TensorBoardInfoaHolds the information about a running TensorBoard instance. Attributes: version: Version of the running TensorBoard. start_time: Seconds since epoch. pid: ID of the process running TensorBoard. port: Port on which TensorBoard is running. path_prefix: Relative prefix to the path, may be empty. logdir: Data location used by the TensorBoard server, may be empty. db: Database connection used by the TensorBoard server, may be empty. cache_key: Opaque, as given by `cache_key` below. r start_timepidport path_prefixlogdirdb cache_keyN)__name__ __module__ __qualname____doc__str__annotations__intC/home/dcms/DCMS/lib/python3.12/site-packages/tensorboard/manager.pyrr#s2 LO H I K GNrrcV|jrd|jzSd|jzS)zFormat the data location for the given TensorBoardInfo. Args: info: A TensorBoardInfo value. Returns: A human-readable string describing the logdir or database connection used by the server: e.g., "logdir /tmp/logs". zdb %sz logdir %s)rr infos rdata_source_from_infor<s) ww  T[[((rc tjt}|jD]9\}}t t |||rt d|d|dt |||jtjk7r)t dtjd|jtj|}tj|ddS)aConvert a `TensorBoardInfo` to string form to be stored on disk. The format returned by this function is opaque and should only be interpreted by `_info_from_string`. Args: info: A valid `TensorBoardInfo` object. Raises: ValueError: If any field on `info` is not of the correct type. Returns: A string representation of the provided `TensorBoardInfo`. expected of type , but found: zexpected 'version' to be T) sort_keysindent) typingget_type_hintsritems isinstancegetattr ValueErrorrVERSION dataclassesasdictjsondumps)rfield_name_to_typekey field_type json_values r_info_to_stringr4Ls ..?-335Z'$,j9 GD#$68   ||w&  .  ##D)J ::jD ;;rc tjt} tj|}t |tst d|t|j}t|}||z }|rt dt|i}|jD]3\}}t |||st d|d|d||||||<5tdi|S#t $rt d|wxYw)aParse a `TensorBoardInfo` object from its string representation. Args: info_string: A string representation of a `TensorBoardInfo`, as produced by a previous call to `_info_to_string`. Returns: A `TensorBoardInfo` value. Raises: ValueError: If the provided string is not valid JSON, or if it is missing any required fields, or if any field is of incorrect type. zinvalid JSON: znot a JSON object: zTensorBoardInfo missing keys: rr r!r) r%r&rr.loadsr*r(dict frozensetkeyssortedr') info_stringr0r3 expected_keys actual_keys missing_keysfieldsr1r2s r_info_from_stringr@ks ..?>ZZ ,  j$ 'J@AA05578MJ'K ;.L282F H  F-335&Z*S/:6 JsO5 !os &  $V $$/ >{<==>s C##C;ct|ttfstd|d|||d}t j t j|ddjd}t|jdS) aCompute a `TensorBoardInfo.cache_key` field. The format returned by this function is opaque. Clients may only inspect it by comparing it for equality with other results from this function. Args: working_directory: The directory from which TensorBoard was launched and relative to which paths like `--logdir` and `--db` are resolved. arguments: The command-line args to TensorBoard, as `sys.argv[1:]`. Should be a list (or tuple), not an unparsed string. If you have a raw shell command, use `shlex.split` before passing it to this function. configure_kwargs: A dictionary of additional argument values to override the textual `arguments`, with the same semantics as in `tensorboard.program.TensorBoard.configure`. May be an empty dictionary. Returns: A string such that if two (prospective or actual) TensorBoard invocations have the same cache key then it is safe to use one in place of the other. The converse is not guaranteed: it is often safe to change the order of TensorBoard arguments, or to explicitly set them to their default values, or to move them between `arguments` and `configure_kwargs`, but such invocations may yield distinct cache keys. z6'arguments' should be a list of arguments, but found: z& (use `shlex.split` if given a string)working_directory argumentsconfigure_kwargsT),:)r# separatorszutf-8ascii) r(listtuple TypeErrorbase64 b64encoder.r/encoderdecode)rCrDrEdatumraws rrrs}: i$ /7@ C  /, E    5DZ@GGP C szz'" ##rcltjjtjd} tj |tj |d|S#t$rI}|jtjk(r tjj|rnYd}~|Sd}~wwxYw)aEGet path to directory in which to store info files. The directory returned by this function is "owned" by this module. If the contents of the directory are modified other than via the public functions of this module, subsequent behavior is undefined. The directory will be created if it does not exist. z.tensorboard-infoiN) ospathjointempfile gettempdirmakedirschmodOSErrorerrnoEEXISTisdir)rUes r _get_info_dirr`s 77<<++-/B CD D u K  77ell "rww}}T':    KsA!! B3*>B..B3c|tjjtdtjzS)zGet path to info file for the current process. As with `_get_info_dir`, the info directory will be created if it does not exist. z pid-%d.info)rTrUrVr`getpidrrr_get_info_file_pathrcs& 77<< )D EErcdt|z}ttd5}|j|dddy#1swYyxYw)aYWrite TensorBoardInfo to the current process's info file. This should be called by `main` once the server is ready. When the server shuts down, `remove_info_file` should be called. Args: tensorboard_info: A valid `TensorBoardInfo` object. Raises: ValueError: If any field on `info` is not of the correct type. z%s wN)r4openrcwrite)tensorboard_infopayloadoutfiles rwrite_info_filerksE'788G !#S )W gs >Ac tjty#t$r)}|jtj k(rnYd}~yd}~wwxYw)zRemove the current process's TensorBoardInfo file, if it exists. If the file does not exist, no action is taken and no error is raised. N)rTunlinkrcr[r\ENOENT)r_s rremove_info_filerosD  %'(  77ell "    s AA  Act}g}tj|D]d}tjj ||} t |5}|j }ddd t}|j|f|S#1swY*xYw#t$r(}|jtjk(rYd}~d}~wwxYw#t$r)tjjd|dYwxYw)aFReturn TensorBoardInfo values for running TensorBoard processes. This function may not provide a perfect snapshot of the set of running processes. Its result set may be incomplete if the user has cleaned their /tmp/ directory while TensorBoard processes are running. It may contain extraneous entries if TensorBoard processes exited uncleanly (e.g., with SIGKILL or SIGQUIT). Entries in the info directory that do not represent valid `TensorBoardInfo` values will be silently ignored. Returns: A fresh list of `TensorBoardInfo` objects. Nzinvalid info file: %rT)exc_info)r`rTlistdirrUrVrfreadIOErrorr\EACCESr@appendr*r get_loggerdebug)info_dirresultsfilenamefilepathinfilecontentsr_rs rget_allrsHGJJx(!77<<(3 h )6!;;= ) !$X.D NN4 -!. N) ) ) ww%,,&     ! ! # ) )' *  sH BB "B+ C  B B C CCC /C=<C=ceZdZUdZeed<y) StartReusedaPossible return value of the `start` function. Indicates that a call to `start` was compatible with an existing TensorBoard process, which can be reused according to the provided info. Attributes: info: A `TensorBoardInfo` object. rNrrrrrrrrrrr1s rrceZdZUdZeed<y) StartLaunchedzPossible return value of the `start` function. Indicates that a call to `start` successfully launched a new TensorBoard process, which is available with the provided info. Attributes: info: A `TensorBoardInfo` object. rNrrrrrr@s rrc<eZdZUdZeed<eeed<eeed<y) StartFailedaTPossible return value of the `start` function. Indicates that a call to `start` tried to launch a new TensorBoard instance, but the subprocess exited with the given exit code and output streams. (If the contents of the output streams are no longer available---e.g., because the user has emptied /tmp/---then the corresponding values will be `None`.) Attributes: exit_code: As `Popen.returncode` (negative for signal). stdout: Error message to stdout if the stream could not be read. stderr: Error message to stderr if the stream could not be read. exit_codestdoutstderrN)rrrrrrrrrrrrrNs" N SM SMrrc,eZdZUdZeed<eeed<y)StartExecFailedaPossible return value of the `start` function. Indicates that a call to `start` failed to invoke the subprocess. Attributes: os_error: `OSError` due to `Popen` invocation. explicit_binary: If the TensorBoard executable was chosen via the `TENSORBOARD_BINARY` environment variable, then this field contains the path to that binary; otherwise `None`. os_errorexplicit_binaryN)rrrrr[rrrrrrrrcs c]"rrceZdZUdZeed<y) StartTimedOuta\Possible return value of the `start` function. Indicates that a call to `start` launched a TensorBoard process, but that process neither exited nor wrote its info file within the allowed timeout period. The process may still be running under the included PID. Attributes: pid: ID of the process running TensorBoard. r N)rrrrrrrrrrrts  Hrr<)secondscDttj|i}t|}|r t |St j d\}}t j d\}}tj}tjjdd} tj| dn| g|z|| } tj|tj|d } ||jz} tj| krxtj | | j#}| t%|t'|t'| St|}|r t)|Stj| krxt+| j, S#t$rA} t| | cYd} ~ tj|tj|Sd} ~ wwxYw#tj|tj|wxYw)aStart a new TensorBoard instance, or reuse a compatible one. If the cache key determined by the provided arguments and the current working directory (see `cache_key`) matches the cache key of a running TensorBoard process (see `get_all`), that process will be reused. Otherwise, a new TensorBoard process will be spawned with the provided arguments, using the `tensorboard` binary from the system path. Args: arguments: List of strings to be passed as arguments to `tensorboard`. (If you have a raw command-line string, see `shlex.split`.) timeout: `datetime.timedelta` object describing how long to wait for the subprocess to initialize a TensorBoard server and write its `TensorBoardInfo` file. If the info file is not written within this time period, `start` will assume that the subprocess is stuck in a bad state, and will give up on waiting for it and return a `StartTimedOut` result. Note that in such a case the subprocess will not be killed. Default value is 60 seconds. Returns: A `StartReused`, `StartLaunched`, `StartFailed`, or `StartTimedOut` object. rBrz.tensorboard-stdout-)prefixz.tensorboard-stderr-TENSORBOARD_BINARYN tensorboard)rr)rrg?)rrr)r )rrTgetcwd_find_matching_instancerrWmkstemptimeenvironget subprocessPopenr[rclose total_secondssleeppollr_maybe_read_filerrr )rDtimeoutthis_cache_keymatch stdout_fd stdout_path stderr_fd stderr_pathstart_time_seconds explicit_tbpr_poll_interval_secondsend_time_secondssubprocess_resultrs rstartrs4))+N $N 3E &&'//7MNY '//7MNY **..!5tz)_find_matching_instance..s affr)r1N)rrr:)rinfosr candidates candidates rrrsQ IE#(H4DNNi,G$HJHJ,<=   Is AAc t|5}|jcdddS#1swYyxYw#t$r,}|jtjk(rYd}~yYd}~yd}~wwxYw)zRead the given file, if it exists. Args: filename: A path to a file. Returns: A string containing the file contents, or `None` if the file does not exist. N)rfrsrtr\rn)r{r}r_s rrrsU (^ !v;;= ! ! !  77ell " #s( 3' 3033 A(A##A()$rrMr,datetimer\r.rTrrWrr%rrrtensorboard.utilr dataclassrrr4r@rr`rcrkrorrrrrr timedeltarrrrrrrslE  'd#$0 ) <>(%V,$^,F""(Vd#  $ d#  $ d#$(d# # #$ # d#   $  0X//;D(N r