Coverage for lib/ansible/utils/color.py : 64%

# (c) 2012-2014, Michael DeHaan <michael.dehaan@gmail.com> 

# 

# This file is part of Ansible 

# 

# Ansible is free software: you can redistribute it and/or modify 

# it under the terms of the GNU General Public License as published by 

# the Free Software Foundation, either version 3 of the License, or 

# (at your option) any later version. 

# 

# Ansible is distributed in the hope that it will be useful, 

# but WITHOUT ANY WARRANTY; without even the implied warranty of 

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

# GNU General Public License for more details. 

# 

# You should have received a copy of the GNU General Public License 

# along with Ansible. If not, see <http://www.gnu.org/licenses/>. 

from __future__ import (absolute_import, division, print_function) 

__metaclass__ = type 

import re 

import sys 

from ansible import constants as C 

ANSIBLE_COLOR = True 

26 ↛ 27line 26 didn't jump to line 27, because the condition on line 26 was never trueif C.ANSIBLE_NOCOLOR: 

ANSIBLE_COLOR = False 

elif not hasattr(sys.stdout, 'isatty') or not sys.stdout.isatty(): 

ANSIBLE_COLOR = False 

else: 

try: 

import curses 

curses.setupterm() 

if curses.tigetnum('colors') < 0: 

ANSIBLE_COLOR = False 

36 ↛ 38line 36 didn't jump to line 38, because the exception caught by line 36 didn't happen except ImportError: 

# curses library was not found 

pass 

except curses.error: 

# curses returns an error (e.g. could not find terminal) 

ANSIBLE_COLOR = False 

43 ↛ 57line 43 didn't jump to line 57, because the condition on line 43 was never falseif C.ANSIBLE_FORCE_COLOR: 

ANSIBLE_COLOR = True 

# --- begin "pretty" 

# 

# pretty - A miniature library that provides a Python print and stdout 

# wrapper that makes colored terminal text easier to use (e.g. without 

# having to mess around with ANSI escape sequences). This code is public 

# domain - there is no license except that you must leave this header. 

# 

# Copyright (C) 2008 Brian Nez <thedude at bri1 dot com> 

# 

# http://nezzen.net/2008/06/23/colored-text-in-python-using-ansi-escape-sequences/ 

codeCodes = { 

'black': u'0;30', 'bright gray': u'0;37', 

'blue': u'0;34', 'white': u'1;37', 

'green': u'0;32', 'bright blue': u'1;34', 

'cyan': u'0;36', 'bright green': u'1;32', 

'red': u'0;31', 'bright cyan': u'1;36', 

'purple': u'0;35', 'bright red': u'1;31', 

'yellow': u'0;33', 'bright purple': u'1;35', 

'dark gray': u'1;30', 'bright yellow': u'1;33', 

'magenta': u'0;35', 'bright magenta': u'1;35', 

'normal': u'0', 

} 

def parsecolor(color): 

"""SGR parameter string for the specified color name.""" 

matches = re.match(r"color(?P<color>[0-9]+)" 

r"|(?P<rgb>rgb(?P<red>[0-5])(?P<green>[0-5])(?P<blue>[0-5]))" 

r"|gray(?P<gray>[0-9]+)", color) 

76 ↛ 78line 76 didn't jump to line 78, because the condition on line 76 was never false if not matches: 

return codeCodes[color] 

if matches.group('color'): 

return u'38;5;%d' % int(matches.group('color')) 

if matches.group('rgb'): 

return u'38;5;%d' % (16 + 36 * int(matches.group('red')) + 

6 * int(matches.group('green')) + 

int(matches.group('blue'))) 

if matches.group('gray'): 

return u'38;5;%d' % (232 + int(matches.group('gray'))) 

def stringc(text, color): 

"""String in color.""" 

91 ↛ 95line 91 didn't jump to line 95, because the condition on line 91 was never false if ANSIBLE_COLOR: 

color_code = parsecolor(color) 

return u"\n".join([u"\033[%sm%s\033[0m" % (color_code, t) for t in text.split(u'\n')]) 

else: 

return text 

def colorize(lead, num, color): 

""" Print 'lead' = 'num' in 'color' """ 

s = u"%s=%-4s" % (lead, str(num)) 

if num != 0 and ANSIBLE_COLOR and color is not None: 

s = stringc(s, color) 

return s 

def hostcolor(host, stats, color=True): 

if ANSIBLE_COLOR and color: 

108 ↛ 109line 108 didn't jump to line 109, because the condition on line 108 was never true if stats['failures'] != 0 or stats['unreachable'] != 0: 

return u"%-37s" % stringc(host, C.COLOR_ERROR) 

110 ↛ 113line 110 didn't jump to line 113, because the condition on line 110 was never false elif stats['changed'] != 0: 

return u"%-37s" % stringc(host, C.COLOR_CHANGED) 

else: 

return u"%-37s" % stringc(host, C.COLOR_OK) 

return u"%-26s" % host