Ejemplo n.º 1
0
 def blue(self, text):
     self._restoreColor()
     if platform.system() == 'Windows':  # pragma: no cover
         # Default blue in windows is unreadable (such awful defaults...)
         return termstyle.cyan(text)
     else:
         return termstyle.blue(text)
Ejemplo n.º 2
0
    def __init__(self,
                 cfg_yaml_path=None,
                 silent_mode=False,
                 virtual_mode=False):
        #       super(InteractivePlatform, self).__init__()
        cmd.Cmd.__init__(self)
        self.virtual_mode = virtual_mode
        self.platform = CPlatform(silent_mode)
        if cfg_yaml_path is None:
            try:
                cfg_yaml_path = raw_input(
                    termstyle.cyan(
                        "Please enter a readable .yaml configuration file path: "
                    ))
                cfg_yaml_path = os.path.abspath(cfg_yaml_path)
            except KeyboardInterrupt:
                exit(-1)
        try:
            self.device_cfg = CDeviceCfg(cfg_yaml_path)
            self.platform.load_platform_data_from_file(self.device_cfg)
            if not virtual_mode:
                # if not virtual mode, try to establish a phyisical connection to platform
                self.platform.launch_connection(self.device_cfg)

        except Exception as inst:
            print(termstyle.magenta(inst))
            exit(-1)
Ejemplo n.º 3
0
 def blue(self, text):
     self._restoreColor()
     if platform.system() == 'Windows': # pragma: no cover
         # Default blue in windows is unreadable (such awful defaults...)
         return termstyle.cyan(text)
     else:
         return termstyle.blue(text)
Ejemplo n.º 4
0
	def _fmt_message_sqlalchemy_line(self, line, match, color):
		logger = match.group(1)
		level = match.group(2)
		verb = match.group(3)
		rest = match.group(4) if len(match.groups()) > 3 else ''
		if verb in ['SELECT']:
			head = color("%s: %s:" % (logger, level))
			self._in_multiline_sql = verb
			self._in_multiline_sql_color = termstyle.cyan
			return "%s %s %s" % (head, termstyle.bold(termstyle.cyan(verb)), termstyle.cyan(rest))
		elif verb in ['BEGIN', 'SAVEPOINT', 'RELEASE', 'ROLLBACK', 'COMMIT']:
			head = color("%s: %s:" % (logger, level))
			self._in_multiline_sql = verb
			self._in_multiline_sql_color = termstyle.green
			return "%s %s %s" % (head, termstyle.bold(termstyle.green(verb)), termstyle.green(rest))
		elif verb in ['INSERT', 'UPDATE', 'DELETE', 'TRUNCATE']:
			head = color("%s: %s:" % (logger, level))
			self._in_multiline_sql = verb
			self._in_multiline_sql_color = termstyle.red
			return "%s %s %s" % (head, termstyle.bold(termstyle.red(verb)), termstyle.red(rest))
		return color(line)
Ejemplo n.º 5
0
    def _format_traceback_line(self, tb):
        """
        Formats the file / lineno / function line of a traceback element.

        Returns None is the line is not relevent to the user i.e. inside the test runner.
        """
        if self._is_relevant_tb_level(tb):
            return None

        f = tb.tb_frame
        filename = f.f_code.co_filename
        lineno = tb.tb_lineno
        linecache.checkcache(filename)
        function_name = f.f_code.co_name

        line_contents = linecache.getline(filename, lineno, f.f_globals).strip()

        return "    %s line %s in %s\n      %s" % (
            termstyle.blue(self._relative_path(filename) if self.use_relative_path else filename),
            termstyle.bold(termstyle.cyan(lineno)),
            termstyle.cyan(function_name),
            line_contents
        )
Ejemplo n.º 6
0
	def _file_line(self, tb):
		"""formats the file / lineno / function line of a traceback element"""
		prefix = "file://"
		prefix = ""

		f = tb.tb_frame
		if '__unittest' in f.f_globals:
			# this is the magical flag that prevents unittest internal
			# code from junking up the stacktrace
			return None

		filename = f.f_code.co_filename
		lineno = tb.tb_lineno
		linecache.checkcache(filename)
		function_name = f.f_code.co_name

		line_contents = linecache.getline(filename, lineno, f.f_globals).strip()

		return "    %s line %s in %s\n      %s" % (
			termstyle.blue(prefix, self._relative_path(filename)),
			termstyle.bold(termstyle.cyan(lineno)),
			termstyle.cyan(function_name),
			line_contents)
