# Copyright (C) 2008-2012 Chris Ball # Gianluca Montecchi # Niall Douglas (s_sourceforge@nedprod.com) # Robert Lehmann # Thomas Habets # Valtteri Kokkoniemi # W. Trevor King # # This file is part of Bugs Everywhere. # # Bugs Everywhere is free software: you can redistribute it and/or modify it # under the terms of the GNU General Public License as published by the Free # Software Foundation, either version 2 of the License, or (at your option) any # later version. # # Bugs Everywhere is distributed in the hope that it will be useful, but # WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or # FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for # more details. # # You should have received a copy of the GNU General Public License along with # Bugs Everywhere. If not, see . """Define :py:class:`Bug` for representing bugs. """ import copy import os import os.path import errno import sys import time import types try: # import core module, Python >= 2.5 from xml.etree import ElementTree except ImportError: # look for non-core module from elementtree import ElementTree import xml.sax.saxutils import libbe import libbe.util.id from libbe.storage.util.properties import Property, doc_property, \ local_property, defaulting_property, checked_property, cached_property, \ primed_property, change_hook_property, settings_property import libbe.storage.util.settings_object as settings_object import libbe.storage.util.mapfile as mapfile import libbe.comment as comment import libbe.util.utility as utility if libbe.TESTING == True: import doctest ### Define and describe valid bug categories # Use a tuple of (category, description) tuples since we don't have # ordered dicts in Python yet http://www.python.org/dev/peps/pep-0372/ # in order of increasing severity. (name, description) pairs severity_def = ( ("target", "The issue is a target or milestone, not a bug."), ("wishlist","A feature that could improve usefulness, but not a bug."), ("minor","The standard bug level."), ("serious","A bug that requires workarounds."), ("critical","A bug that prevents some features from working at all."), ("fatal","A bug that makes the package unusable.")) # in order of increasing resolution # roughly following http://www.bugzilla.org/docs/3.2/en/html/lifecycle.html active_status_def = ( ("unconfirmed","A possible bug which lacks independent existance confirmation."), ("open","A working bug that has not been assigned to a developer."), ("assigned","A working bug that has been assigned to a developer."), ("test","The code has been adjusted, but the fix is still being tested.")) inactive_status_def = ( ("closed", "The bug is no longer relevant."), ("fixed", "The bug should no longer occur."), ("wontfix","It's not a bug, it's a feature.")) ### Convert the description tuples to more useful formats severity_values = () severity_description = {} severity_index = {} def load_severities(severity_def): global severity_values global severity_description global severity_index if severity_def == None: return severity_values = tuple([val for val,description in severity_def]) severity_description = dict(severity_def) severity_index = {} for i,severity in enumerate(severity_values): severity_index[severity] = i load_severities(severity_def) active_status_values = [] inactive_status_values = [] status_values = [] status_description = {} status_index = {} def load_status(active_status_def, inactive_status_def): global active_status_values global inactive_status_values global status_values global status_description global status_index if active_status_def == None: active_status_def = globals()["active_status_def"] if inactive_status_def == None: inactive_status_def = globals()["inactive_status_def"] active_status_values = tuple([val for val,description in active_status_def]) inactive_status_values = tuple([val for val,description in inactive_status_def]) status_values = active_status_values + inactive_status_values status_description = dict(tuple(active_status_def) + tuple(inactive_status_def)) status_index = {} for i,status in enumerate(status_values): status_index[status] = i load_status(active_status_def, inactive_status_def) class Bug (settings_object.SavedSettingsObject): """A bug (or issue) is a place to store attributes and attach :py:class:`~libbe.comment.Comment`\s. In mailing-list terms, a bug is analogous to a thread. Bugs are normally stored in :py:class:`~libbe.bugdir.BugDir`\s. >>> b = Bug() >>> print b.status open >>> print b.severity minor There are two formats for time, int and string. Setting either one will adjust the other appropriately. The string form is the one stored in the bug's settings file on disk. >>> print type(b.time) >>> print type(b.time_string) >>> b.time = 0 >>> print b.time_string Thu, 01 Jan 1970 00:00:00 +0000 >>> b.time_string="Thu, 01 Jan 1970 00:01:00 +0000" >>> b.time 60 >>> print b.settings["time"] Thu, 01 Jan 1970 00:01:00 +0000 """ settings_properties = [] required_saved_properties = [] _prop_save_settings = settings_object.prop_save_settings _prop_load_settings = settings_object.prop_load_settings def _versioned_property(settings_properties=settings_properties, required_saved_properties=required_saved_properties, **kwargs): if "settings_properties" not in kwargs: kwargs["settings_properties"] = settings_properties if "required_saved_properties" not in kwargs: kwargs["required_saved_properties"]=required_saved_properties return settings_object.versioned_property(**kwargs) @_versioned_property(name="severity", doc="A measure of the bug's importance", default="minor", check_fn=lambda s: s in severity_values, require_save=True) def severity(): return {} @_versioned_property(name="status", doc="The bug's current status", default="open", check_fn=lambda s: s in status_values, require_save=True) def status(): return {} @property def active(self): return self.status in active_status_values @_versioned_property(name="creator", doc="The user who entered the bug into the system") def creator(): return {} @_versioned_property(name="reporter", doc="The user who reported the bug") def reporter(): return {} @_versioned_property(name="assigned", doc="The developer in charge of the bug") def assigned(): return {} @_versioned_property(name="time", doc="An RFC 2822 timestamp for bug creation") def time_string(): return {} def _get_time(self): if self.time_string == None: self._cached_time_string = None self._cached_time = None return None if (not hasattr(self, '_cached_time_string') or self.time_string != self._cached_time_string): self._cached_time_string = self.time_string self._cached_time = utility.str_to_time(self.time_string) return self._cached_time def _set_time(self, value): if not hasattr(self, '_cached_time') or value != self._cached_time: self.time_string = utility.time_to_str(value) self._cached_time_string = self.time_string self._cached_time = value time = property(fget=_get_time, fset=_set_time, doc="An integer version of .time_string") def _extra_strings_check_fn(value): return utility.iterable_full_of_strings(value, \ alternative=settings_object.EMPTY) def _extra_strings_change_hook(self, old, new): self.extra_strings.sort() # to make merging easier self._prop_save_settings(old, new) @_versioned_property(name="extra_strings", doc="Space for an array of extra strings. Useful for storing state for functionality implemented purely in becommands/.py.", default=[], check_fn=_extra_strings_check_fn, change_hook=_extra_strings_change_hook, mutable=True) def extra_strings(): return {} @_versioned_property(name="summary", doc="A one-line bug description") def summary(): return {} def _get_comment_root(self, load_full=False): if self.storage != None and self.storage.is_readable(): return comment.load_comments(self, load_full=load_full) else: return comment.Comment(self, uuid=comment.INVALID_UUID) @Property @cached_property(generator=_get_comment_root) @local_property("comment_root") @doc_property(doc="The trunk of the comment tree. We use a dummy root comment by default, because there can be several comment threads rooted on the same parent bug. To simplify comment interaction, we condense these threads into a single thread with a Comment dummy root.") def comment_root(): return {} def __init__(self, bugdir=None, uuid=None, from_storage=False, load_comments=False, summary=None): settings_object.SavedSettingsObject.__init__(self) self.bugdir = bugdir self.storage = None self.uuid = uuid self.id = libbe.util.id.ID(self, 'bug') if from_storage == False: if uuid == None: self.uuid = libbe.util.id.uuid_gen() self.time = int(time.time()) # only save to second precision self.summary = summary dummy = self.comment_root if self.bugdir != None: self.storage = self.bugdir.storage if from_storage == False: if self.storage != None and self.storage.is_writeable(): self.save() def __repr__(self): return "Bug(uuid=%r)" % self.uuid def __str__(self): return self.string(shortlist=True) def __cmp__(self, other): return cmp_full(self, other) # serializing methods def _setting_attr_string(self, setting): value = getattr(self, setting) if value == None: return "" if type(value) not in types.StringTypes: return str(value) return value def string(self, shortlist=False, show_comments=False): if shortlist == False: if self.time == None: timestring = "" else: htime = utility.handy_time(self.time) timestring = "%s (%s)" % (htime, self.time_string) info = [("ID", self.uuid), ("Short name", self.id.user()), ("Severity", self.severity), ("Status", self.status), ("Assigned", self._setting_attr_string("assigned")), ("Reporter", self._setting_attr_string("reporter")), ("Creator", self._setting_attr_string("creator")), ("Created", timestring)] for estr in self.extra_strings: info.append(('Extra string', estr)) longest_key_len = max([len(k) for k,v in info]) infolines = [" %*s : %s\n" %(longest_key_len,k,v) for k,v in info] bugout = "".join(infolines) + "%s" % self.summary.rstrip('\n') else: statuschar = self.status[0] severitychar = self.severity[0] chars = "%c%c" % (statuschar, severitychar) bugout = "%s:%s: %s" % (self.id.user(),chars,self.summary.rstrip('\n')) if show_comments == True: self.comment_root.sort(cmp=libbe.comment.cmp_time, reverse=True) comout = self.comment_root.string_thread(flatten=False) output = bugout + '\n' + comout.rstrip('\n') else : output = bugout return output def xml(self, indent=0, show_comments=False): """ >>> bugA = Bug(uuid='0123', summary='Need to test Bug.xml()') >>> bugA.uuid = 'bugA' >>> bugA.time_string = 'Thu, 01 Jan 1970 00:00:00 +0000' >>> bugA.creator = u'Frank' >>> bugA.extra_strings += ['TAG: very helpful'] >>> commA = bugA.comment_root.new_reply(body='comment A') >>> commA.uuid = 'commA' >>> commA.date = 'Thu, 01 Jan 1970 00:01:00 +0000' >>> commB = commA.new_reply(body='comment B') >>> commB.uuid = 'commB' >>> commB.date = 'Thu, 01 Jan 1970 00:02:00 +0000' >>> commC = commB.new_reply(body='comment C') >>> commC.uuid = 'commC' >>> commC.date = 'Thu, 01 Jan 1970 00:03:00 +0000' >>> print(bugA.xml(show_comments=True)) # doctest: +REPORT_UDIFF bugA /bug minor open Frank Thu, 01 Jan 1970 00:00:00 +0000 Need to test Bug.xml() TAG: very helpful commA /bug/commA Thu, 01 Jan 1970 00:01:00 +0000 text/plain comment A commB /bug/commB commA Thu, 01 Jan 1970 00:02:00 +0000 text/plain comment B commC /bug/commC commB Thu, 01 Jan 1970 00:03:00 +0000 text/plain comment C >>> print(bugA.xml(show_comments=True, indent=2)) ... # doctest: +REPORT_UDIFF bugA /bug minor open Frank Thu, 01 Jan 1970 00:00:00 +0000 Need to test Bug.xml() TAG: very helpful commA /bug/commA Thu, 01 Jan 1970 00:01:00 +0000 text/plain comment A commB /bug/commB commA Thu, 01 Jan 1970 00:02:00 +0000 text/plain comment B commC /bug/commC commB Thu, 01 Jan 1970 00:03:00 +0000 text/plain comment C """ if self.time == None: timestring = "" else: timestring = utility.time_to_str(self.time) info = [('uuid', self.uuid), ('short-name', self.id.user()), ('severity', self.severity), ('status', self.status), ('assigned', self.assigned), ('reporter', self.reporter), ('creator', self.creator), ('created', timestring), ('summary', self.summary)] lines = [''] for (k,v) in info: if v is not None: lines.append(' <%s>%s' % (k,xml.sax.saxutils.escape(v),k)) for estr in self.extra_strings: lines.append(' %s' % estr) if show_comments == True: comout = self.comment_root.xml_thread(indent=indent+2) if comout: comout = comout[indent:] # strip leading indent spaces lines.append(comout) lines.append('') istring = ' '*indent sep = '\n' + istring return istring + sep.join(lines).rstrip('\n') def from_xml(self, xml_string, preserve_uuids=False, verbose=True): u""" Note: If a bug uuid is given, set .alt_id to it's value. >>> bugA = Bug(uuid="0123", summary="Need to test Bug.from_xml()") >>> bugA.date = "Thu, 01 Jan 1970 00:00:00 +0000" >>> bugA.creator = u'Fran\xe7ois' >>> bugA.extra_strings += ['TAG: very helpful'] >>> commA = bugA.comment_root.new_reply(body='comment A') >>> commB = bugA.comment_root.new_reply(body='comment B') >>> commC = commA.new_reply(body='comment C') >>> xml = bugA.xml(show_comments=True) >>> bugB = Bug() >>> bugB.from_xml(xml, verbose=True) >>> bugB.xml(show_comments=True) == xml False >>> bugB.uuid = bugB.alt_id >>> for comm in bugB.comments(): ... comm.uuid = comm.alt_id ... comm.alt_id = None >>> bugB.xml(show_comments=True) == xml True >>> bugB.explicit_attrs # doctest: +NORMALIZE_WHITESPACE ['severity', 'status', 'creator', 'time', 'summary'] >>> len(list(bugB.comments())) 3 >>> bugC = Bug() >>> bugC.from_xml(xml, preserve_uuids=True) >>> bugC.uuid == bugA.uuid True """ if type(xml_string) == types.UnicodeType: xml_string = xml_string.strip().encode('unicode_escape') if hasattr(xml_string, 'getchildren'): # already an ElementTree Element bug = xml_string else: bug = ElementTree.XML(xml_string) if bug.tag != 'bug': raise utility.InvalidXML( \ 'bug', bug, 'root element must be ') tags=['uuid','short-name','severity','status','assigned', 'reporter', 'creator','created','summary','extra-string'] self.explicit_attrs = [] uuid = None estrs = [] comments = [] for child in bug.getchildren(): if child.tag == 'short-name': pass elif child.tag == 'comment': comm = comment.Comment(bug=self) comm.from_xml( child, preserve_uuids=preserve_uuids, verbose=verbose) comments.append(comm) continue elif child.tag in tags: if child.text == None or len(child.text) == 0: text = settings_object.EMPTY else: text = xml.sax.saxutils.unescape(child.text) if not isinstance(text, unicode): text = text.decode('unicode_escape') text = text.strip() if child.tag == 'uuid' and not preserve_uuids: uuid = text continue # don't set the bug's uuid tag. elif child.tag == 'created': if text is not settings_object.EMPTY: self.time = utility.str_to_time(text) self.explicit_attrs.append('time') continue elif child.tag == 'extra-string': estrs.append(text) continue # don't set the bug's extra_string yet. attr_name = child.tag.replace('-','_') self.explicit_attrs.append(attr_name) setattr(self, attr_name, text) elif verbose == True: print >> sys.stderr, 'Ignoring unknown tag %s in %s' \ % (child.tag, comment.tag) if uuid != self.uuid: if not hasattr(self, 'alt_id') or self.alt_id == None: self.alt_id = uuid self.extra_strings = estrs self.add_comments(comments, ignore_missing_references=True) def add_comment(self, comment, *args, **kwargs): """ Add a comment too the current bug, under the parent specified by comment.in_reply_to. Note: If a bug uuid is given, set .alt_id to it's value. >>> bugA = Bug(uuid='0123', summary='Need to test Bug.add_comment()') >>> bugA.creator = 'Jack' >>> commA = bugA.comment_root.new_reply(body='comment A') >>> commA.uuid = 'commA' >>> commB = comment.Comment(body='comment B') >>> commB.uuid = 'commB' >>> bugA.add_comment(commB) >>> commC = comment.Comment(body='comment C') >>> commC.uuid = 'commC' >>> commC.in_reply_to = commA.uuid >>> bugA.add_comment(commC) >>> print bugA.xml(show_comments=True) # doctest: +ELLIPSIS 0123 /012 minor open Jack ... Need to test Bug.add_comment() commA /012/commA ... text/plain comment A commC /012/commC commA ... text/plain comment C commB /012/commB ... text/plain comment B """ self.add_comments([comment], **kwargs) def add_comments(self, comments, default_parent=None, ignore_missing_references=False): """ Convert a raw list of comments to single root comment. If a comment does not specify a parent with .in_reply_to, the parent defaults to .comment_root, but you can specify another default parent via default_parent. """ uuid_map = {} if default_parent == None: default_parent = self.comment_root for c in list(self.comments()) + comments: assert c.uuid != None assert c.uuid not in uuid_map uuid_map[c.uuid] = c if c.alt_id != None: uuid_map[c.alt_id] = c uuid_map[None] = self.comment_root uuid_map[comment.INVALID_UUID] = self.comment_root if default_parent != self.comment_root: assert default_parent.uuid in uuid_map, default_parent.uuid for c in comments: if c.in_reply_to == None \ and default_parent.uuid != comment.INVALID_UUID: c.in_reply_to = default_parent.uuid elif c.in_reply_to == comment.INVALID_UUID: c.in_reply_to = None try: parent = uuid_map[c.in_reply_to] except KeyError: if ignore_missing_references == True: print >> sys.stderr, \ 'Ignoring missing reference to %s' % c.in_reply_to parent = default_parent if parent.uuid != comment.INVALID_UUID: c.in_reply_to = parent.uuid else: raise comment.MissingReference(c) c.bug = self parent.append(c) def merge(self, other, accept_changes=True, accept_extra_strings=True, accept_comments=True, change_exception=False): """ Merge info from other into this bug. Overrides any attributes in self that are listed in other.explicit_attrs. >>> bugA = Bug(uuid='0123', summary='Need to test Bug.merge()') >>> bugA.date = 'Thu, 01 Jan 1970 00:00:00 +0000' >>> bugA.creator = 'Frank' >>> bugA.extra_strings += ['TAG: very helpful'] >>> bugA.extra_strings += ['TAG: favorite'] >>> commA = bugA.comment_root.new_reply(body='comment A') >>> commA.uuid = 'uuid-commA' >>> bugB = Bug(uuid='3210', summary='More tests for Bug.merge()') >>> bugB.date = 'Fri, 02 Jan 1970 00:00:00 +0000' >>> bugB.creator = 'John' >>> bugB.explicit_attrs = ['creator', 'summary'] >>> bugB.extra_strings += ['TAG: very helpful'] >>> bugB.extra_strings += ['TAG: useful'] >>> commB = bugB.comment_root.new_reply(body='comment B') >>> commB.uuid = 'uuid-commB' >>> bugA.merge(bugB, accept_changes=False, accept_extra_strings=False, ... accept_comments=False, change_exception=False) >>> print bugA.creator Frank >>> bugA.merge(bugB, accept_changes=False, accept_extra_strings=False, ... accept_comments=False, change_exception=True) Traceback (most recent call last): ... ValueError: Merge would change creator "Frank"->"John" for bug 0123 >>> print bugA.creator Frank >>> bugA.merge(bugB, accept_changes=True, accept_extra_strings=False, ... accept_comments=False, change_exception=True) Traceback (most recent call last): ... ValueError: Merge would add extra string "TAG: useful" for bug 0123 >>> print bugA.creator John >>> print bugA.extra_strings ['TAG: favorite', 'TAG: very helpful'] >>> bugA.merge(bugB, accept_changes=True, accept_extra_strings=True, ... accept_comments=False, change_exception=True) Traceback (most recent call last): ... ValueError: Merge would add comment uuid-commB (alt: None) to bug 0123 >>> print bugA.extra_strings ['TAG: favorite', 'TAG: useful', 'TAG: very helpful'] >>> bugA.merge(bugB, accept_changes=True, accept_extra_strings=True, ... accept_comments=True, change_exception=True) >>> print bugA.xml(show_comments=True) # doctest: +ELLIPSIS 0123 /012 minor open John ... More tests for Bug.merge() TAG: favorite TAG: useful TAG: very helpful uuid-commA /012/uuid-commA ... text/plain comment A uuid-commB /012/uuid-commB ... text/plain comment B """ if hasattr(other, 'explicit_attrs'): for attr in other.explicit_attrs: old = getattr(self, attr) new = getattr(other, attr) if old != new: if accept_changes: setattr(self, attr, new) elif change_exception: raise ValueError( ('Merge would change {} "{}"->"{}" for bug {}' ).format(attr, old, new, self.uuid)) for estr in other.extra_strings: if not estr in self.extra_strings: if accept_extra_strings == True: self.extra_strings += [estr] elif change_exception == True: raise ValueError, \ 'Merge would add extra string "%s" for bug %s' \ % (estr, self.uuid) for o_comm in other.comments(): try: s_comm = self.comment_root.comment_from_uuid(o_comm.uuid) except KeyError, e: try: s_comm = self.comment_root.comment_from_uuid(o_comm.alt_id) except KeyError, e: s_comm = None if s_comm == None: if accept_comments == True: o_comm_copy = copy.copy(o_comm) o_comm_copy.bug = self o_comm_copy.id = libbe.util.id.ID(o_comm_copy, 'comment') self.comment_root.add_reply(o_comm_copy) elif change_exception == True: raise ValueError, \ 'Merge would add comment %s (alt: %s) to bug %s' \ % (o_comm.uuid, o_comm.alt_id, self.uuid) else: s_comm.merge(o_comm, accept_changes=accept_changes, accept_extra_strings=accept_extra_strings, change_exception=change_exception) # methods for saving/loading/acessing settings and properties. def load_settings(self, settings_mapfile=None): if settings_mapfile == None: settings_mapfile = self.storage.get( self.id.storage('values'), '{}\n') try: settings = mapfile.parse(settings_mapfile) except mapfile.InvalidMapfileContents, e: raise Exception('Invalid settings file for bug %s\n' '(BE version missmatch?)' % self.id.user()) self._setup_saved_settings(settings) def save_settings(self): mf = mapfile.generate(self._get_saved_settings()) self.storage.set(self.id.storage('values'), mf) def save(self): """ Save any loaded contents to storage. Because of lazy loading of comments, this is actually not too inefficient. However, if self.storage.is_writeable() == True, then any changes are automatically written to storage as soon as they happen, so calling this method will just waste time (unless something else has been messing with your stored files). """ assert self.storage != None, "Can't save without storage" if self.bugdir != None: parent = self.bugdir.id.storage() else: parent = None self.storage.add(self.id.storage(), parent=parent, directory=True) self.storage.add(self.id.storage('values'), parent=self.id.storage(), directory=False) self.save_settings() if len(self.comment_root) > 0: comment.save_comments(self) def load_comments(self, load_full=True): if load_full == True: # Force a complete load of the whole comment tree self.comment_root = self._get_comment_root(load_full=True) else: # Setup for fresh lazy-loading. Clear _comment_root, so # next _get_comment_root returns a fresh version. Turn of # writing temporarily so we don't write our blank comment # tree to disk. w = self.storage.writeable self.storage.writeable = False self.comment_root = None self.storage.writeable = w def remove(self): self.storage.recursive_remove(self.id.storage()) # methods for managing comments def uuids(self): for comment in self.comments(): yield comment.uuid def comments(self): for comment in self.comment_root.traverse(): yield comment def new_comment(self, body=None): comm = self.comment_root.new_reply(body=body) return comm def comment_from_uuid(self, uuid, *args, **kwargs): return self.comment_root.comment_from_uuid(uuid, *args, **kwargs) # methods for id generation def sibling_uuids(self): if self.bugdir != None: return self.bugdir.uuids() return [] # The general rule for bug sorting is that "more important" bugs are # less than "less important" bugs. This way sorting a list of bugs # will put the most important bugs first in the list. When relative # importance is unclear, the sorting follows some arbitrary convention # (i.e. dictionary order). def cmp_severity(bug_1, bug_2): """ Compare the severity levels of two bugs, with more severe bugs comparing as less. >>> bugA = Bug() >>> bugB = Bug() >>> bugA.severity = bugB.severity = "wishlist" >>> cmp_severity(bugA, bugB) == 0 True >>> bugB.severity = "minor" >>> cmp_severity(bugA, bugB) > 0 True >>> bugA.severity = "critical" >>> cmp_severity(bugA, bugB) < 0 True """ if not hasattr(bug_2, "severity") : return 1 return -cmp(severity_index[bug_1.severity], severity_index[bug_2.severity]) def cmp_status(bug_1, bug_2): """ Compare the status levels of two bugs, with more "open" bugs comparing as less. >>> bugA = Bug() >>> bugB = Bug() >>> bugA.status = bugB.status = "open" >>> cmp_status(bugA, bugB) == 0 True >>> bugB.status = "closed" >>> cmp_status(bugA, bugB) < 0 True >>> bugA.status = "fixed" >>> cmp_status(bugA, bugB) > 0 True """ if not hasattr(bug_2, "status") : return 1 val_2 = status_index[bug_2.status] return cmp(status_index[bug_1.status], status_index[bug_2.status]) def cmp_attr(bug_1, bug_2, attr, invert=False): """ Compare a general attribute between two bugs using the conventional comparison rule for that attribute type. If ``invert==True``, sort *against* that convention. >>> attr="severity" >>> bugA = Bug() >>> bugB = Bug() >>> bugA.severity = "critical" >>> bugB.severity = "wishlist" >>> cmp_attr(bugA, bugB, attr) < 0 True >>> cmp_attr(bugA, bugB, attr, invert=True) > 0 True >>> bugB.severity = "critical" >>> cmp_attr(bugA, bugB, attr) == 0 True """ if not hasattr(bug_2, attr) : return 1 val_1 = getattr(bug_1, attr) val_2 = getattr(bug_2, attr) if val_1 == None: val_1 = None if val_2 == None: val_2 = None if invert == True : return -cmp(val_1, val_2) else : return cmp(val_1, val_2) # alphabetical rankings (a < z) cmp_uuid = lambda bug_1, bug_2 : cmp_attr(bug_1, bug_2, "uuid") cmp_creator = lambda bug_1, bug_2 : cmp_attr(bug_1, bug_2, "creator") cmp_assigned = lambda bug_1, bug_2 : cmp_attr(bug_1, bug_2, "assigned") cmp_reporter = lambda bug_1, bug_2 : cmp_attr(bug_1, bug_2, "reporter") cmp_summary = lambda bug_1, bug_2 : cmp_attr(bug_1, bug_2, "summary") cmp_extra_strings = lambda bug_1, bug_2 : cmp_attr(bug_1, bug_2, "extra_strings") # chronological rankings (newer < older) cmp_time = lambda bug_1, bug_2 : cmp_attr(bug_1, bug_2, "time", invert=True) def cmp_mine(bug_1, bug_2): user_id = libbe.ui.util.user.get_user_id(bug_1.storage) mine_1 = bug_1.assigned != user_id mine_2 = bug_2.assigned != user_id return cmp(mine_1, mine_2) def cmp_comments(bug_1, bug_2): """ Compare two bugs' comments lists. Doesn't load any new comments, so you should call each bug's .load_comments() first if you want a full comparison. """ comms_1 = sorted(bug_1.comments(), key = lambda comm : comm.uuid) comms_2 = sorted(bug_2.comments(), key = lambda comm : comm.uuid) result = cmp(len(comms_1), len(comms_2)) if result != 0: return result for c_1,c_2 in zip(comms_1, comms_2): result = cmp(c_1, c_2) if result != 0: return result return 0 DEFAULT_CMP_FULL_CMP_LIST = \ (cmp_status, cmp_severity, cmp_assigned, cmp_time, cmp_creator, cmp_reporter, cmp_comments, cmp_summary, cmp_uuid, cmp_extra_strings) class BugCompoundComparator (object): def __init__(self, cmp_list=DEFAULT_CMP_FULL_CMP_LIST): self.cmp_list = cmp_list def __call__(self, bug_1, bug_2): for comparison in self.cmp_list : val = comparison(bug_1, bug_2) if val != 0 : return val return 0 cmp_full = BugCompoundComparator() # define some bonus cmp_* functions def cmp_last_modified(bug_1, bug_2): """ Like cmp_time(), but use most recent comment instead of bug creation for the timestamp. """ def last_modified(bug): time = bug.time for comment in bug.comment_root.traverse(): if comment.time > time: time = comment.time return time val_1 = last_modified(bug_1) val_2 = last_modified(bug_2) return -cmp(val_1, val_2) if libbe.TESTING == True: suite = doctest.DocTestSuite()