gui.py

#  -*-  coding: utf-8  -*-

import time
import random
import hashlib
import xml.etree.cElementTree as ET

import tkinter as tk
from tkinter import scrolledtext as st

import requests


settings = {
    # `ToUserName` & `FromUserName` will be placed in the XML data posted to
    # the given URL.
    "ToUserName": "gh_bea8cf2a04fd",
    "FromUserName": "oLXjgjiWeAS1gfe4ECchYewwoyTc",

    # URL of your Wexin handler.
    "url": "http://localhost:8080/weixin",
    # Path to client certificate
    # Put certificate and secret key into a single PEM file
    "cert": "",

    # These will be displayed in GUI.
    "mp_display_name": "APP",
    "me_display_name": "ME",

    # The token you submitted to Weixin MP. Used to generate signature.
    "token": "B0e8alq5ZmMjcnG5gwwLRPW2",
}


TPL_TEXT = '''
<xml>
    <ToUserName><![CDATA[%(to)s]]></ToUserName>
    <FromUserName><![CDATA[%(from)s]]></FromUserName>
    <CreateTime>%(time)d</CreateTime>
    <MsgType><![CDATA[text]]></MsgType>
    <Content><![CDATA[%(content)s]]></Content>
    <MsgId>%(id)s</MsgId>
</xml>
'''

TPL_EVENT = '''
<xml>
    <ToUserName><![CDATA[%(to)s]]></ToUserName>
    <FromUserName><![CDATA[%(from)s]]></FromUserName>
    <CreateTime>%(time)d</CreateTime>
    <MsgType><![CDATA[event]]></MsgType>
    <Event><![CDATA[%(event)s]]></Event>
    <EventKey><![CDATA[%(key)s]]></EventKey>
</xml>'''


def post(qs, data):
    headers = {'Content-Type': 'text/xml'}
    r = requests.post(settings["url"] + qs, data=data.encode(), cert=settings["cert"], verify=False, headers=headers)
    return r.content


def send():
    s = e.get()

    if not s:
        return

    # Simulation for clicking the menu.
    # Usage: EVENT_TYPE@EVENT_KEY
    #        'c@KEY_FIND'   - 'CLICK' event with 'KEY_FIND' as event key
    #        'v@www.qq.com' - 'VIEW' event with 'www.qq.com' as event key
    if s.startswith("c@") or s.startswith("v@"):
        event, key = s.split("@")
        msg = {
            "to": settings["ToUserName"],
            "from": settings["FromUserName"],
            "time": time.time(),
            "event": "CLICK" if event == "c" else "VIEW",
            "key": key
        }
        qs = "?signature=%s&timestamp=%s&nonce=%s" % \
            mix(int(msg["time"]))
        receive(msg["time"], post(qs, TPL_EVENT % msg))
    # Simulation for sending a message.
    else:
        t.insert(tk.END, settings["me_display_name"]+"\n", "send_name")
        t.insert(tk.END, s+"\n", "send_content")

        msg = {
            "to": settings["ToUserName"],
            "from": settings["FromUserName"],
            "time": time.time(),
            "content": s,
            "id": str(random.random())[-10:]
        }

        qs = "?signature=%s&timestamp=%s&nonce=%s" % \
            mix(int(msg["time"]))
        receive(msg["time"], post(qs, TPL_TEXT % msg))


def receive(start, response):
    if time.time() - start > 4.95:
        return

    if not response:
        print("No response.")
        return

    print("Received:\n%s\n" % response.decode())
    try:
        et = ET.fromstring(response)
    except ET.ParseError:
        print("Bad XML.")
        return

    to = et.find("ToUserName").text
    fr = et.find("FromUserName").text

    type = et.find("MsgType").text
    if type == "text":
        c = et.find("Content").text
    elif type == "news":
        l = ["[news]"]
        for i in et.find("Articles").findall("item"):
            l.append("Title={0}".format(i.find("Title").text))
            l.append("Description={0}".format(i.find("Description").text))
            l.append("PicUrl={0}".format(i.find("PicUrl").text))
            l.append("Url={0}".format(i.find("Url").text))
            l.append("---")
        c = "\n".join(l)
    elif type == "image":
        c = "[image]\nMediaId={0}\n---".format(et.find("Image").find("MediaId").text)
    else:
        print("Unknown response.")
        return

    t.insert(tk.END, settings["mp_display_name"]+"\n", "receive_name")
    t.insert(tk.END, c+"\n", "receive_content")
    t.yview_moveto(1.0)


def mix(time):
    timestamp = str(time)

    # I don't know how Weixin generate the 9-digit nonce, so I turn to random.
    nonce = str(int(random.random()))[-9:]

    l = [timestamp, nonce, settings["token"]]
    l.sort()
    signature = hashlib.sha1("".join(l).encode()).hexdigest()

    return signature, timestamp, nonce


def follow():
    msg = {
        "to": settings["ToUserName"],
        "from": settings["FromUserName"],
        "time": time.time(),
        "event": "subscribe",
        "key": ""
    }
    qs = "?signature=%s&timestamp=%s&nonce=%s" % \
        mix(int(msg["time"]))
    receive(msg["time"], post(qs, TPL_EVENT % msg))


def unfollow():
    msg = {
        "to": settings["ToUserName"],
        "from": settings["FromUserName"],
        "time": time.time(),
        "event": "unsubscribe",
        "key": ""       # `EventKey` in `unsubscribe` event is empty.
    }
    qs = "?signature=%s&timestamp=%s&nonce=%s" % \
        mix(int(msg["time"]))
    receive(msg["time"], post(qs, TPL_EVENT % msg))


top = tk.Tk()
top.title("微信模拟器")

t = st.ScrolledText(top, width=45, height=30)
t.pack()

t.tag_add("send_name", "1.0", "1.end")
t.tag_config("send_name", font=("Arial", "10", "bold"), spacing3=5, rmargin=6,
             justify=tk.RIGHT)
t.tag_add("send_content", "2.0", "2.end")
t.tag_config("send_content", spacing3=2, justify=tk.RIGHT, rmargin=6)

t.tag_add("receive_name", "1.0", "1.end")
t.tag_config("receive_name", font=("Arial", "10", "bold"), spacing3=5, lmargin1=2)
t.tag_add("receive_content", "2.0", "2.end")
t.tag_config("receive_content", lmargin1=2)

e = tk.Entry(top)
e.pack(side=tk.LEFT)

b = tk.Button(top, text="发送", command=send)
b.pack(side=tk.LEFT)

a = tk.Button(top, text="关注公众帐号", command=follow)
a.pack(side=tk.RIGHT)

a = tk.Button(top, text="取消关注", command=unfollow)
a.pack(side=tk.RIGHT)

if __name__ == "__main__":
    top.mainloop()