def do_start_and_return (self, line):
     """Start T-Rex run and once in 'Running' mode, return to cmd prompt"""
     print termstyle.green("*** Starting T-Rex run, wait until in 'Running' state ***")
     try:
         ret = self.trex.start_trex(**self.run_params)
         print termstyle.green("*** End of scenario (T-Rex is probably still running!) ***")
     except TRexException as inst:
         print termstyle.red(inst)
 def do_stop_trex (self, line):
     """Try to stop T-Rex run (if T-Rex is currently running)"""
     print termstyle.green("*** Starting T-Rex termination ***")
     try:
         ret = self.trex.stop_trex()
         print termstyle.green("*** End of scenario (T-Rex is not running now) ***")
     except TRexException as inst:
         print termstyle.red(inst)
 def do_stop_trex(self, line):
     """Try to stop T-Rex run (if T-Rex is currently running)"""
     print termstyle.green("*** Starting T-Rex termination ***")
     try:
         ret = self.trex.stop_trex()
         print termstyle.green(
             "*** End of scenario (T-Rex is not running now) ***")
     except TRexException as inst:
         print termstyle.red(inst)
 def do_kill_indiscriminately (self, line):
     """Force killing of running T-Rex process (if exists) on the server."""
     print termstyle.green("*** Starting T-Rex termination ***")
     ret = self.trex.force_kill()
     if ret:
         print termstyle.green("*** End of scenario (T-Rex is not running now) ***")
     elif ret is None:
         print termstyle.magenta("*** End of scenario (T-Rex termination aborted) ***")
     else:
         print termstyle.red("*** End of scenario (T-Rex termination failed) ***")
 def do_start_and_return(self, line):
     """Start T-Rex run and once in 'Running' mode, return to cmd prompt"""
     print termstyle.green(
         "*** Starting T-Rex run, wait until in 'Running' state ***")
     try:
         ret = self.trex.start_trex(**self.run_params)
         print termstyle.green(
             "*** End of scenario (T-Rex is probably still running!) ***")
     except TRexException as inst:
         print termstyle.red(inst)
 def do_cancel_reservation (self, user):
     """Cancels a current reservation of T-Rex to a certain user"""
     try:
         if not user:
             ret = self.trex.cancel_reservation()
         else:
             ret = self.trex.cancel_reservation(user.split(' ')[0])
         print termstyle.green("*** T-Rex reservation canceled successfully ***")
     except TRexException as inst:
         print termstyle.red(inst)
 def do_reserve_trex(self, user):
     """Reserves the usage of T-Rex to a certain user"""
     try:
         if not user:
             ret = self.trex.reserve_trex()
         else:
             ret = self.trex.reserve_trex(user.split(' ')[0])
         print termstyle.green("*** T-Rex reserved successfully ***")
     except TRexException as inst:
         print termstyle.red(inst)
 def do_reserve_trex (self, user):
     """Reserves the usage of T-Rex to a certain user"""
     try:
         if not user:
             ret = self.trex.reserve_trex()
         else:
             ret = self.trex.reserve_trex(user.split(' ')[0])
         print termstyle.green("*** T-Rex reserved successfully ***")
     except TRexException as inst:
         print termstyle.red(inst)
 def do_cancel_reservation(self, user):
     """Cancels a current reservation of T-Rex to a certain user"""
     try:
         if not user:
             ret = self.trex.cancel_reservation()
         else:
             ret = self.trex.cancel_reservation(user.split(' ')[0])
         print termstyle.green(
             "*** T-Rex reservation canceled successfully ***")
     except TRexException as inst:
         print termstyle.red(inst)
 def do_poll_once (self, line):
     """Performs a single poll of T-Rex current data dump (if T-Rex is running) and prompts and short version of latest result_obj"""
     print termstyle.green("*** Trying T-Rex single poll ***")
     try:
         last_res = dict()
         if self.trex.is_running(dump_out = last_res):
             obj = self.trex.get_result_obj()
             print obj
         else:
             print termstyle.magenta("T-Rex isn't currently running.")
         print termstyle.green("*** End of scenario (T-Rex is posssibly still running!) ***")
     except TRexException as inst:
         print termstyle.red(inst)
 def do_kill_indiscriminately(self, line):
     """Force killing of running T-Rex process (if exists) on the server."""
     print termstyle.green("*** Starting T-Rex termination ***")
     ret = self.trex.force_kill()
     if ret:
         print termstyle.green(
             "*** End of scenario (T-Rex is not running now) ***")
     elif ret is None:
         print termstyle.magenta(
             "*** End of scenario (T-Rex termination aborted) ***")
     else:
         print termstyle.red(
             "*** End of scenario (T-Rex termination failed) ***")
 def do_poll_once(self, line):
     """Performs a single poll of T-Rex current data dump (if T-Rex is running) and prompts and short version of latest result_obj"""
     print termstyle.green("*** Trying T-Rex single poll ***")
     try:
         last_res = dict()
         if self.trex.is_running(dump_out=last_res):
             obj = self.trex.get_result_obj()
             print obj
         else:
             print termstyle.magenta("T-Rex isn't currently running.")
         print termstyle.green(
             "*** End of scenario (T-Rex is posssibly still running!) ***")
     except TRexException as inst:
         print termstyle.red(inst)
    def do_run_until_finish (self, sample_rate):
        """Starts T-Rex and sample server until run is done."""
        print termstyle.green("*** Starting T-Rex run_until_finish scenario ***")

        if not sample_rate: # use default sample rate if not passed
            sample_rate = 5
        try:
            sample_rate = int(sample_rate)
            ret = self.trex.start_trex(**self.run_params)
            self.trex.sample_to_run_finish(sample_rate)
            print termstyle.green("*** End of T-Rex run ***")
        except ValueError as inst:
            print termstyle.magenta("Provided illegal sample rate value. Please try again.\n[", inst,"]")
        except TRexException as inst:
            print termstyle.red(inst)
    def do_run_until_finish(self, sample_rate):
        """Starts T-Rex and sample server until run is done."""
        print termstyle.green(
            "*** Starting T-Rex run_until_finish scenario ***")

        if not sample_rate:  # use default sample rate if not passed
            sample_rate = 5
        try:
            sample_rate = int(sample_rate)
            ret = self.trex.start_trex(**self.run_params)
            self.trex.sample_to_run_finish(sample_rate)
            print termstyle.green("*** End of T-Rex run ***")
        except ValueError as inst:
            print termstyle.magenta(
                "Provided illegal sample rate value. Please try again.\n[",
                inst, "]")
        except TRexException as inst:
            print termstyle.red(inst)
    def do_run_until_condition (self, sample_rate):
        """Starts T-Rex and sample server until condition is satisfied."""
        print termstyle.green("*** Starting T-Rex run until condition is satisfied scenario ***")

        def condition (result_obj):
                return result_obj.get_current_tx_rate()['m_tx_pps'] > 200000

        if not sample_rate: # use default sample rate if not passed
            sample_rate = 5
        try:
            sample_rate = int(sample_rate)
            ret = self.trex.start_trex(**self.run_params)
            ret_val = self.trex.sample_until_condition(condition, sample_rate)
            print ret_val
            print termstyle.green("*** End of T-Rex run ***")
        except ValueError as inst:
            print termstyle.magenta("Provided illegal sample rate value. Please try again.\n[", inst,"]")
        except TRexException as inst:
            print termstyle.red(inst)
 def do_run_and_poll (self, sample_rate):
     """Starts T-Rex and sample server manually until run is done."""
     print termstyle.green("*** Starting T-Rex run and manually poll scenario ***")
     if not sample_rate: # use default sample rate if not passed
         sample_rate = 5
     try:
         sample_rate = int(sample_rate)
         ret = self.trex.start_trex(**self.run_params)
         last_res = dict()
         while self.trex.is_running(dump_out = last_res):
             obj = self.trex.get_result_obj()
             if (self.verbose):
                 print obj
             # do WHATEVER here
             time.sleep(sample_rate)
             
         print termstyle.green("*** End of T-Rex run ***")
     except ValueError as inst:
         print termstyle.magenta("Provided illegal sample rate value. Please try again.\n[", inst,"]")
     except TRexException as inst:
         print termstyle.red(inst)
    def do_run_until_condition(self, sample_rate):
        """Starts T-Rex and sample server until condition is satisfied."""
        print termstyle.green(
            "*** Starting T-Rex run until condition is satisfied scenario ***")

        def condition(result_obj):
            return result_obj.get_current_tx_rate()['m_tx_pps'] > 200000

        if not sample_rate:  # use default sample rate if not passed
            sample_rate = 5
        try:
            sample_rate = int(sample_rate)
            ret = self.trex.start_trex(**self.run_params)
            ret_val = self.trex.sample_until_condition(condition, sample_rate)
            print ret_val
            print termstyle.green("*** End of T-Rex run ***")
        except ValueError as inst:
            print termstyle.magenta(
                "Provided illegal sample rate value. Please try again.\n[",
                inst, "]")
        except TRexException as inst:
            print termstyle.red(inst)
    def do_run_and_poll(self, sample_rate):
        """Starts T-Rex and sample server manually until run is done."""
        print termstyle.green(
            "*** Starting T-Rex run and manually poll scenario ***")
        if not sample_rate:  # use default sample rate if not passed
            sample_rate = 5
        try:
            sample_rate = int(sample_rate)
            ret = self.trex.start_trex(**self.run_params)
            last_res = dict()
            while self.trex.is_running(dump_out=last_res):
                obj = self.trex.get_result_obj()
                if (self.verbose):
                    print obj
                # do WHATEVER here
                time.sleep(sample_rate)

            print termstyle.green("*** End of T-Rex run ***")
        except ValueError as inst:
            print termstyle.magenta(
                "Provided illegal sample rate value. Please try again.\n[",
                inst, "]")
        except TRexException as inst:
            print termstyle.red(inst)
 def _show(self):
     if self.pidfile.is_locked():
         print termstyle.red("T-Rex server daemon is running")
     else:
         print termstyle.red("T-Rex server daemon is NOT running")
 def _show (self):
     if self.pidfile.is_locked():
         print termstyle.red("T-Rex server daemon is running")
     else:
         print termstyle.red("T-Rex server daemon is NOT running")