a d=ic? @sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z eddZeeeddd ddZeeeeedZed efd efd efd efdefdefdefdeffZedeZddZddZddZddZddZddZddZd d!Zd"d#Zed$d%Z ed&d%Z!ed'd(Z"ed)d*Z#ed+d,Z$ej%d-d.fd/d0Z&d1d2Z'd3d4Z(dS)5z>Private utilities for managing multiple TensorBoard processes.N)version) tb_logging _FieldType)serialized_type runtime_type serialize deserializecCs|SNnr r d/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tensorboard/manager.py2rcCs|Sr r r r r r r3rr start_timepidportZ path_prefixlogdirdb cache_keyTensorBoardInfocCs|jrd|jSd|jSdS)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 %sN)rrinfor r r data_source_from_infoOs  rcstD]8}t|}tt||jstd||jt|fqjtjkr^tdtjjffddtD}tj|dddS)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 %r of type %s, but found: %rz*expected 'version' to be %r, but found: %rcs"i|]}|t|t|qSr )_TENSORBOARD_INFO_FIELDSrgetattr).0krr r zsz#_info_to_string..T) sort_keysindent) r isinstancerr ValueErrorrVERSIONjsondumps)rkey field_type json_valuer rr _info_to_string_s$   r+cCszt|}Wn ty.td|fYn0t|tsHtd|ftt}t|}||}|rvtdt|fi}tD]F}t|}t|||jstd||j||f| ||||<q~t fi|S)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: %rznot a JSON object: %rz TensorBoardInfo missing keys: %rr) r&loadsr$r#dict frozensetrsortedrrr)Z info_stringr*Z expected_keysZ actual_keysZ missing_keysfieldsr(r)r r r _info_from_strings0   r1cCsRt|ttfstd|f|||d}ttj|dddd}t | dS)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. z^'arguments' should be a list of arguments, but found: %r (use `shlex.split` if given a string)working_directory argumentsconfigure_kwargsT),:)r! separatorszutf-8ascii) r#listtuple TypeErrorbase64 b64encoder&r'encodestrdecode)r3r4r5Zdatumrawr r r rsc Csrtjtd}zt|Wn@ty`}z(|jtjkrJtj |rJnWYd}~nd}~00t |d|S)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-infoNi) ospathjointempfile gettempdirmakedirsOSErrorerrnoEEXISTisdirchmod)rDer r r _get_info_dirs  rOcCstjtdtS)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)rCrDrErOgetpidr r r r _get_info_file_pathsrQcCsFdt|}ttd}||Wdn1s80YdS)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)r+openrQwrite)Ztensorboard_infopayloadoutfiler r r write_info_files rWc CsJzttWn4tyD}z|jtjkr.nWYd}~n d}~00dS)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)rCunlinkrQrIrJENOENT)rNr r r remove_info_file s  rZc Cst}g}t|D]}tj||}z4t|}|}Wdn1sN0YWn@ty}z(|jtj krWYd}~qnWYd}~n d}~00z t |}Wn&t yt jd|ddYq0||q|S)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)rOrClistdirrDrErSreadIOErrorrJEACCESr1r$rZ get_loggerdebugappend)info_dirresultsfilenamefilepathinfilecontentsrNrr r r get_alls* *     rh StartReusedr StartLaunched StartFailed exit_codestdoutstderrStartExecFailedZos_errorZexplicit_binary StartTimedOutr<)secondsc Csttt|id}t|}|r(t|dStjdd\}}tjdd\}}t}tj dd} zz&t j | durrdn| g|||d } WnFt y} z.t | | d WYd} ~ Wt|t|Sd} ~ 00Wt|t|nt|t|0d } ||} t| krht| | }|durLt|t|t|d St|}|r t|dSq t| jd S)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. r2rz.tensorboard-stdout-)prefixz.tensorboard-stderr-ZTENSORBOARD_BINARYN tensorboard)rnrorqg?rlrs)rrCgetcwd_find_matching_instancerirFmkstemptimeenvironget subprocessPopenrIrpclose total_secondssleeppollrk_maybe_read_filerjrrr)r4timeoutZthis_cache_keymatchZ stdout_fdZ stdout_pathZ stderr_fdZ stderr_pathZstart_time_secondsZ explicit_tbprNZpoll_interval_secondsZend_time_secondsZsubprocess_resultrr r r startvsT          rcs8t}fdd|D}t|dddD] }|SdS)zFind a running TensorBoard instance compatible with the cache key. Returns: A `TensorBoardInfo` object, or `None` if none matches the cache key. csg|]}|jkr|qSr r)rrrr r rz+_find_matching_instance..cSs|jSr )r)xr r r rrz)_find_matching_instance..)r(N)rhr/)rinfos candidates candidater rr rys ryc Cszz6t|}|WdWS1s*0YWn>tyt}z&|jtjkr`WYd}~dSWYd}~n d}~00dS)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)rSr]r^rJrY)rdrfrNr r r rs  , r))__doc__r= collectionsdatetimerJr&rCr~rFr{rwrZtensorboard.utilr namedtuplerintZ _type_intr@Z _type_str OrderedDictrrrr+r1rrOrQrWrZrhrirjrkrprr timedeltarryrr r r r s    ",/ 1   G