1 files changed, 80 insertions, 0 deletions
diff --git a/src/console/trex_console.py b/src/console/trex_console.py
new file mode 100755
index 00000000..404a2ee0
--- /dev/null
+++ b/src/console/trex_console.py
@@ -0,0 +1,80 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*- 
+import cmd
+from trex_rpc_client import RpcClient
+
+class TrexConsole(cmd.Cmd):
+    """Trex Console"""
+   
+    def __init__(self, rpc_client):
+        cmd.Cmd.__init__(self)
+        self.prompt = "TRex > "
+        self.intro  = "\n-=TRex Console V1.0=-\n" 
+        self.rpc_client = rpc_client
+        self.verbose = False
+
+    # a cool hack - i stole this function and added space
+    def completenames(self, text, *ignored):
+        dotext = 'do_'+text
+        return [a[3:]+' ' for a in self.get_names() if a.startswith(dotext)]
+
+    def do_verbose (self, line):
+        '''shows or set verbose mode\nusage: verbose [on/off]\n'''
+        if line == "":
+            print "verbose is " + ("on" if self.verbose else "off")
+        elif line == "on":
+            self.verbose = True
+            self.rpc_client.set_verbose(True)
+            print "verbose set to on\n"
+
+        elif line == "off":
+            self.verbose = False
+            self.rpc_client.set_verbose(False)
+            print "verbose set to off\n"
+        else:
+            print "please specify 'on' or 'off'\n"
+
+
+    def do_query_server(self, line):
+        '''\nquery the RPC server for supported remote commands\n'''
+        rc, msg = self.rpc_client.query_rpc_server()
+        if not rc:
+            print "\n*** Failed to query RPC server: " + str(msg)
+
+        print "RPC server supports the following commands: \n\n" + msg
+
+    def do_ping (self, line):
+        '''pings the RPC server \n'''
+        print "Pinging RPC server"
+
+        rc, msg = self.rpc_client.ping_rpc_server()
+        if rc:
+            print "[SUCCESS]\n"
+        else:
+            print "[FAILED]\n"
+
+    def do_quit(self, line):
+        '''\nexit the client\n'''
+        return True
+
+    def default(self, line):
+        print "'{0}' is an unrecognized command\n".format(line)
+
+    # aliasing
+    do_EOF = do_q = do_quit 
+
+def main ():
+    # RPC client
+    try:
+        rpc_client = RpcClient(5050)
+        rpc_client.connect()
+    except Exception as e:
+        print "\n*** " + str(e) + "\n"
+        exit(-1)
+
+    # console
+    console = TrexConsole(rpc_client)
+    console.cmdloop()
+
+if __name__ == '__main__':
+    main()