From 7351ddb52a2162fc2ac303c687a2ed50cbb0843e Mon Sep 17 00:00:00 2001
From: imarom <imarom@cisco.com>
Date: Wed, 27 Jan 2016 10:47:40 -0500
Subject: API: example for bi-directional flows

---
 .../client/trex_stateless_client.py                   | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

(limited to 'scripts/automation/trex_control_plane/client/trex_stateless_client.py')

diff --git a/scripts/automation/trex_control_plane/client/trex_stateless_client.py b/scripts/automation/trex_control_plane/client/trex_stateless_client.py
index 65e69938..3d4dbc93 100755
--- a/scripts/automation/trex_control_plane/client/trex_stateless_client.py
+++ b/scripts/automation/trex_control_plane/client/trex_stateless_client.py
@@ -110,6 +110,10 @@ class LoggerApi(object):
 
 # default logger - to stdout
 class DefaultLogger(LoggerApi):
+
+    def __init__ (self):
+        super(DefaultLogger, self).__init__()
+
     def write (self, msg, newline = True):
         if newline:
             print msg
@@ -972,17 +976,22 @@ class STLClient(object):
         Sets verbose level
 
         :parameters:
-            level : enum
-                LoggerApi.VERBOSE_QUIET
-                LoggerApi.VERBOSE_NORMAL
-                LoggerApi.VERBOSE_HIGH
+            level : str
+                "high"
+                "low"
+                "normal"
 
         :raises:
             None
 
     """
     def set_verbose (self, level):
-        self.logger.set_verbose(level)
+        modes = {'low' : LoggerApi.VERBOSE_QUIET, 'normal': LoggerApi.VERBOSE_REGULAR, 'high': LoggerApi.VERBOSE_HIGH}
+
+        if not level in modes.keys():
+            raise STLArgumentError('level', level)
+
+        self.logger.set_verbose(modes[level])
 
 
     """
-- 
cgit 1.2.3-korg