ai-on-wechat/channel/wechatmp/receive.py

# -*- coding: utf-8 -*-#
# filename: receive.py
import xml.etree.ElementTree as ET
from bridge.context import ContextType
from channel.chat_message import ChatMessage
from common.log import logger


def parse_xml(web_data):
    if len(web_data) == 0:
        return None
    xmlData = ET.fromstring(web_data)
    return WeChatMPMessage(xmlData)

class WeChatMPMessage(ChatMessage):
    def __init__(self, xmlData):
        super().__init__(xmlData)
        self.to_user_id = xmlData.find('ToUserName').text
        self.from_user_id = xmlData.find('FromUserName').text
        self.create_time = xmlData.find('CreateTime').text
        self.msg_type = xmlData.find('MsgType').text
        self.msg_id = xmlData.find('MsgId').text
        self.is_group = False
        
        # reply to other_user_id
        self.other_user_id = self.from_user_id

        if self.msg_type == 'text':
            self.ctype = ContextType.TEXT
            self.content = xmlData.find('Content').text.encode("utf-8")
        elif self.msg_type == 'voice':
            self.ctype = ContextType.TEXT
            self.content = xmlData.find('Recognition').text.encode("utf-8")  # 接收语音识别结果
        elif self.msg_type == 'image':
            # not implemented
            self.pic_url = xmlData.find('PicUrl').text
            self.media_id = xmlData.find('MediaId').text
        elif self.msg_type == 'event':
            self.event = xmlData.find('Event').text
        else: # video, shortvideo, location, link
            # not implemented
            pass