Python environ.partial_format_prompt函数代码示例

def test_format_prompt_with_broken_template():
    for p in ('{user', '{user}{hostname'):
        assert partial_format_prompt(p) == p

    # '{{user' will be parsed to '{user'
    for p in ('{{user}', '{{user'):
        assert 'user' in partial_format_prompt(p)

def test_format_prompt_with_broken_template():
    for p in ('{user', '{user}{hostname'):
        assert_equal(partial_format_prompt(p), p)
        assert_equal(format_prompt(p), p)

    # '{{user' will be parsed to '{user'
    for p in ('{{user}', '{{user'):
        assert_in('user', partial_format_prompt(p))
        assert_in('user', format_prompt(p))

 def prompt(self):
     """Obtains the current prompt string."""
     global RL_LIB, RL_CAN_RESIZE
     if RL_CAN_RESIZE:
         # This is needed to support some system where line-wrapping doesn't
         # work. This is a bug in upstream Python, or possibly readline.
         RL_LIB.rl_reset_screen_size()
     if self.need_more_lines:
         if self.mlprompt is None:
             try:
                 self.mlprompt = multiline_prompt(curr=self._current_prompt)
             except Exception:  # pylint: disable=broad-except
                 print_exception()
                 self.mlprompt = '<multiline prompt error> '
         return self.mlprompt
     env = builtins.__xonsh_env__  # pylint: disable=no-member
     p = env.get('PROMPT')
     try:
         p = partial_format_prompt(p)
     except Exception:  # pylint: disable=broad-except
         print_exception()
     hide = True if self._force_hide is None else self._force_hide
     p = ansi_partial_color_format(p, style=env.get('XONSH_COLOR_STYLE'),
                                   hide=hide)
     self._current_prompt = p
     self.settitle()
     return p

def test_format_prompt_with_broken_template_in_func():
    for p in (
        lambda: '{user',
        lambda: '{{user',
        lambda: '{{user}',
        lambda: '{user}{hostname',
    ):
        # '{{user' will be parsed to '{user'
        assert 'user' in partial_format_prompt(p)

 def prompt_tokens(self, cli):
     """Returns a list of (token, str) tuples for the current prompt."""
     p = builtins.__xonsh_env__.get('PROMPT')
     try:
         p = partial_format_prompt(p)
     except Exception:  # pylint: disable=broad-except
         print_exception()
     toks = partial_color_tokenize(p)
     self.settitle()
     return toks

 def rprompt_tokens(self, cli):
     """Returns a list of (token, str) tuples for the current right
     prompt.
     """
     p = builtins.__xonsh_env__.get('RIGHT_PROMPT')
     if len(p) == 0:
         return []
     try:
         p = partial_format_prompt(p)
     except Exception:  # pylint: disable=broad-except
         print_exception()
     toks = partial_color_tokenize(p)
     return toks

 def rprompt_tokens(self, cli):
     """Returns a list of (token, str) tuples for the current right
     prompt.
     """
     p = builtins.__xonsh_env__.get('RIGHT_PROMPT')
     # partial_format_prompt does handle empty strings properly,
     # but this avoids descending into it in the common case of
     # $RIGHT_PROMPT == ''.
     if isinstance(p, str) and len(p) == 0:
         return []
     try:
         p = partial_format_prompt(p)
     except Exception:  # pylint: disable=broad-except
         print_exception()
     toks = partial_color_tokenize(p)
     return toks

def test_format_prompt():
    formatter_dict = {
        'a_string': 'cat',
        'none': (lambda: None),
        'f': (lambda: 'wakka'),
        }
    cases = {
        'my {a_string}': 'my cat',
        'my {none}{a_string}': 'my cat',
        '{f} jawaka': 'wakka jawaka',
        }
    for p, exp in cases.items():
        obs = format_prompt(template=p, formatter_dict=formatter_dict)
        yield assert_equal, exp, obs
    for p, exp in cases.items():
        obs = partial_format_prompt(template=p, formatter_dict=formatter_dict)
        yield assert_equal, exp, obs

 def prompt(self):
     """Obtains the current prompt string."""
     if self.need_more_lines:
         if self.mlprompt is None:
             try:
                 self.mlprompt = multiline_prompt()
             except Exception:  # pylint: disable=broad-except
                 print_exception()
                 self.mlprompt = '<multiline prompt error> '
         return self.mlprompt
     env = builtins.__xonsh_env__  # pylint: disable=no-member
     p = env.get('PROMPT')
     try:
         p = partial_format_prompt(p)
     except Exception:  # pylint: disable=broad-except
         print_exception()
     self.settitle()
     return p

 def settitle(self):
     """Sets terminal title."""
     _ = self
     env = builtins.__xonsh_env__  # pylint: disable=no-member
     term = env.get('TERM', None)
     # Shells running in emacs sets TERM to "dumb" or "eterm-color".
     # Do not set title for these to avoid garbled prompt.
     if (term is None and not ON_WINDOWS) or term in ['dumb', 'eterm-color',
                                                      'linux']:
         return
     t = env.get('TITLE')
     if t is None:
         return
     t = partial_format_prompt(t)
     if ON_WINDOWS and 'ANSICON' not in env:
         t = escape_windows_cmd_string(t)
         os.system('title {}'.format(t))
     else:
         os.write(1, "\x1b]2;{0}\x07".format(t).encode())

 def settitle(self):
     """Sets terminal title."""
     env = builtins.__xonsh_env__  # pylint: disable=no-member
     term = env.get('TERM', None)
     # Shells running in emacs sets TERM to "dumb" or "eterm-color".
     # Do not set title for these to avoid garbled prompt.
     if (term is None and not ON_WINDOWS) or term in ['dumb', 'eterm-color',
                                                      'linux']:
         return
     t = env.get('TITLE')
     if t is None:
         return
     t = partial_format_prompt(t)
     if ON_WINDOWS and 'ANSICON' not in env:
         t = escape_windows_cmd_string(t)
         os.system('title {}'.format(t))
     else:
         with open(1, 'wb', closefd=False) as f:
             # prevent xonsh from answering interative questions
             # on the next command by writing the title
             f.write("\x1b]2;{0}\x07".format(t).encode())
             f.flush()

def test_format_prompt_with_invalid_func():
    def p():
        foo = bar  # raises exception
        return '{user}'
    assert_is_instance(partial_format_prompt(p), str)
    assert_is_instance(format_prompt(p), str)

def test_format_prompt_with_format_spec(inp, exp, formatter_dict):
    obs = partial_format_prompt(template=inp, formatter_dict=formatter_dict)
    assert exp == obs