fix when contacts are deleted

2015-11-09 10:48:41 -05:00
parent 13e3d456e8
commit 42ee381c7e
1 changed files with 4 additions and 3 deletions
@@ -58,10 +58,11 @@ SELECT {} FROM message
            msg = WeChatMsg(row)
            if not WeChatMsg.filter_type(msg.type):
                self.msgs_by_talker[msg.talker].append(msg)
-            #if msg.type > 10000 or msg.type < 0:
-                #print repr(msg).split('|')[0]
+
+        # It's possible that messages are kept in database after contacts been deleted
+        # TODO handle this with a random contact name
        self.msgs_by_talker = {self.contacts[k]: sorted(v, key=lambda x: x.createTime)
-                           for k, v in self.msgs_by_talker.iteritems()}
+                           for k, v in self.msgs_by_talker.iteritems() if k in self.contacts}
        for k, v in self.msgs_by_talker.iteritems():
            for msg in v:
                msg.talker_name = ensure_unicode(k)