django 公众号开发:receive.py/reply.py的封装和使用

一、封装receive.py和reply.py：

在任意app模块下创建receive.py和reply.py文件。

 receive.py代码：

# -*- coding: utf-8 -*-#
# filename: receive.py
import xml.etree.ElementTree as ET


def parse_xml(web_data):
    if len(web_data) == 0:
        return None
    xmlData = ET.fromstring(web_data)
    msg_type = xmlData.find('MsgType').text
    if msg_type == 'text':
        return TextMsg(xmlData)
    elif msg_type == 'image':
        return ImageMsg(xmlData)
    elif msg_type == 'event':
        event_type = xmlData.find('Event').text
        if event_type in ('subscribe', 'unsubscribe'):
            return Subscribe(xmlData)
        elif event_type == 'CLICK':
            return Click(xmlData)
        elif event_type == 'VIEW':
            return View(xmlData)


class Msg(object):
    def __init__(self, xmlData):
        self.ToUserName = xmlData.find('ToUserName').text
        self.FromUserName = xmlData.find('FromUserName').text
        self.CreateTime = xmlData.find('CreateTime').text
        self.MsgType = xmlData.find('MsgType').text
        self.MsgId = xmlData.find('MsgId').text


class TextMsg(Msg):
    def __init__(self, xmlData):
        Msg.__init__(self, xmlData)
        self.Content = xmlData.find('Content').text.encode("utf-8")


class ImageMsg(Msg):
    def __init__(self, xmlData):
        Msg.__init__(self, xmlData)
        self.PicUrl = xmlData.find('PicUrl').text
        self.MediaId = xmlData.find('MediaId').text


class EventMsg(object):
    def __init__(self, xmlData):
        self.ToUserName = xmlData.find('ToUserName').text
        self.FromUserName = xmlData.find('FromUserName').text
        self.CreateTime = xmlData.find('CreateTime').text
        self.MsgType = xmlData.find('MsgType').text
        self.Event = xmlData.find('Event').text

class Subscribe(EventMsg):
    def __init__(self, xmlData):
        EventMsg.__init__(self, xmlData)

class Click(EventMsg):
    def __init__(self, xmlData):
        EventMsg.__init__(self, xmlData)
        self.EventKey = xmlData.find('EventKey').text

class View(EventMsg):
    def __init__(self, xmlData):
        EventMsg.__init__(self, xmlData)
        self.EventKey = xmlData.find('EventKey').text
        self.MenuId = xmlData.find('MenuId').text

reply.py代码：

# -*- coding: utf-8 -*-#
# filename: reply.py
import time
import logging
logger = logging.getLogger('log')
class Msg(object):
    def __init__(self):
        pass

    def send(self):
        return "success"

class TextMsg(Msg):
    def __init__(self, toUserName, fromUserName, content):
        self.__dict = dict()
        self.__dict['ToUserName'] = toUserName
        self.__dict['FromUserName'] = fromUserName
        self.__dict['CreateTime'] = int(time.time())
        self.__dict['Content'] = content

    def send(self):
        XmlForm = """
            <xml>
                <ToUserName><![CDATA[{ToUserName}]]></ToUserName>
                <FromUserName><![CDATA[{FromUserName}]]></FromUserName>
                <CreateTime>{CreateTime}</CreateTime>
                <MsgType><![CDATA[text]]></MsgType>
                <Content><![CDATA[{Content}]]></Content>
            </xml>
            """
        return XmlForm.format(**self.__dict)


class ImageMsg(Msg):
    def __init__(self, toUserName, fromUserName, mediaId):
        self.__dict = dict()
        self.__dict['ToUserName'] = toUserName
        self.__dict['FromUserName'] = fromUserName
        self.__dict['CreateTime'] = int(time.time())
        self.__dict['MediaId'] = mediaId

    def send(self):
        XmlForm = """
            <xml>
                <ToUserName><![CDATA[{ToUserName}]]></ToUserName>
                <FromUserName><![CDATA[{FromUserName}]]></FromUserName>
                <CreateTime>{CreateTime}</CreateTime>
                <MsgType><![CDATA[image]]></MsgType>
                <Image>
                <MediaId><![CDATA[{MediaId}]]></MediaId>
                </Image>
            </xml>
            """
        return XmlForm.format(**self.__dict)


class ArticleMsg(Msg):
    def __init__(self, toUserName, fromUserName, Title,Description,Picurl,Url):
        logger.info(Description)
        self.__dict = dict()
        self.__dict['ToUserName'] = toUserName
        self.__dict['FromUserName'] = fromUserName
        self.__dict['CreateTime'] = int(time.time())
        self.__dict['ArticleTitle'] = str(Title),
        self.__dict['ArticleDescription'] = str(Description),
        self.__dict['ArticlePicurl'] = str(Picurl),
        self.__dict['Url'] = Url

    def send(self):
        logger.info(self.__dict['ArticleTitle'])
        logger.info(self.__dict['ArticleDescription'])
        logger.info(self.__dict['ArticlePicurl'])
        logger.info(self.__dict['Url'])
        XmlForm = """
            <xml>
              <ToUserName><![CDATA[{ToUserName}]]></ToUserName>
              <FromUserName><![CDATA[{FromUserName}]]></FromUserName>
              <CreateTime>{CreateTime}</CreateTime>
              <MsgType><![CDATA[news]]></MsgType>
              <ArticleCount>1</ArticleCount>
              <Articles>
                <item>
                  <Title><![CDATA[{ArticleTitle[0]}]]></Title>
                  <Description><![CDATA[{ArticleDescription[0]}]]></Description>
                  <PicUrl><![CDATA[{ArticlePicurl[0]}]]></PicUrl>
                  <Url><![CDATA[{Url}]]></Url>
                </item>
              </Articles>
            </xml>

            """
        return XmlForm.format(**self.__dict)

二、使用：

引入receive.py和reply.py

from connectModule import  receive, reply

 调用方法

webData = request.body
recMsg = receive.parse_xml(webData)  #调用parsex_xml()方法格式化request参数
    if isinstance(recMsg, receive.Msg): # 判断recMsg是否和receive.py中的Msg对象相符
        toUser = recMsg.FromUserName
        fromUser = recMsg.ToUserName
        content = 'adsfasdfasdfadfsa'
        replyMsg = reply.TextMsg(toUser, fromUser, content)    #调用reply.TextMsg方法
        return replyMsg.send()    #返回send()