import string
import xml.etree.ElementTree as ET


def parse_xml(data):
    if len(data) == 0:
        return None
    xmlData = ET.fromstring(data)
    msg_type = xmlData.find('MsgType').text
    if msg_type == 'text':
        return ReqTextMsg(xmlData)
    elif msg_type == 'image':
        return ReqImageMsg(xmlData)
    else :
        return None


class ReqMsg(object):
    def __init__(self, xmlData):
        self.to_username = xmlData.find('ToUserName').text
        self.from_username = xmlData.find('FromUserName').text
        self.create_time = xmlData.find('CreateTime').text
        self.msg_id = xmlData.find('MsgId').text
        self.msg_type = xmlData.find('MsgType').text

class ReqTextMsg(ReqMsg):
    def __init__(self, xmlData):
        ReqMsg.__init__(self, xmlData)
        self.content:string = xmlData.find('Content').text


class ReqImageMsg(ReqMsg):
    def __init__(self, xmlData):
        ReqMsg.__init__(self, xmlData)
        self.pic_url = xmlData.find('PicUrl').text
        self.media_id = xmlData.find('MediaId').text