# -*- coding: utf-8 -*-
from WXApi import *
from flask import Flask, Response, request
app = Flask(__name__)

BASE_URL = '/wechatrequest'
BASE_TOKEN = 'wechatrequest'


def service_route(ask):
    # type: (WeChatObject) -> ReplyObject
    if isinstance(ask, ImageMsg):
        return ImageReply(ask, ask.media_id)

    if isinstance(ask, VoiceMsg):
        ques = "{0}".format(ask.recognition)
        return VoiceReply(ask, ask.media_id)

    if isinstance(ask, TextMsg):
        msg = ask.content
        return TextMsg(ask)

    if isinstance(ask, LinkMsg):
        return NewsReply(ask, title=ask.title, desc=ask.description, 
        picture_url='',link=ask.link)

    if isinstance(ask, SubEvent):
        return TextReply(ask, "欢迎关注")
    
    if isinstance(ask, UnSubEvent):
        return EmptyReply(ask)

    return TextReply(ask, "我不懂你在说什么才是最大的痛吧 😪")


# 用于处理微信的数据请求
@app.route(BASE_URL, methods=['GET', 'POST'])
def app_main():
    query = url_decode(request.query_string)
    # 认证是否是微信的合法请求
    if auth_signature(BASE_TOKEN, query):
        if 'echostr' in query:
            return Response(query.get('echostr'), 200)
    else:
        return Response(status=401)
    if 'POST' != request.method:
        return Response(status=400)
    # 认证成功,处理POST数据
    try:
        json_req = xml2event(request.data)
        res_data = service_route(json_req).create_xml() + ""
    except Exception as e:
        return Response(status=200)
    else:
        return Response(res_data, 200)