libbe/diff.py

   1 """Compare two bug trees"""
   2 from libbe import cmdutil, bugdir
   3
   4 def diff(old_tree, new_tree):
   5     old_bug_map = old_tree.bug_map()
   6     new_bug_map = new_tree.bug_map()
   7     added = []
   8     removed = []
   9     modified = []
  10     for old_bug in old_bug_map.itervalues():
  11         new_bug = new_bug_map.get(old_bug.uuid)
  12         if new_bug is None :
  13             removed.append(old_bug)
  14         else:
  15             if old_bug != new_bug:
  16                 modified.append((old_bug, new_bug))
  17     for new_bug in new_bug_map.itervalues():
  18         if not old_bug_map.has_key(new_bug.uuid):
  19             added.append(new_bug)
  20     return (removed, modified, added)
  21
  22
  23 def reference_diff(bugdir, spec=None):
  24     return diff(bugdir.get_reference_bugdir(spec), bugdir)
  25
  26 def diff_report(diff_data, bug_dir):
  27     (removed, modified, added) = diff_data
  28     bugs = list(bug_dir.list())
  29     def modified_cmp(left, right):
  30         return bugdir.cmp_severity(left[1], right[1])
  31
  32     added.sort(bugdir.cmp_severity)
  33     removed.sort(bugdir.cmp_severity)
  34     modified.sort(modified_cmp)
  35
  36     if len(added) > 0:
  37         print "New bug reports:"
  38         for bug in added:
  39             print cmdutil.bug_summary(bug, bugs, no_target=True)
  40
  41     if len(modified) > 0:
  42         printed = False
  43         for old_bug, new_bug in modified:
  44             change_str = bug_changes(old_bug, new_bug, bugs)
  45             if change_str is None:
  46                 continue
  47             if not printed:
  48                 printed = True
  49                 print "Modified bug reports:"
  50             print change_str
  51
  52     if len(removed) > 0:
  53         print "Removed bug reports:"
  54         for bug in removed:
  55             print cmdutil.bug_summary(bug, bugs, no_target=True)
  56
  57 def change_lines(old, new, attributes):
  58     change_list = []
  59     for attr in attributes:
  60         old_attr = getattr(old, attr)
  61         new_attr = getattr(new, attr)
  62         if old_attr != new_attr:
  63             change_list.append("%s: %s -> %s" % (attr, old_attr, new_attr))
  64     if len(change_list) >= 0:
  65         return change_list
  66     else:
  67         return None
  68
  69 def bug_changes(old, new, bugs):
  70     change_list = change_lines(old, new, ("time", "creator", "severity",
  71     "target", "summary", "status", "assigned"))
  72     if len(change_list) == 0:
  73         return None
  74     return "%s%s\n" % (cmdutil.bug_summary(new, bugs, shortlist=True),
  75                            "\n".join(change_list))
  76
  77