def _colorize(color, message): """Add a color to the message.""" try: return getattr(colorama.Fore, color) + message + colorama.Fore.RESET except AttributeError: return message