[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