libbe/diff.py

   1 # Copyright (C) 2005 Aaron Bentley and Panometrics, Inc.
   2 # <abentley@panoramicfeedback.com>
   3 #
   4 #    This program is free software; you can redistribute it and/or modify
   5 #    it under the terms of the GNU General Public License as published by
   6 #    the Free Software Foundation; either version 2 of the License, or
   7 #    (at your option) any later version.
   8 #
   9 #    This program is distributed in the hope that it will be useful,
  10 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 #    GNU General Public License for more details.
  13 #
  14 #    You should have received a copy of the GNU General Public License
  15 #    along with this program; if not, write to the Free Software
  16 #    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17 """Compare two bug trees"""
  18 from libbe import cmdutil, bugdir
  19
  20 def diff(old_tree, new_tree):
  21     old_bug_map = old_tree.bug_map()
  22     new_bug_map = new_tree.bug_map()
  23     added = []
  24     removed = []
  25     modified = []
  26     for old_bug in old_bug_map.itervalues():
  27         new_bug = new_bug_map.get(old_bug.uuid)
  28         if new_bug is None :
  29             removed.append(old_bug)
  30         else:
  31             if old_bug != new_bug:
  32                 modified.append((old_bug, new_bug))
  33     for new_bug in new_bug_map.itervalues():
  34         if not old_bug_map.has_key(new_bug.uuid):
  35             added.append(new_bug)
  36     return (removed, modified, added)
  37
  38
  39 def reference_diff(bugdir, spec=None):
  40     return diff(bugdir.get_reference_bugdir(spec), bugdir)
  41
  42 def diff_report(diff_data, bug_dir):
  43     (removed, modified, added) = diff_data
  44     bugs = list(bug_dir.list())
  45     def modified_cmp(left, right):
  46         return bugdir.cmp_severity(left[1], right[1])
  47
  48     added.sort(bugdir.cmp_severity)
  49     removed.sort(bugdir.cmp_severity)
  50     modified.sort(modified_cmp)
  51
  52     if len(added) > 0:
  53         print "New bug reports:"
  54         for bug in added:
  55             print cmdutil.bug_summary(bug, bugs, no_target=True)
  56
  57     if len(modified) > 0:
  58         printed = False
  59         for old_bug, new_bug in modified:
  60             change_str = bug_changes(old_bug, new_bug, bugs)
  61             if change_str is None:
  62                 continue
  63             if not printed:
  64                 printed = True
  65                 print "Modified bug reports:"
  66             print change_str
  67
  68     if len(removed) > 0:
  69         print "Removed bug reports:"
  70         for bug in removed:
  71             print cmdutil.bug_summary(bug, bugs, no_target=True)
  72
  73 def change_lines(old, new, attributes):
  74     change_list = []
  75     for attr in attributes:
  76         old_attr = getattr(old, attr)
  77         new_attr = getattr(new, attr)
  78         if old_attr != new_attr:
  79             change_list.append((attr, old_attr, new_attr))
  80     if len(change_list) >= 0:
  81         return change_list
  82     else:
  83         return None
  84
  85 def bug_changes(old, new, bugs):
  86     change_list = change_lines(old, new, ("time", "creator", "severity",
  87     "target", "summary", "status", "assigned"))
  88     if len(change_list) == 0:
  89         return None
  90     return "%s%s\n" % (cmdutil.bug_summary(new, bugs, shortlist=True),
  91                        "\n".join(["%s: %s -> %s" % f for f in change_list]))
  92
  93