Skip to content

Commit d92b1c1

Browse files
authored
fix ansi color reset bug (#2970)
2 parents 0652d36 + 3ff1a64 commit d92b1c1

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

src/_nebari/utils.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,7 @@ def process_streams(
6464

6565
outputs = {"stdout": [], "stderr": []}
6666
partial = {"stdout": b"", "stderr": b""}
67+
reset_code = b"\x1b[0m" # ANSI reset code
6768

6869
try:
6970
while True:
@@ -109,6 +110,15 @@ def process_streams(
109110
sys.stderr.flush()
110111
else:
111112
outputs["stderr"].append(line_w_newline)
113+
114+
# Add reset code when we're done processing output
115+
if print_stdout:
116+
sys.stdout.buffer.write(reset_code)
117+
sys.stdout.flush()
118+
if print_stderr:
119+
sys.stderr.buffer.write(reset_code)
120+
sys.stderr.flush()
121+
112122
finally:
113123
sel.close()
114124
if process.stdout:

0 commit comments

Comments
 (0)