[dts] [Patch V1] framework/texttable.py: Add the class of supporting trex sent this patch on behalf of chen hongli
wang fei
feix.y.wang at intel.com
Wed Sep 20 14:13:51 CEST 2017
Signed-off-by: wang fei <feix.y.wang at intel.com>
---
framework/texttable.py | 29 +++++++++++++++++++++++++++++
1 file changed, 29 insertions(+)
diff --git a/framework/texttable.py b/framework/texttable.py
index 2f37a0c..f423bf7 100644
--- a/framework/texttable.py
+++ b/framework/texttable.py
@@ -92,6 +92,7 @@ Brian Peterson:
import sys
import string
+import re
from functools import reduce
try:
@@ -122,6 +123,34 @@ def len(iterable):
except:
return iterable.__len__()
+TEXT_CODES = {'bold': {'start': '\x1b[1m',
+ 'end': '\x1b[22m'},
+ 'cyan': {'start': '\x1b[36m',
+ 'end': '\x1b[39m'},
+ 'blue': {'start': '\x1b[34m',
+ 'end': '\x1b[39m'},
+ 'red': {'start': '\x1b[31m',
+ 'end': '\x1b[39m'},
+ 'magenta': {'start': '\x1b[35m',
+ 'end': '\x1b[39m'},
+ 'green': {'start': '\x1b[32m',
+ 'end': '\x1b[39m'},
+ 'yellow': {'start': '\x1b[33m',
+ 'end': '\x1b[39m'},
+ 'underline': {'start': '\x1b[4m',
+ 'end': '\x1b[24m'}}
+
+class TextCodesStripper:
+ keys = [re.escape(v['start']) for k,v in TEXT_CODES.items()]
+ keys += [re.escape(v['end']) for k,v in TEXT_CODES.items()]
+ pattern = re.compile("|".join(keys))
+
+ @staticmethod
+ def strip (s):
+ return re.sub(TextCodesStripper.pattern, '', s)
+
+def ansi_len (iterable):
+ return len(TextCodesStripper.strip(iterable))
class ArraySizeError(Exception):
--
2.7.4
More information about the dts
mailing list