draft #2

1 files changed, 9 insertions, 3 deletions

diff --git a/scripts/automation/trex_control_plane/common/trex_types.py b/scripts/automation/trex_control_plane/common/trex_types.py
index 337f0a70..aada5bfc 100644
--- a/scripts/automation/trex_control_plane/common/trex_types.py
+++ b/scripts/automation/trex_control_plane/common/trex_types.py
@@ -14,16 +14,17 @@ class RpcResponseStatus(namedtuple('RpcResponseStatus', ['success', 'id', 'msg']
 # simple class to represent complex return value
 class RC():
 
-    def __init__ (self, rc = None, data = None):
+    def __init__ (self, rc = None, data = None, is_warn = False):
         self.rc_list = []
 
         if (rc != None):
-            tuple_rc = namedtuple('RC', ['rc', 'data'])
-            self.rc_list.append(tuple_rc(rc, data))
+            tuple_rc = namedtuple('RC', ['rc', 'data', 'is_warn'])
+            self.rc_list.append(tuple_rc(rc, data, is_warn))
 
     def __nonzero__ (self):
         return self.good()
 
+
     def add (self, rc):
         self.rc_list += rc.rc_list
 
@@ -33,6 +34,9 @@ class RC():
     def bad (self):
         return not self.good()
 
+    def warn (self):
+        return any([x.is_warn for x in self.rc_list])
+
     def data (self):
         d = [x.data if x.rc else "" for x in self.rc_list]
         return (d if len(d) != 1 else d[0])
@@ -83,3 +87,5 @@ def RC_OK(data = ""):
 def RC_ERR (err):
     return RC(False, err)
 
+def RC_WARN (warn):
+    return RC(True, warn, is_warn = True)