buildtools/checkdeps/results.py

   1 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5
   6 """Results object and results formatters for checkdeps tool."""
   7
   8
   9 import json
  10
  11
  12 class DependencyViolation(object):
  13   """A single dependency violation."""
  14
  15   def __init__(self, include_path, violated_rule, rules):
  16     # The include or import path that is in violation of a rule.
  17     self.include_path = include_path
  18
  19     # The violated rule.
  20     self.violated_rule = violated_rule
  21
  22     # The set of rules containing self.violated_rule.
  23     self.rules = rules
  24
  25
  26 class DependeeStatus(object):
  27   """Results object for a dependee file."""
  28
  29   def __init__(self, dependee_path):
  30     # Path of the file whose nonconforming dependencies are listed in
  31     # self.violations.
  32     self.dependee_path = dependee_path
  33
  34     # List of DependencyViolation objects that apply to the dependee
  35     # file.  May be empty.
  36     self.violations = []
  37
  38   def AddViolation(self, violation):
  39     """Adds a violation."""
  40     self.violations.append(violation)
  41
  42   def HasViolations(self):
  43     """Returns True if this dependee is violating one or more rules."""
  44     return not not self.violations
  45
  46
  47 class ResultsFormatter(object):
  48   """Base class for results formatters."""
  49
  50   def AddError(self, dependee_status):
  51     """Add a formatted result to |self.results| for |dependee_status|,
  52     which is guaranteed to return True for
  53     |dependee_status.HasViolations|.
  54     """
  55     raise NotImplementedError()
  56
  57   def GetResults(self):
  58     """Returns the results.  May be overridden e.g. to process the
  59     results that have been accumulated.
  60     """
  61     raise NotImplementedError()
  62
  63   def PrintResults(self):
  64     """Prints the results to stdout."""
  65     raise NotImplementedError()
  66
  67
  68 class NormalResultsFormatter(ResultsFormatter):
  69   """A results formatting object that produces the classical,
  70   detailed, human-readable output of the checkdeps tool.
  71   """
  72
  73   def __init__(self, verbose):
  74     self.results = []
  75     self.verbose = verbose
  76
  77   def AddError(self, dependee_status):
  78     lines = []
  79     lines.append('\nERROR in %s' % dependee_status.dependee_path)
  80     for violation in dependee_status.violations:
  81       lines.append(self.FormatViolation(violation, self.verbose))
  82     self.results.append('\n'.join(lines))
  83
  84   @staticmethod
  85   def FormatViolation(violation, verbose=False):
  86     lines = []
  87     if verbose:
  88       lines.append('  For %s' % violation.rules)
  89     lines.append(
  90         '  Illegal include: "%s"\n    Because of %s' %
  91         (violation.include_path, str(violation.violated_rule)))
  92     return '\n'.join(lines)
  93
  94   def GetResults(self):
  95     return self.results
  96
  97   def PrintResults(self):
  98     for result in self.results:
  99       print result
 100     if self.results:
 101       print '\nFAILED\n'
 102
 103
 104 class JSONResultsFormatter(ResultsFormatter):
 105   """A results formatter that outputs results to a file as JSON."""
 106
 107   def __init__(self, output_path, wrapped_formatter=None):
 108     self.output_path = output_path
 109     self.wrapped_formatter = wrapped_formatter
 110
 111     self.results = []
 112
 113   def AddError(self, dependee_status):
 114     self.results.append({
 115         'dependee_path': dependee_status.dependee_path,
 116         'violations': [{
 117             'include_path': violation.include_path,
 118             'violated_rule': violation.violated_rule.AsDependencyTuple(),
 119         } for violation in dependee_status.violations]
 120     })
 121
 122     if self.wrapped_formatter:
 123       self.wrapped_formatter.AddError(dependee_status)
 124
 125   def GetResults(self):
 126     with open(self.output_path, 'w') as f:
 127       f.write(json.dumps(self.results))
 128
 129     return self.results
 130
 131   def PrintResults(self):
 132     if self.wrapped_formatter:
 133       self.wrapped_formatter.PrintResults()
 134       return
 135
 136     print self.results
 137
 138
 139 class TemporaryRulesFormatter(ResultsFormatter):
 140   """A results formatter that produces a single line per nonconforming
 141   include. The combined output is suitable for directly pasting into a
 142   DEPS file as a list of temporary-allow rules.
 143   """
 144
 145   def __init__(self):
 146     self.violations = set()
 147
 148   def AddError(self, dependee_status):
 149     for violation in dependee_status.violations:
 150       self.violations.add(violation.include_path)
 151
 152   def GetResults(self):
 153     return ['  "!%s",' % path for path in sorted(self.violations)]
 154
 155   def PrintResults(self):
 156     for result in self.GetResults():
 157       print result
 158
 159
 160 class CountViolationsFormatter(ResultsFormatter):
 161   """A results formatter that produces a number, the count of #include
 162   statements that are in violation of the dependency rules.
 163
 164   Note that you normally want to instantiate DepsChecker with
 165   ignore_temp_rules=True when you use this formatter.
 166   """
 167
 168   def __init__(self):
 169     self.count = 0
 170
 171   def AddError(self, dependee_status):
 172     self.count += len(dependee_status.violations)
 173
 174   def GetResults(self):
 175     return '%d' % self.count
 176
 177   def PrintResults(self):
 178     print self.count