|
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061 |
-
-
- from bridge.context import ContextType
- from channel.chat_message import ChatMessage
- from common.tmp_dir import TmpDir
- from common.log import logger
- from lib.itchat.content import *
- from lib import itchat
-
- class WeChatMessage(ChatMessage):
-
- def __init__(self, itchat_msg, is_group=False):
- super().__init__( itchat_msg)
- self.msg_id = itchat_msg['MsgId']
- self.create_time = itchat_msg['CreateTime']
- self.is_group = is_group
-
- if itchat_msg['Type'] == TEXT:
- self.ctype = ContextType.TEXT
- self.content = itchat_msg['Text']
- elif itchat_msg['Type'] == VOICE:
- self.ctype = ContextType.VOICE
- self.content = TmpDir().path() + itchat_msg['FileName'] # content直接存临时目录路径
- self._prepare_fn = lambda: itchat_msg.download(self.content)
- elif itchat_msg['Type'] == PICTURE and itchat_msg['MsgType'] == 3:
- self.ctype = ContextType.IMAGE
- self.content = TmpDir().path() + itchat_msg['FileName'] # content直接存临时目录路径
- self._prepare_fn = lambda: itchat_msg.download(self.content)
- else:
- raise NotImplementedError("Unsupported message type: {}".format(itchat_msg['Type']))
-
- self.from_user_id = itchat_msg['FromUserName']
- self.to_user_id = itchat_msg['ToUserName']
-
- user_id = itchat.instance.storageClass.userName
- nickname = itchat.instance.storageClass.nickName
-
- # 虽然from_user_id和to_user_id用的少，但是为了保持一致性，还是要填充一下
- # 以下很繁琐，一句话总结：能填的都填了。
- if self.from_user_id == user_id:
- self.from_user_nickname = nickname
- if self.to_user_id == user_id:
- self.to_user_nickname = nickname
- try: # 陌生人时候, 'User'字段可能不存在
- self.other_user_id = itchat_msg['User']['UserName']
- self.other_user_nickname = itchat_msg['User']['NickName']
- if self.other_user_id == self.from_user_id:
- self.from_user_nickname = self.other_user_nickname
- if self.other_user_id == self.to_user_id:
- self.to_user_nickname = self.other_user_nickname
- except KeyError as e: # 处理偶尔没有对方信息的情况
- logger.warn("[WX]get other_user_id failed: " + str(e))
- if self.from_user_id == user_id:
- self.other_user_id = self.to_user_id
- else:
- self.other_user_id = self.from_user_id
-
- if self.is_group:
- self.is_at = itchat_msg['IsAt']
- self.actual_user_id = itchat_msg['ActualUserName']
- self.actual_user_nickname = itchat_msg['ActualNickName']
|
