U Vc @sHdZddlZddlmZdZddZddZd d Zd d Z d dZ dS)zLoad plugin assets from disk.N)tfZpluginscCstjjtj||S)z2Helper that returns if parent/item is a directory.)riogfileisdirospathjoin)parentitemr Z/tmp/pip-unpacked-wheel-g8kmtpbc/tensorboard/backend/event_processing/plugin_asset_util.py _IsDirectorysr cCstj|t|S)z-Returns the plugin directory for plugin_name.)rrr _PLUGINS_DIR)logdir plugin_namer r r PluginDirectorysrcsPtj|tztjj}Wntjj k r<gYSXfdd|DS)abList all the plugins that have registered assets in logdir. If the plugins_dir does not exist, it returns an empty list. This maintains compatibility with old directories that have no plugins written. Args: logdir: A directory that was created by a TensorFlow events writer. Returns: a list of plugin names, as strings cs*g|]"}|dst|r|dqS/)endswithr rstrip.0xZ plugins_dirr r 6s zListPlugins..) rrrrrrrlistdirerrors NotFoundError)rentriesr rr ListPlugins#s   rcCsFt||}zddtjj|DWStjjk r@gYSXdS)aList all the assets that are available for given plugin in a logdir. Args: logdir: A directory that was created by a TensorFlow summary.FileWriter. plugin_name: A string name of a plugin to list assets for. Returns: A string list of available plugin assets. If the plugin subdirectory does not exist (either because the logdir doesn't exist, or because the plugin didn't register) an empty list is returned. cSsg|]}|dqSr)rrr r r rLszListAssets..N)rrrrrrr)rrZ plugin_dirr r r ListAssets=s r c Cstjt|||}z2tjj|d}|W5QRWSQRXWnXtj j k rjt d|Yn6tj j k r}zt d||fW5d}~XYnXdS)akRetrieve a particular plugin asset from a logdir. Args: logdir: A directory that was created by a TensorFlow summary.FileWriter. plugin_name: The plugin we want an asset from. asset_name: The name of the requested asset. Returns: string contents of the plugin asset. Raises: KeyError: if the asset does not exist. rzAsset path %s not foundz(Couldn't read asset path: %s, OpError %sN) rrrrrrrZGFilereadrrKeyErrorZOpError)rrZ asset_nameZ asset_pathfer r r RetrieveAssetQs  r&) __doc__os.pathrZtensorboard.compatrrr rrr r&r r r r s