From 678042ae6c558821f550f018896318348128ac25 Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Fri, 13 Mar 2009 07:26:17 -0400 Subject: Start moving common reporter functionality into Reporter. --- coverage/annotate.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) (limited to 'coverage/annotate.py') diff --git a/coverage/annotate.py b/coverage/annotate.py index 2d7d04a7..c4fe810f 100644 --- a/coverage/annotate.py +++ b/coverage/annotate.py @@ -2,12 +2,11 @@ import os, re -from coverage.codeunit import code_unit_factory +from coverage.report import Reporter -class AnnotateReporter: +class AnnotateReporter(Reporter): def __init__(self, coverage, ignore_errors=False): - self.coverage = coverage - self.ignore_errors = ignore_errors + super(AnnotateReporter, self).__init__(coverage, ignore_errors) self.directory = None @@ -15,10 +14,10 @@ class AnnotateReporter: else_re = re.compile(r"\s*else\s*:\s*(#|$)") def report(self, morfs, directory=None, omit_prefixes=None): - morfs = morfs or self.coverage.data.executed_files() - code_units = code_unit_factory(morfs, self.coverage.file_locator, omit_prefixes) + self.find_code_units(morfs, omit_prefixes) + self.directory = directory - for cu in code_units: + for cu in self.code_units: try: filename, statements, excluded, missing, _ = self.coverage.analyze(cu) self.annotate_file(filename, statements, excluded, missing) -- cgit v1.2.1