B һd@sDdZddlZddlmZddlmZddlmZGdddeZdS)zABase Class of TensorFlow Debugger (tfdbg) Command-Line Interface.N) 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%U/var/www/html/venv/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)DszBaseUI.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. )rrN)rr rZextend_comp_items)r#prefixhandlerZ help_inforr%r%r&r Ns zBaseUI.register_command_handlercOs|jj||dS)zAWrapper around TabCompletionRegistry.register_tab_comp_context().N)rr)r#argskwargsr%r%r&risz 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_commandtitle title_colorZenable_mouse_on_startr%r%r&run_uinsz 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). rNr)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 r2rN)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.r r) highlightN)r parse_argshasattrrr r rZ 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 rr1r6r@propertyrr!r%r%r%r&rs" %  * r)rHrZtensorflow.python.debug.clirrrobjectrr%r%r%r&s