Daiki Kato / mbed-os-lychee

mbed-os for GR-LYCHEE

Dependents:   mbed-os-example-blinky-gr-lychee GR-Boads_Camera_sample GR-Boards_Audio_Recoder GR-Boads_Camera_DisplayApp ... more

tools/colorize.py@0:f782d9c66c49, 2018-02-02 (annotated)

Committer:
dkato
Date:
Fri Feb 02 05:42:23 2018 +0000
Revision:
0:f782d9c66c49
mbed-os for GR-LYCHEE

Who changed what in which revision?

UserRevisionLine numberNew contents of line
dkato 0:f782d9c66c49 1 # mbed SDK
dkato 0:f782d9c66c49 2 # Copyright (c) 2016 ARM Limited
dkato 0:f782d9c66c49 3 #
dkato 0:f782d9c66c49 4 # Licensed under the Apache License, Version 2.0 (the "License");
dkato 0:f782d9c66c49 5 # you may not use this file except in compliance with the License.
dkato 0:f782d9c66c49 6 # You may obtain a copy of the License at
dkato 0:f782d9c66c49 7 #
dkato 0:f782d9c66c49 8 # http://www.apache.org/licenses/LICENSE-2.0
dkato 0:f782d9c66c49 9 #
dkato 0:f782d9c66c49 10 # Unless required by applicable law or agreed to in writing, software
dkato 0:f782d9c66c49 11 # distributed under the License is distributed on an "AS IS" BASIS,
dkato 0:f782d9c66c49 12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
dkato 0:f782d9c66c49 13 # See the License for the specific language governing permissions and
dkato 0:f782d9c66c49 14 # limitations under the License.
dkato 0:f782d9c66c49 15
dkato 0:f782d9c66c49 16 """ This python file is responsible for generating colorized notifiers.
dkato 0:f782d9c66c49 17 """
dkato 0:f782d9c66c49 18
dkato 0:f782d9c66c49 19 import sys
dkato 0:f782d9c66c49 20 import re
dkato 0:f782d9c66c49 21 from colorama import init, Fore, Back, Style
dkato 0:f782d9c66c49 22 init()
dkato 0:f782d9c66c49 23
dkato 0:f782d9c66c49 24 COLORS = {
dkato 0:f782d9c66c49 25 'none' : "",
dkato 0:f782d9c66c49 26 'default' : Style.RESET_ALL,
dkato 0:f782d9c66c49 27
dkato 0:f782d9c66c49 28 'black' : Fore.BLACK,
dkato 0:f782d9c66c49 29 'red' : Fore.RED,
dkato 0:f782d9c66c49 30 'green' : Fore.GREEN,
dkato 0:f782d9c66c49 31 'yellow' : Fore.YELLOW,
dkato 0:f782d9c66c49 32 'blue' : Fore.BLUE,
dkato 0:f782d9c66c49 33 'magenta' : Fore.MAGENTA,
dkato 0:f782d9c66c49 34 'cyan' : Fore.CYAN,
dkato 0:f782d9c66c49 35 'white' : Fore.WHITE,
dkato 0:f782d9c66c49 36
dkato 0:f782d9c66c49 37 'on_black' : Back.BLACK,
dkato 0:f782d9c66c49 38 'on_red' : Back.RED,
dkato 0:f782d9c66c49 39 'on_green' : Back.GREEN,
dkato 0:f782d9c66c49 40 'on_yellow' : Back.YELLOW,
dkato 0:f782d9c66c49 41 'on_blue' : Back.BLUE,
dkato 0:f782d9c66c49 42 'on_magenta' : Back.MAGENTA,
dkato 0:f782d9c66c49 43 'on_cyan' : Back.CYAN,
dkato 0:f782d9c66c49 44 'on_white' : Back.WHITE,
dkato 0:f782d9c66c49 45 }
dkato 0:f782d9c66c49 46
dkato 0:f782d9c66c49 47 COLOR_MATCHER = re.compile(r"(\w+)(\W+on\W+\w+)?")
dkato 0:f782d9c66c49 48 def colorstring_to_escapecode(color_string):
dkato 0:f782d9c66c49 49 """ Convert a color string from a string into an ascii escape code that
dkato 0:f782d9c66c49 50 will print that color on the terminal.
dkato 0:f782d9c66c49 51
dkato 0:f782d9c66c49 52 Positional arguments:
dkato 0:f782d9c66c49 53 color_string - the string to parse
dkato 0:f782d9c66c49 54 """
dkato 0:f782d9c66c49 55 match = re.match(COLOR_MATCHER, color_string)
dkato 0:f782d9c66c49 56 if match:
dkato 0:f782d9c66c49 57 return COLORS[match.group(1)] + \
dkato 0:f782d9c66c49 58 (COLORS[match.group(2).strip().replace(" ", "_")]
dkato 0:f782d9c66c49 59 if match.group(2) else "")
dkato 0:f782d9c66c49 60 else:
dkato 0:f782d9c66c49 61 return COLORS['default']
dkato 0:f782d9c66c49 62
dkato 0:f782d9c66c49 63
dkato 0:f782d9c66c49 64 def print_in_color_notifier(color_map, print_fn):
dkato 0:f782d9c66c49 65 """ Wrap a toolchain notifier in a colorizer. This colorizer will wrap
dkato 0:f782d9c66c49 66 notifications in a color if the severity matches a color in the *color_map*.
dkato 0:f782d9c66c49 67 """
dkato 0:f782d9c66c49 68 def wrap(event, silent=False):
dkato 0:f782d9c66c49 69 """The notification function inself"""
dkato 0:f782d9c66c49 70 file_desc = sys.stdout
dkato 0:f782d9c66c49 71 self = event['toolchain']
dkato 0:f782d9c66c49 72 if file_desc.isatty() and 'severity' in event and \
dkato 0:f782d9c66c49 73 event['severity'] in color_map:
dkato 0:f782d9c66c49 74 file_desc.write(colorstring_to_escapecode(
dkato 0:f782d9c66c49 75 color_map[event['severity']]))
dkato 0:f782d9c66c49 76 print_fn(self, event, silent)
dkato 0:f782d9c66c49 77 file_desc.write(colorstring_to_escapecode('default'))
dkato 0:f782d9c66c49 78 else:
dkato 0:f782d9c66c49 79 print_fn(self, event, silent)
dkato 0:f782d9c66c49 80 return wrap