Merge branch 'master-demo'

1 files changed, 109 insertions, 1 deletions

diff --git a/scripts/automation/trex_control_plane/console/trex_console.py b/scripts/automation/trex_control_plane/console/trex_console.py
index e9d8ad84..27a5eeab 100755
--- a/scripts/automation/trex_control_plane/console/trex_console.py
+++ b/scripts/automation/trex_control_plane/console/trex_console.py
@@ -1,5 +1,22 @@
 #!/usr/bin/env python
-# -*- coding: utf-8 -*- 
+# -*- coding: utf-8 -*-
+
+"""
+Dan Klein, Itay Marom
+Cisco Systems, Inc.
+
+Copyright (c) 2015-2015 Cisco Systems, Inc.
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+    http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
 import cmd
 import json
 import ast
@@ -10,10 +27,14 @@ import string
 import sys
 import tty, termios
 import trex_root_path
+from common.trex_streams import *
 
 
 from client_utils.jsonrpc_client import TrexStatelessClient
 import trex_status
+from collections import namedtuple
+
+LoadedStreamList = namedtuple('LoadedStreamList', ['loaded', 'compiled'])
 
 #
 
@@ -114,6 +135,8 @@ class TrexConsole(cmd.Cmd):
         self.verbose = False
 
         self.postcmd(False, "")
+
+        self.user_streams = {}
       
 
     # a cool hack - i stole this function and added space
@@ -404,6 +427,91 @@ class TrexConsole(cmd.Cmd):
 
             print "{:<30} {:<30}".format(cmd + " - ", help)
 
+    def do_load_stream_list(self, line):
+        '''Loads a YAML stream list serialization into user console \n'''
+        args = line.split()
+        if args >= 2:
+            name = args[0]
+            yaml_path = args[1]
+            try:
+                multiplier = args[2]
+            except IndexError:
+                multiplier = 1
+            stream_list = CStreamList()
+            loaded_obj = stream_list.load_yaml(yaml_path, multiplier)
+            # print self.rpc_client.pretty_json(json.dumps(loaded_obj))
+            if name in self.user_streams:
+                print "Picked name already exist. Please pick another name."
+            else:
+                try:
+                    compiled_streams = stream_list.compile_streams()
+                    self.user_streams[name] = LoadedStreamList(loaded_obj,
+                                                               [StreamPack(v.stream_id, v.stream.dump_compiled())
+                                                                for k, v in compiled_streams.items()])
+
+                    print "Stream '{0}' loaded successfully".format(name)
+                except Exception as e:
+                    raise
+            return
+        else:
+            print "please provide load name and YAML path, separated by space.\n" \
+                  "Optionally, you may provide a third argument to specify multiplier."
+
+    def do_show_stream_list(self, line):
+        '''Shows the loaded stream list named [name] \n'''
+        args = line.split()
+        if args:
+            list_name = args[0]
+            try:
+                stream = self.user_streams[list_name]
+                if len(args) >= 2 and args[1] == "full":
+                    print self.rpc_client.pretty_json(json.dumps(stream.compiled))
+                else:
+                    print self.rpc_client.pretty_json(json.dumps(stream.loaded))
+            except KeyError as e:
+                print "Unknown stream list name provided"
+        else:
+            print "\nAvailable stream lists:\n{0}".format(', '.join([x
+                                                                  for x in self.user_streams.keys()]))
+
+    def complete_show_stream_list (self, text, line, begidx, endidx):
+        return [x
+                for x in self.user_streams.keys()
+                if x.startswith(text)]
+
+    def do_attach(self, line):
+        args = line.split()
+        if len(args) >= 1:
+            try:
+                stream_list = self.user_streams[args[0]]
+                port_list = self.parse_ports_from_line(' '.join(args[1:]))
+                owned = set(self.rpc_client.get_owned_ports())
+                if set(port_list).issubset(owned):
+                    rc, resp_list = self.rpc_client.add_stream(port_list, stream_list.compiled)
+                    if not rc:
+                        print "\n*** " + resp_list + "\n"
+                        return
+                else:
+                    print "Not all desired ports are aquired.\n" \
+                          "Acquired ports are: {acq}\n" \
+                          "Requested ports:    {req}\n" \
+                          "Missing ports:      {miss}".format(acq=list(owned),
+                                                              req=port_list,
+                                                              miss=list(set(port_list).difference(owned)))
+            except KeyError as e:
+                cause = e.args[0]
+                print "Provided stream list name '{0}' doesn't exists.".format(cause)
+        else:
+            print "Please provide list name and ports to attach to, or leave empty to attach to all ports."
+
+
+
+
+
+
+
+
+
 
 
     # adds a very simple stream