diff options
Diffstat (limited to 'scripts/automation/trex_control_plane/console/trex_console.py')
| -rwxr-xr-x | scripts/automation/trex_control_plane/console/trex_console.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/scripts/automation/trex_control_plane/console/trex_console.py b/scripts/automation/trex_control_plane/console/trex_console.py index 49d58fe0..3e9dec47 100755 --- a/scripts/automation/trex_control_plane/console/trex_console.py +++ b/scripts/automation/trex_control_plane/console/trex_console.py @@ -116,6 +116,8 @@ class TRexConsole(TRexGeneralCmd): def __init__(self, stateless_client, verbose=False): self.stateless_client = stateless_client + self.stateless_client.set_prompt_redraw_cb(self.prompt_redraw) + TRexGeneralCmd.__init__(self) self.tui = trex_tui.TrexTUI(stateless_client) @@ -130,6 +132,9 @@ class TRexConsole(TRexGeneralCmd): ################### internal section ######################## + def prompt_redraw (self): + sys.stdout.write(self.prompt) + def verify_connected(f): @wraps(f) def wrap(*args): |