Skip to content

Commit bd2168a

Browse files
APX103mmmay0722
authored andcommitted
[fix] Fix long log display error
1 parent 2420cfd commit bd2168a

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

webqa_agent/utils/task_display_util.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import asyncio
22
import logging
3+
import re
34
import os
45
import sys
56
import time
@@ -39,6 +40,10 @@ def __exit__(self, exc_type, exc, tb):
3940
TaskInfo(name=self.name, start=self.start_time, end=end_time, error=error))
4041
return False
4142

43+
def remove_ansi_escape_sequences(text):
44+
ansi_escape = re.compile(r'\x1B[@-_][0-?]*[ -/]*[@-~]')
45+
return ansi_escape.sub('', text)
46+
4247

4348
class Display:
4449
display = None
@@ -124,7 +129,7 @@ def _render_frame(self):
124129
length = min(self.num_log, len(lines))
125130
for ln in range(length):
126131
line = lines[-length + ln]
127-
if len(line) >= col:
132+
if len(remove_ansi_escape_sequences(str(line))) >= col:
128133
out.write("... (to long) \n")
129134
else:
130135
out.write(line + "\n")

0 commit comments

Comments
 (0)