建议支持多号联动扩展 about flask-wechatpy HOT 1 OPEN

from flask.views import View, request
from flask import abort, current_app
from wechatpy import parse_message
from wechatpy.utils import check_signature
from wechatpy.exceptions import InvalidSignatureException, InvalidAppIdException
from wechatpy.crypto import WeChatCrypto
from wechatpy.replies import BaseReply


class WechatServerView(View):
    methods = ['GET', 'POST']

    @property
    def appid(self):
        raise NotImplemented

    @property
    def secret(self):
        raise NotImplemented

    @property
    def token(self):
        raise NotImplemented

    @property
    def aes_key(self):
        return None

    def dispatch_request(self):
        signature, timestamp, nonce = [request.args.get(k) for k in ('signature', 'timestamp', 'nonce')]

        # check signature
        try:
            check_signature(self.token, signature, timestamp, nonce)
        except InvalidSignatureException as ex:
            current_app.logger.warning('signature failed.')
            return abort(403)

        if request.method == 'GET':
            return request.args.get('echostr')

        raw_msg = request.data
        current_app.logger.debug(raw_msg)
        if self.aes_key:
            crypto = WeChatCrypto(self.token, self.aes_key, self.appid)
            try:
                raw_msg = crypto.decrypt_message(
                    raw_msg,
                    signature,
                    timestamp,
                    nonce
                )
            except (InvalidAppIdException, InvalidSignatureException):
                current_app.logger.warning('decode message failed.')
                return abort(403)

        wechat_msg = parse_message(raw_msg)
        res = self.process_wechat_msg(wechat_msg)
        xml = ''

        if isinstance(res, BaseReply):
            xml = res.render()

        if self.aes_key:
            crypto = WeChatCrypto(
                self.token,
                self.aes_key,
                self.appid
            )
            xml = crypto.encrypt_message(xml, nonce, timestamp)

        return xml

    def process_wechat_msg(self, msg):
        raise NotImplemented