B )`#@shdZddlmZddlmZddlmZddlZddlmZddlmZddlm Z Gd d d e Z dS) zABase Class of TensorFlow Debugger (tfdbg) Command-Line Interface.)absolute_import)division)print_functionN) cli_config)command_parser)debugger_cli_commonc@sxeZdZdZdZddgZdZdZddd Zd d Z dd d Z ddZ dddZ ddZ ddZeddZdddZdS)BaseUIz#Base class of tfdbg user interface.ztfdbg> exitquitzERROR: zINFO: NcCs||_t|_t|_|jdg|jtjjgtjj |pHt |_ t jdt jd|_|j}|d}|jdtd|jdtd|d}|jd |j|jd gd d S) zConstructor of the base class. Args: on_ui_exit: (`Callable`) the callback to be called when the UI exits. config: An instance of `cli_config.CLIConfig()` carrying user-facing configurations. zconfig command) descriptionusageset property_name)typeproperty_valueshowconfigcfg)prefix_aliasesN)Z _on_ui_exitrZCommandHandlerRegistry_command_handler_registryZTabCompletionRegistry_tab_completion_registryregister_tab_comp_contextCLI_EXIT_COMMANDSZ HELP_COMMANDZHELP_COMMAND_ALIASESrZ CLIConfig_configargparseArgumentParserSUPPRESS_config_argparseradd_subparsers add_parser add_argumentstrregister_command_handler_config_command_handler format_help)selfZ on_ui_exitrZ subparsers set_parserr(R/home/dcms/DCMS/lib/python3.7/site-packages/tensorflow/python/debug/cli/base_ui.py__init__#s(       zBaseUI.__init__cCs|jj|ddS)zSet an introductory message to the help output of the command registry. Args: help_intro: (RichTextLines) Rich text lines appended to the beginning of the output of the command "help", as introductory information. ) help_introN)rset_help_intro)r&r+r(r(r)r,HszBaseUI.set_help_introcCs:|jj||||d|jd|g|r6|jd|dS)aA wrapper around CommandHandlerRegistry.register_command_handler(). In addition to calling the wrapped register_command_handler() method, this method also registers the top-level tab-completion context based on the command prefixes and their aliases. See the doc string of the wrapped method for more details on the args. Args: prefix: (str) command prefix. handler: (callable) command handler. help_info: (str) help information. prefix_aliases: (list of str) aliases of the command prefix. )rr N)rr#rZextend_comp_items)r&prefixhandlerZ help_inforr(r(r)r#Rs zBaseUI.register_command_handlercOs|jj||dS)zAWrapper around TabCompletionRegistry.register_tab_comp_context().N)rr)r&argskwargsr(r(r)rmsz BaseUI.register_tab_comp_contextTcCs tddS)aRun the UI until user- or command- triggered exit. Args: init_command: (str) Optional command to run on CLI start up. title: (str) Optional title to display in the CLI. title_color: (str) Optional color of the title, e.g., "yellow". enable_mouse_on_start: (bool) Whether the mouse mode is to be enabled on start-up. Returns: An exit token of arbitrary type. Can be None. z%run_ui() is not implemented in BaseUIN)NotImplementedError)r&Z init_commandtitleZ title_colorZenable_mouse_on_startr(r(r)run_uirsz BaseUI.run_uicCsD|}|sdgdfSt|}t|\}}|d|dd|fS)aaParse a command string into prefix and arguments. Args: command: (str) Command string to be parsed. Returns: prefix: (str) The command prefix. args: (list of str) The command arguments (i.e., not including the prefix). output_file_path: (str or None) The path to save the screen output to (if any). r Nr)stripr parse_commandZextract_output_file_path)r&commandZ command_itemsZoutput_file_pathr(r(r)_parse_commands    zBaseUI._parse_commandcCsr|}|sd}d}d}nN|d}t|dkrBd}|d}d}n&|d}|d}d|ddd}|||fS)aMAnalyze raw input to tab-completer. Args: text: (str) the full, raw input text to be tab-completed. Returns: context: (str) the context str. For example, If text == "print_tensor softmax", returns "print_tensor". If text == "print", returns "". If text == "", returns "". prefix: (str) the prefix to be tab-completed, from the last word. For example, if text == "print_tensor softmax", returns "softmax". If text == "print", returns "print". If text == "", returns "". except_last_word: (str) the input text, except the last word. For example, if text == "print_tensor softmax", returns "print_tensor". If text == "print_tensor -a softmax", returns "print_tensor -a". If text == "print", returns "". If text == "", returns "". r  r4rN)lstripsplitlenjoin)r&textcontextr-Zexcept_last_worditemsr(r(r)_analyze_tab_complete_inputs  z"BaseUI._analyze_tab_complete_inputcCs|jS)z/Obtain the CLIConfig of this `BaseUI` instance.)r)r&r(r(r)rsz BaseUI.configcCsR~|j|}t|drDt|drD|j|j|j|jj|jdS|jSdS)z)Command handler for the "config" command.rr)Z highlightN)r parse_argshasattrrrrrZ summarize)r&r/Z screen_infoparsedr(r(r)r$s  zBaseUI._config_command_handler)NN)N)NNNT)N)__name__ __module__ __qualname____doc__Z CLI_PROMPTrZERROR_MESSAGE_PREFIXZINFO_MESSAGE_PREFIXr*r,r#rr3r8rBpropertyrr$r(r(r(r)rs" %  * r) rI __future__rrrrZtensorflow.python.debug.clirrrobjectrr(r(r(r)s