B )`N@srdZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z e j Z Gdd d eZdS) zGConfigurations for TensorFlow Debugger (TFDBG) command-line interfaces.)absolute_import)division)print_functionN)debugger_cli_common)gfilec@s`eZdZdZdZddgZdddZdd Zd d Zd d Z ddZ ddZ dddZ ddZ dS) CLIConfigz?Client-facing configurations for TFDBG command-line interfaces.z .tfdbg_config)Zgraph_recursion_depth)Z mouse_modeTNcCsb|p ||_t|j|_t|jrP|}x| D]\}}||j|<q:W| i|_ dS)N) _default_config_file_path_config_file_path collections OrderedDict_DEFAULT_CONFIG_configrZExists_load_from_fileitems _save_to_file_set_callbacks)selfZconfig_file_pathconfigkeyvaluerU/home/dcms/DCMS/lib/python3.7/site-packages/tensorflow/python/debug/cli/cli_config.py__init__(s  zCLIConfig.__init__cCs ||jkrtd||j|S)Nz %s is not a valid property name.)rKeyError)r property_namerrrget4s  z CLIConfig.getcCs||jkrtd||j|}t|trpt|trf|dkrFd}qn|dkrXd}qntd|qt|}n8t|trt|}n$t|trt|}ntdt |||j|<| ||j kr|j ||jdS) aSet the value of a property. Supports limitd property value types: `bool`, `int` and `str`. Args: property_name: Name of the property. property_val: Value of the property. If the property has `bool` type and this argument has `str` type, the `str` value will be parsed as a `bool` Raises: ValueError: if a `str` property_value fails to be parsed as a `bool`. KeyError: if `property_name` is an invalid property name. z %s is not a valid property name.)1truetyesyonT)0falsefnonoffFz&Invalid string value for bool type: %szUnsupported property type: %sN) rr isinstanceboolstrlower ValueErrorint TypeErrortyperr)rrZ property_valZorig_valrrrset9s*               z CLIConfig.setcCs4||jkrtd|t|s&td||j|<dS)aSet a set-callback for given property. Args: property_name: Name of the property. callback: The callback as a `callable` of signature: def cbk(config): where config is the config after it is set to the new value. The callback is invoked each time the set() method is called with the matching property_name. Raises: KeyError: If property_name does not exist. TypeError: If `callback` is not callable. z %s is not a valid property name.z-The callback object provided is not callable.N)rrcallabler/r)rrcallbackrrr set_callbackcs   zCLIConfig.set_callbackcCstjtjd|jS)N~)ospathjoin expanduser_CONFIG_FILE_NAME)rrrrr xsz#CLIConfig._default_config_file_pathc CsFy,t|jd}t|j|WdQRXWntk r@YnXdS)Nw)rOpenr jsondumprIOError)r config_filerrrr{s zCLIConfig._save_to_filecCstddtdg}xj|jD]\\}}||kr2dnd}td}|t|d|g7}|td7}|tt||d7}||qWt|S) zGet a text summary of the config. Args: highlight: A property name to highlight in the output. Returns: A `RichTextLines` output. zCommand-line configuration:ZboldNz Z underlinez: )Z font_attr)RLrrr+appendrZ#rich_text_lines_from_rich_line_list)rZ highlightlinesnamevalZhighlight_attrlinerrr summarizes  zCLIConfig.summarizec CsryRt|jd:}t|}t}x t|D]}||||<q2W|SQRXWnt t fk rlt SXdS)Nr) rr<r r=loadr r sortedkeysr?r-dict)rr@Z config_dictrrrrrrs zCLIConfig._load_from_file)N)N)__name__ __module__ __qualname____doc__r:r rrr1r4r rrHrrrrrrs * r)rQ __future__rrrr r=r6Ztensorflow.python.debug.clirZtensorflow.python.platformrZRichLinerBobjectrrrrrs