diff --git a/pidcat.py b/pidcat.py index c95c25d..61f5088 100755 --- a/pidcat.py +++ b/pidcat.py @@ -46,6 +46,7 @@ parser.add_argument('-i', '--ignore-tag', dest='ignored_tag', action='append', help='Filter output by ignoring specified tag(s)') parser.add_argument('-v', '--version', action='version', version='%(prog)s ' + __version__, help='Print the version number and exit') parser.add_argument('-a', '--all', dest='all', action='store_true', default=False, help='Print all log messages') +parser.add_argument('--colorized', '--colorized', dest='colorized', action='store_true', default=False, help='Colorized log messages') args = parser.parse_args() min_level = LOG_LEVELS_MAP[args.min_level.upper()] @@ -90,6 +91,11 @@ RESET = '\033[0m' +colorized = False +if len(sys.argv) > 1: + if sys.argv[1] == "--colorized": + colorized = True + def termcolor(fg=None, bg=None): codes = [] if fg is not None: codes.append('3%d' % fg) @@ -355,6 +361,6 @@ def tag_in_tags_regex(tag, tags): for matcher in RULES: replace = RULES[matcher] message = matcher.sub(replace, message) - - linebuf += indent_wrap(message) + lineFg = color if colorized else WHITE + linebuf += indent_wrap(colorize(message, fg=lineFg)) print(linebuf.encode('utf-8'))