cntrl+c in middle of line can abort the line instead of exiting

1 files changed, 17 insertions, 3 deletions

diff --git a/scripts/automation/trex_control_plane/console/trex_console.py b/scripts/automation/trex_control_plane/console/trex_console.py
index fe909c09..37e33cb7 100755
--- a/scripts/automation/trex_control_plane/console/trex_console.py
+++ b/scripts/automation/trex_control_plane/console/trex_console.py
@@ -557,6 +557,20 @@ class TRexConsole(TRexGeneralCmd):
     
              print "{:<30} {:<30}".format(cmd + " - ", help)
 
+    # a custorm cmdloop wrapper
+    def start(self):
+        while True:
+            try:
+                self.cmdloop()
+                break
+            except KeyboardInterrupt as e:
+                 if not readline.get_line_buffer():
+                     raise KeyboardInterrupt
+                 else:
+                     print ""
+                     self.intro = None
+                     continue
+
     # aliases
     do_exit = do_EOF = do_q = do_quit
     do_h = do_history
@@ -650,11 +664,11 @@ def main():
         if options.tui:
             console.do_tui("")
         else:
-            console.cmdloop()
-
+            console.start()
+            
     except KeyboardInterrupt as e:
         print "\n\n*** Caught Ctrl + C... Exiting...\n\n"
-        
+
     finally:
         stateless_client.disconnect()