xonsh/xontrib/prompt_ret_code.xsh
Jean-Benoist Leger b6d339a825 ret_code should return None not '' when rc==0
When the return code is 0, nothing is displayed in the prompt. Empty string was
returned. Returning None have the same behavior, but allows user to use the
syntax "{retcode:{} }" in the prompt for different spacing.
2017-01-02 01:27:34 +01:00

34 lines
912 B
Text

from xonsh.tools import ON_WINDOWS as _ON_WINDOWS
def _ret_code_color():
if __xonsh_history__.rtns:
color = 'blue' if __xonsh_history__.rtns[-1] == 0 else 'red'
else:
color = 'blue'
if _ON_WINDOWS:
if color == 'blue':
return '{BOLD_INTENSE_CYAN}'
elif color == 'red':
return '{BOLD_INTENSE_RED}'
else:
if color == 'blue':
return '{BOLD_BLUE}'
elif color == 'red':
return '{BOLD_RED}'
def _ret_code():
if __xonsh_history__.rtns:
return_code = __xonsh_history__.rtns[-1]
if return_code != 0:
return '[{}]'.format(return_code)
return None
$PROMPT = $PROMPT.replace('{prompt_end}{NO_COLOR}',
'{ret_code_color}{ret_code}{prompt_end}{NO_COLOR}')
$PROMPT_FIELDS['ret_code_color'] = _ret_code_color
$PROMPT_FIELDS['ret_code'] = _ret_code