diff --git a/wechat/msg.py b/wechat/msg.py index 46a63b6..a76bd0b 100644 --- a/wechat/msg.py +++ b/wechat/msg.py @@ -91,6 +91,7 @@ class WeChatMsg(object): return self.content elif self.type == TYPE_REDENVELOPE: pq = PyQuery(self.content_xml_ready, parser='xml') + title = pq('sendertitle').text() return u"[RED ENVELOPE]\n{}".format(title) else: diff --git a/wechat/parser.py b/wechat/parser.py index f524ea1..251fd21 100644 --- a/wechat/parser.py +++ b/wechat/parser.py @@ -135,6 +135,7 @@ SELECT {} FROM message if values['chat'].endswith('@chatroom'): values['chat'] = self.contacts[values['chat']] content = values['content'] + if values['isSend'] == 1: values['talker'] = self.username elif values['type'] == TYPE_SYSTEM: @@ -142,7 +143,8 @@ SELECT {} FROM message else: talker = content[:content.find(':')] values['talker'] = self.contacts.get(talker, talker) - values['content'] = content[content.find('\n') + 1:] + + values['content'] = content[content.find('\n') + 1:] else: tk_id = values['talker'] values['chat'] = self.contacts[tk_id]