def __prevent_duplicate_runs(self):
if self.action == 'start' and self.pidfile.is_locked():
print termstyle.green("Server daemon is already running")
exit(1)
elif self.action == 'stop' and not self.pidfile.is_locked():
print termstyle.green("Server daemon is not running")
exit(1)
def do_push_files(self, filepaths):
"""Pushes a custom file to be stored locally on TRex server.\
\nPush multiple files by specifying their path separated by ' ' (space)."""
try:
filepaths = filepaths.split(' ')
print termstyle.green("*** Starting pushing files ({trex_files}) to TRex. ***".format(
trex_files=', '.join(filepaths))
)
ret_val = self.trex.push_files(filepaths)
if ret_val:
print termstyle.green("*** End of TRex push_files method (success) ***")
else:
print termstyle.magenta("*** End of TRex push_files method (failed) ***")
except IOError as inst:
print termstyle.magenta(inst)
def __prompt_init_msg(self):
if self.action == 'start':
print termstyle.green("Starting daemon server...")
elif self.action == 'stop':
print termstyle.green("Stopping daemon server...")
class InteractiveStatelessTRex(cmd.Cmd):
intro = termstyle.green(
"\nInteractive shell to play with Cisco's TRex stateless API.\
\nType help to view available pre-defined scenarios\n(c) All rights reserved.\n"
)
prompt = '> '
def __init__(self, trex_host, trex_port, virtual, verbose):
cmd.Cmd.__init__(self)
self.verbose = verbose
self.virtual = virtual
self.trex = CTRexStatelessClient(trex_host, trex_port, self.virtual)
self.DEFAULT_RUN_PARAMS = dict(m=1.5,
nc=True,
p=True,
d=100,
f='avl/sfr_delay_10_1g.yaml',
l=1000)
self.run_params = dict(self.DEFAULT_RUN_PARAMS)
def do_transmit(self, line):
"""Transmits a request over using a given link to server.\
\nuse: transmit [method_name] [method_params]"""
if line == "":
print "\nUsage: [method name] [param dict as string]\n"
print "Example: rpc test_add {'x': 12, 'y': 17}\n"
return
args = line.split(' ', 1) # args will have max length of 2
method_name = args[0]
params = None
bad_parse = False
try:
params = ast.literal_eval(args[1])
if not isinstance(params, dict):
bad_parse = True
except ValueError as e1:
bad_parse = True
except SyntaxError as e2:
bad_parse = True
if bad_parse:
print "\nValue should be a valid dict: '{0}'".format(args[1])
print "\nUsage: [method name] [param dict as string]\n"
print "Example: rpc test_add {'x': 12, 'y': 17}\n"
return
response = self.trex.transmit(method_name, params)
if not self.virtual:
# expect response
rc, msg = response
if rc:
print "\nServer Response:\n\n" + json.dumps(msg) + "\n"
else:
print "\n*** " + msg + "\n"
def do_push_files(self, filepaths):
"""Pushes a custom file to be stored locally on TRex server.\
\nPush multiple files by specifying their path separated by ' ' (space)."""
try:
filepaths = filepaths.split(' ')
print termstyle.green(
"*** Starting pushing files ({trex_files}) to TRex. ***".
format(trex_files=', '.join(filepaths)))
ret_val = self.trex.push_files(filepaths)
if ret_val:
print termstyle.green(
"*** End of TRex push_files method (success) ***")
else:
print termstyle.magenta(
"*** End of TRex push_files method (failed) ***")
except IOError as inst:
print termstyle.magenta(inst)
