def diff_color(input_file, cfg):
"""
colorizes a diff file
"""
cov = cfg.data
with paged_echo() as pager:
term_width = click.get_terminal_size()[0]
modified = []
measured = cov.get_data().measured_files()
diff = PatchSet(input_file)
for thing in diff:
if thing.is_modified_file or thing.is_added_file:
target = thing.target_file
if target.startswith('b/') or target.startswith('a/'):
target = target[2:]
if abspath(target) in measured:
covdata = cov._analyze(abspath(target))
modified.append((thing, covdata))
# pager.echo(abspath(target))
else:
msg = "skip: {}".format(target)
msg = msg + (' ' * (term_width - len(msg)))
pager.echo(colors.color(msg, bg='yellow', fg='black'))
total_added_lines = 0
total_covered_lines = 0
for (patch, covdata) in modified:
fname = str(patch.target_file) + (' ' * (term_width - len(patch.target_file)))
pager.echo(colors.color(fname, bg='cyan', fg='black'))
for hunk in patch:
for line in hunk:
kw = dict()
if line.is_added:
total_added_lines += 1
if line.target_line_no in covdata.missing:
pager.echo(colors.color(u'\u258c', fg='red', bg=52), nl=False, color=True)
kw['bg'] = 52
else:
total_covered_lines += 1
pager.echo(colors.color(u'\u258f', fg='green'), nl=False, color=True)
else:
pager.echo(' ', nl=False)
out = u"{}".format(line).strip()
if line.is_added:
kw['fg'] = 'green'
elif line.is_removed:
kw['fg'] = 'red'
pager.echo(colors.color(out, **kw))
if total_added_lines == 0:
raise click.ClickException("No covered lines at all")
percent_covered = (total_covered_lines / float(total_added_lines))
msg = u"{} covered of {} added lines".format(total_covered_lines, total_added_lines)
print_banner(msg, percent_covered, pager=pager)
return
target_fname = abspath(target_fname)
for fname in cov.get_data().measured_files():
if target_fname == abspath(fname):
match.append(fname)
if len(match) != 1:
if len(match) == 0:
# this file wasn't in the coverage data, so we just dump
# it to stdout as-is. (FIXME: ideally, also
# syntax-highlighted anyway)
with open(target_fname, 'r') as f:
for line in f.readlines():
sys.stdout.write(line)
return
else:
raise RuntimeError("Multiple matches: %s" % ', '.join(match))
fname = match[0]
covdata = cov._analyze(fname)
percent = 1.0 # if no statements, it's all covered, right?
if covdata.numbers.n_statements:
percent = float(covdata.numbers.n_statements - covdata.numbers.n_missing) / covdata.numbers.n_statements
total_statements = covdata.numbers.n_statements
total_missing = covdata.numbers.n_missing
fill = min(click.get_terminal_size()[0], 80)
print_banner(fname, percent, fill)
# it was tempting to write/override/wrap this Formatter and mess
# with the background color based on our coverage data -- and
# that's not a terrible idea, but the way TerminalFormatter is
# written, it's not very nice. Basically, we'd have to wrap the
# output stream, look for ANSI reset codes, and re-do the
# background color after each reset (for "uncovered" lines)... so
# I didn't do that. Instead we just hack it by splitting on the
# reset codes (see below)
formatter = TerminalFormatter(style=style)
lines = highlight(
open(fname).read(), get_lexer_by_name('python'),
formatter=formatter,
)
lines = lines.split(u'\n')
for (i, line) in enumerate(lines):
# assert type(line) is unicode
spaces = fill - len(colors.strip_color(line))
spaces = u' ' * spaces
if (i + 1) not in covdata.missing:
if (i + 1) in covdata.excluded:
line = colors.strip_color(line)
pager.echo(colors.color(u'\u258f', fg=46, bg=236) + colors.color(line + spaces, bg=236, fg=242), color=True)
elif cfg.branch and (i + 1) in covdata.branch_lines():
line = colors.strip_color(line)
pager.echo(colors.color(u'\u258f', bg=52, fg=160) + colors.color(line + spaces, bg=52), color=True)
else:
pager.echo(u'{}{}{}'.format(colors.color(u'\u258f', fg=46), line, spaces), color=True)
else:
# HACK-O-MATIC, uhm. Yeah, so what we're doing here is
# splitting the output from the formatter on the ANSI
# "reset" code, and the re-assembling it with the "52"
# (dark red) background color. I appoligize in advance;
# PRs with improvements encouraged!
reset_code = u"\x1b[39;49;00m"
segments = (line + spaces).split(reset_code)
reset_plus_bg = u"\x1b[39;49;00m\x1b[39;49;48;5;52m"
out = u"\x1b[39;49;48;5;52m" + reset_plus_bg.join(segments)
pager.echo(colors.color(u'\u258f', bg=52, fg=160) + out, color=True)
def term_color(target_fname, cfg, style='monokai'):
cov = cfg.data
target_fname = realpath(target_fname)
match = filter( lambda f: target_fname == realpath(f), cov.data.measured_files() )
if len(match) > 1:
raise RuntimeError("Multiple matches: %s" % ', '.join(match))
if len(match) == 0:
# this file wasn't in the coverage data, so we just dump
# it to stdout as-is. (FIXME: ideally, also
# syntax-highlighted anyway)
with open(target_fname, 'r') as f:
sys.stdout.write( f.read() )
return
fname = match[0]
covdata = cov._analyze(fname)
percent = 1.0 # if no statements, it's all covered, right?
if covdata.numbers.n_statements:
percent = float(covdata.numbers.n_statements - covdata.numbers.n_missing) / covdata.numbers.n_statements
total_statements = covdata.numbers.n_statements
total_missing = covdata.numbers.n_missing
fill = min(click.get_terminal_size()[0], 80)
print_banner(fname, percent, fill)
# it was tempting to write/override/wrap this Formatter and mess
# with the background color based on our coverage data -- and
# that's not a terrible idea, but the way TerminalFormatter is
# written, it's not very nice. Basically, we'd have to wrap the
# output stream, look for ANSI reset codes, and re-do the
# background color after each reset (for "uncovered" lines)...
# so I didn't do that. Instead we just hack it by clearing *all*
# formatting from the "uncovered" lines and make them all "grey on
# red"
formatter = TerminalFormatter(style=style)
lines = highlight(
open(fname).read(), get_lexer_by_name('python'),
formatter=formatter,
)
lines = lines.split(u'\n')
for (i, line) in enumerate(lines):
# assert type(line) is unicode
spaces = fill - len(colors.strip_color(line))
spaces = u' ' * spaces
if (i + 1) not in covdata.missing:
if (i + 1) in covdata.excluded:
line = colors.strip_color(line)
click.echo(colors.color(u'\u258f', fg=46, bg=236) + colors.color(line + spaces, bg=236, fg=242), color=True)
elif cfg.branch and (i + 1) in covdata.branch_lines():
line = colors.strip_color(line)
click.echo(colors.color(u'\u258f', bg=52, fg=160) + colors.color(line + spaces, bg=52), color=True)
else:
click.echo(u'{}{}{}'.format(colors.color(u'\u258f', fg=46), line, spaces), color=True)
else:
# HACK-O-MATIC, uhm. Yeah, so what we're doing here is
# splitting the output from the formatter on the ANSI
# "reset" code, and the re-assembling it with the "52"
# (dark red) background color. I appoligize in advance;
# PRs with improvements encouraged!
reset_code = u"\x1b[39;49;00m"
segments = (line + spaces).split(reset_code)
reset_plus_bg = u"\x1b[39;49;00m\x1b[39;49;48;5;52m"
out = u"\x1b[39;49;48;5;52m" + reset_plus_bg.join(segments)
click.echo(colors.color(u'\u258c', bg=52, fg=160) + out, color=True)
def diff_color(input_file, cfg):
"""
colorizes a diff file
"""
cov = cfg.data
term_width = click.get_terminal_size()[0]
modified = []
measured = cov.data.measured_files()
diff = PatchSet(input_file)
for thing in diff:
if thing.is_modified_file or thing.is_added_file:
target = thing.target_file
if target.startswith('b/') or target.startswith('a/'):
target = target[2:]
if abspath(target) in measured:
covdata = cov._analyze(abspath(target))
modified.append((thing, covdata))
print(abspath(target))
else:
msg = "skip: {}".format(thing.target_file)
msg = msg + (' ' * (term_width - len(msg)))
print(colors.color(msg, bg='yellow', fg='black'))
# else:
# print("THING", dir(thing))
for (patch, covdata) in modified:
fname = str(patch.target_file) + (' ' * (term_width - len(patch.target_file)))
print(colors.color(fname, bg='cyan', fg='black'))
for hunk in patch:
for line in hunk:
kw = dict()
if line.is_added:
if line.target_line_no in covdata.missing:
click.echo(colors.color(u'\u258f', fg='red', bg=52), nl=False, color=True)
kw['bg'] = 52
else:
click.echo(colors.color(u'\u258f', fg='green'), nl=False, color=True)
else:
print(' ', end='')
out = str(line)
if line.is_added:
kw['fg'] = 'green'
elif line.is_removed:
kw['fg'] = 'red'
print(colors.color(out, **kw))
return
target_fname = abspath(target_fname)
for fname in cov.data.measured_files():
if target_fname == abspath(fname):
match.append(fname)
if len(match) != 1:
if len(match) == 0:
# this file wasn't in the coverage data, so we just dump
# it to stdout as-is. (FIXME: ideally, also
# syntax-highlighted anyway)
with open(target_fname, 'r') as f:
for line in f.readlines():
sys.stdout.write(line)
return
else:
raise RuntimeError("Multiple matches: %s" % ', '.join(match))
fname = match[0]
covdata = cov._analyze(fname)
percent = 1.0 # if no statements, it's all covered, right?
if covdata.numbers.n_statements:
percent = float(covdata.numbers.n_statements - covdata.numbers.n_missing) / covdata.numbers.n_statements
total_statements = covdata.numbers.n_statements
total_missing = covdata.numbers.n_missing
fill = min(click.get_terminal_size()[0], 80)
print_banner(fname, percent, fill)
# it was tempting to write/override/wrap this Formatter and mess
# with the background color based on our coverage data -- and
# that's not a terrible idea, but the way TerminalFormatter is
# written, it's not very nice. Basically, we'd have to wrap the
# output stream, look for ANSI reset codes, and re-do the
# background color after each reset (for "uncovered" lines)...
# so I didn't do that. Instead we just hack it by clearing *all*
# formatting from the "uncovered" lines and make them all "grey on
# red"
formatter = TerminalFormatter(style=style)
lines = highlight(
open(fname).read(), get_lexer_by_name('python'),
formatter=formatter,
)
lines = lines.split(u'\n')
for (i, line) in enumerate(lines):
# assert type(line) is unicode
spaces = fill - len(colors.strip_color(line))
spaces = u' ' * spaces
if (i + 1) not in covdata.missing:
if (i + 1) in covdata.excluded:
line = colors.strip_color(line)
click.echo(colors.color(u'\u258f', fg=46, bg=236) + colors.color(line + spaces, bg=236, fg=242), color=True)
elif cfg.branch and (i + 1) in covdata.branch_lines():
line = colors.strip_color(line)
click.echo(colors.color(u'\u258f', bg=52, fg=160) + colors.color(line + spaces, bg=52), color=True)
else:
click.echo(u'{}{}{}'.format(colors.color(u'\u258f', fg=46), line, spaces), color=True)
else:
# HACK-O-MATIC, uhm. Yeah, so what we're doing here is
# splitting the output from the formatter on the ANSI
# "reset" code, and the re-assembling it with the "52"
# (dark red) background color. I appoligize in advance;
# PRs with improvements encouraged!
reset_code = u"\x1b[39;49;00m"
segments = (line + spaces).split(reset_code)
reset_plus_bg = u"\x1b[39;49;00m\x1b[39;49;48;5;52m"
out = u"\x1b[39;49;48;5;52m" + reset_plus_bg.join(segments)
click.echo(colors.color(u'\u258f', bg=52, fg=160) + out, color=True)
def term_color(target_fname, cfg, style='monokai'):
cov = cfg.data
target_fname = realpath(target_fname)
match = [f for f in cov.get_data().measured_files() if target_fname == realpath(f)]
if len(match) > 1:
raise RuntimeError("Multiple matches: %s" % ', '.join(match))
if len(match) == 0:
print("not in coverage")
# this file wasn't in the coverage data, so we just dump
# it to stdout as-is. (FIXME: ideally, also
# syntax-highlighted anyway)
with open(target_fname, 'r') as f:
sys.stdout.write( f.read() )
return
fname = match[0]
covdata = cov._analyze(fname)
percent = 1.0 # if no statements, it's all covered, right?
if covdata.numbers.n_statements:
percent = float(covdata.numbers.n_statements - covdata.numbers.n_missing) / covdata.numbers.n_statements
total_statements = covdata.numbers.n_statements
total_missing = covdata.numbers.n_missing
fill = min(click.get_terminal_size()[0], 80)
print_banner(fname, percent, fill)
# it was tempting to write/override/wrap this Formatter and mess
# with the background color based on our coverage data -- and
# that's not a terrible idea, but the way TerminalFormatter is
# written, it's not very nice. Basically, we'd have to wrap the
# output stream, look for ANSI reset codes, and re-do the
# background color after each reset (for "uncovered" lines)...
# so I didn't do that. Instead we just hack it by clearing *all*
# formatting from the "uncovered" lines and make them all "grey on
# red"
formatter = TerminalFormatter(style=style)
raw_data = open(fname).read()
lexer = get_lexer_by_name('python')
lexer.stripall = False
lexer.stripnl = False
lines = highlight(raw_data, lexer, formatter=formatter)
lines = lines.split(u'\n')
# coverage thinks the first line is called "1", not "0"
offset = 1
for (i, line) in enumerate(lines):
# assert type(line) is unicode
spaces = fill - len(colors.strip_color(line))
spaces = u' ' * spaces
if (i + offset) not in covdata.missing:
if (i + offset) in covdata.excluded:
line = colors.strip_color(line)
click.echo(colors.color(u'\u258f', fg=46, bg=236) + colors.color(line + spaces, bg=236, fg=242), color=True)
elif cfg.branch and (i + offset) in covdata.branch_lines():
line = colors.strip_color(line)
click.echo(colors.color(u'\u258f', bg=52, fg=160) + colors.color(line + spaces, bg=52), color=True)
else:
click.echo(u'{}{}{}'.format(colors.color(u'\u258f', fg=46), line, spaces), color=True)
else:
# HACK-O-MATIC, uhm. Yeah, so what we're doing here is
# splitting the output from the formatter on the ANSI
# "reset" code, and the re-assembling it with the "52"
# (dark red) background color. I appoligize in advance;
# PRs with improvements encouraged!
reset_code = u"\x1b[39;49;00m"
segments = (line + spaces).split(reset_code)
reset_plus_bg = u"\x1b[39;49;00m\x1b[39;49;48;5;52m"
out = u"\x1b[39;49;48;5;52m" + reset_plus_bg.join(segments)
click.echo(colors.color(u'\u258c', bg=52, fg=160) + out, color=True)
