nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

check types of compared objects in __eq__() operators

Browse source
This commit is contained in:
Romain Bignon 2012-03-27 11:55:24 +02:00
commit cedbc3ae77
2 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
weboob/applications/qhavedate/contacts.py
View file

@ -67,6 +67,8 @@ class ThreadMessage(QFrame):
def __eq__(self, m): def __eq__(self, m):
if not isinstance(m, Message):
return False
return self.message == m.message return self.message == m.message
class ContactThread(QWidget): class ContactThread(QWidget):

3
weboob/capabilities/messages.py
View file

@ -117,6 +117,9 @@ class Message(_Message):
return '%s.%s' % (self.thread.id, self._parent_id) return '%s.%s' % (self.thread.id, self._parent_id)
def __eq__(self, msg): def __eq__(self, msg):
if not isinstance(msg, Message):
return False
if self.thread: if self.thread:
return unicode(self.thread.id) == unicode(msg.thread.id) and \ return unicode(self.thread.id) == unicode(msg.thread.id) and \
unicode(self.id) == unicode(msg.id) unicode(self.id) == unicode(msg.id)