Ejemplo n.º 7
0
    def _file_line(self, tb):
        """formats the file / lineno / function line of a traceback element"""
        prefix = "file://"
        prefix = ""

        f = tb.tb_frame
        if '__unittest' in f.f_globals:
            # this is the magical flag that prevents unittest internal
            # code from junking up the stacktrace
            return None

        filename = f.f_code.co_filename
        lineno = tb.tb_lineno
        linecache.checkcache(filename)
        function_name = f.f_code.co_name

        line_contents = linecache.getline(filename, lineno,
                                          f.f_globals).strip()

        return "    %s line %s in %s\n      %s" % (
            termstyle.blue(prefix, self._relative_path(filename)),
            termstyle.bold(termstyle.cyan(lineno)),
            termstyle.cyan(function_name), line_contents)
Ejemplo n.º 8
0
    def __init__(self, cfg_yaml_path = None, silent_mode = False, virtual_mode = False ):
#       super(InteractivePlatform, self).__init__()
        cmd.Cmd.__init__(self)
        self.virtual_mode   = virtual_mode
        self.platform       = CPlatform(silent_mode)
        if cfg_yaml_path is None:
            try:
                cfg_yaml_path = raw_input(termstyle.cyan("Please enter a readable .yaml configuration file path: "))
                cfg_yaml_path = os.path.abspath(cfg_yaml_path)
            except KeyboardInterrupt:
                exit(-1)
        try:
            self.device_cfg = CDeviceCfg(cfg_yaml_path)
            self.platform.load_platform_data_from_file(self.device_cfg)
            if not virtual_mode:
                # if not virtual mode, try to establish a phyisical connection to platform
                self.platform.launch_connection(self.device_cfg)

        except Exception as inst:
            print(termstyle.magenta(inst))
            exit(-1)
Ejemplo n.º 9
0
 def do_exit(self, arg):
     """Quits the application"""
     print termstyle.cyan('Bye Bye!')
     return True
Ejemplo n.º 10
0
        ret = self.trex.force_kill()
        if ret:
            print termstyle.green("*** End of scenario (TRex is not running now) ***")
        elif ret is None:
            print termstyle.magenta("*** End of scenario (TRex termination aborted) ***")
        else:
            print termstyle.red("*** End of scenario (TRex termination failed) ***")

    def do_exit(self, arg):
        """Quits the application"""
        print termstyle.cyan('Bye Bye!')
        return True


if __name__ == "__main__":
    parser = ArgumentParser(description = termstyle.cyan('Run TRex client API demos and scenarios.'),
        usage = """client_interactive_example [options]""" )

    parser.add_argument('-v', '--version', action='version', version='%(prog)s 1.0 \t (C) Cisco Systems Inc.\n')

    parser.add_argument("-t", "--trex-host", required = True, dest="trex_host",
        action="store", help="Specify the hostname or ip to connect with TRex server.",
        metavar="HOST" )
    parser.add_argument("-p", "--trex-port", type=int, default = 8090, metavar="PORT", dest="trex_port", 
        help="Select port on which the TRex server listens. Default port is 8090.", action="store")
    parser.add_argument("-m", "--maxhist", type=int, default = 100, metavar="SIZE", dest="hist_size", 
        help="Specify maximum history size saved at client side. Default size is 100.", action="store")
    parser.add_argument("--verbose", dest="verbose",
        action="store_true", help="Switch ON verbose option at TRex client. Default is: OFF.",
        default = False )
    args = parser.parse_args()
Ejemplo n.º 11
0
 def __cbterminateIDLE(self, *args, **kwargs):
     print termstyle.cyan('__cbterminateIDLE')
     print termstyle.cyan(args)
     print termstyle.cyan(kwargs)
     return self.logout()
Ejemplo n.º 12
0
def color_parts(parts):
    """Adds colors to each part of the citation"""
    return parts._replace(title=green(parts.title), doi=cyan(parts.doi))