1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
from xonsh.tools import ON_WINDOWS as _ON_WINDOWS
import builtins
def _ret_code_color():
if builtins.__xonsh__.history.rtns:
color = "blue" if builtins.__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 builtins.__xonsh__.history.rtns:
return_code = builtins.__xonsh__.history.rtns[-1]
if return_code != 0:
return "[{}]".format(return_code)
return None
def _update():
env = builtins.__xonsh__.env
env["PROMPT"] = env["PROMPT"].replace(
"{prompt_end}{RESET}", "{ret_code_color}{ret_code}{prompt_end}{RESET}"
)
flds = env["PROMPT_FIELDS"]
flds["ret_code_color"] = _ret_code_color
flds["ret_code"] = _ret_code
# xontrib loads updates context
_update()
|
