Python 模拟微信后台 第一弹

富友郑鹏展

好久没有在鱼C更新技术心得了，最近太懒了！ 还是下定决心把自己最近写的项目心得分享出来吧。由于最近公司需要写一个模拟微信后台与客户端来测试公众号在多并发下的性能情况。项目用的是flask来写的。我把它分成3部分，微信后台，微信客户端，当然还有进行测试的公众号端。当然这篇文章不会把它们一次性全部讲出来。所以我打算把它按功能来展开讲，主要也只是讲微信后台。当然我这个项目只是实现了微信的一部分功能而已，想把这个项目的心得写下来也是为了做一次整理。我也不会把整个项目讲出来，会挑里面几个功能来讲。整个项目写下来后，对微信的业务流程有了一定的理解，当然对flask也有了更深入的了解。用flask来写web其实还是蛮爽的。期待小甲鱼的教程。。。。
  好了，开始进入主题吧！那要从哪里开始讲呢？ 那就从客户端发送消息给公众号，公众号回复消息给客户端这个功能说起吧！
  其实无论是从客户端发送消息给公众号，还是公众号回复消息给客户端，都会经过微信后台，后台也会做相应的记录，怎么做呢？当然就需要数据库来记录了，这里我用的数据库是python自带的sqlite。为了让我们的程序更加优雅，更加有利于业务逻辑的编写，所以我采用了ORM框架sqlalchemy来作为数据库对象关系映射引擎，什么是ORM呢？简单点说其实就是用比较优雅的API来写数据库语句，它的底层还是SQL语句，其实就是加了一层语法糖。好了，接下来我需要编写数据库表结构，来存放消息数据。这里需要两个文件，一个database.py,一个models.py。models.py文件就是我们设计的表结构文件，database.py文件是数据库初始化文件。

  下图是models.py文件

1. from sqlalchemy import Column, Integer, String
   
2. from database import Base
   
3. # 消息表
   
4. class Msg(Base):
   
5.     __tablename__ = 'message'
   
6.     MsgId = Column(String(10000), primary_key=True)
   
7.     ToUserName = Column(String(10000), unique=False, nullable=False, index=True)
   
8.     FromUserName = Column(String(10000), unique=False, nullable=False, index=True)
   
9.     CreateTime = Column(Integer, unique=False, nullable=False, index=True)
   
10.     Content = Column(String(10000), unique=False, nullable=False, index=True)
    
11.     MsgType = Column(String(10000), unique=False, nullable=False, index=True)
    
12. 
    
13.     def __init__(self, ToUserName=None, FromUserName=None, CreateTime=None, Content=None, MsgId=None, MsgType=None):
    
14.         self.ToUserName = ToUserName
    
15.         self.FromUserName = FromUserName
    
16.         self.CreateTime = CreateTime
    
17.         self.Content = Content
    
18.         self.MsgId = MsgId
    
19.         self.MsgType = MsgType
    
20. 
    
21.     def __repr__(self):
    
22.         return "{'ToUserName': %s, 'FromUserName': %s, 'CreateTime': %s, 'Content': %s, 'MsgId': %s, 'MsgType': %s}" % (self.ToUserName, self.FromUserName, self.CreateTime, self.Content, self.MsgId, self.MsgType)

复制代码

  下图是database.py文件

1. # coding=utf-8
   
2. 
   
3. from sqlalchemy import create_engine
   
4. from sqlalchemy.orm import scoped_session, sessionmaker
   
5. from sqlalchemy.ext.declarative import declarative_base
   
6. 
   
7. engine = create_engine(r'sqlite:///test.db', convert_unicode=True)
   
8. db_session = scoped_session(sessionmaker(autocommit=False,
   
9.                                          autoflush=True,
   
10.                                          bind=engine))
    
11. Base = declarative_base()
    
12. Base.query = db_session.query_property()
    
13. 
    
14. def init_db():
    
15.     # 在这里导入定义模型所需要的所有模块，这样它们就会正确的注册在
    
16.     # 元数据上。否则你就必须在调用 init_db() 之前导入它们。
    
17.     import models
    
18.     Base.metadata.create_all(bind=engine)

复制代码

  有了数据库的表结构，下面我们就可以写业务逻辑了。先来看后台的代码

1. # coding=utf-8
   
2. 
   
3. import time
   
4. import requests
   
5. import random
   
6. import hashlib
   
7. import re
   
8. from lxml import etree
   
9. from flask import Flask, request, jsonify, send_file, send_from_directory, render_template
   
10. from database import init_db, db_session
    
11. from models import *
    
12. from sqlalchemy import and_
    
13. from datetime import datetime
    
14. from refresh import Reflesh_DataBase
    
15. import urlparse
    
16. import ConfigParser
    
17. import qrcode
    
18. import uuid
    
19. import os
    
20. import sys
    
21. import json
    
22. from gevent.wsgi import WSGIServer
    
23. from gevent import monkey
    
24. monkey.patch_all()
    
25. reload(sys)
    
26. sys.setdefaultencoding('utf-8')
    
27. 
    
28. app = Flask(__name__)
    
29. work_path = os.getcwd()
    
30. f = ConfigParser.ConfigParser()
    
31. f.readfp(open(work_path + '\\TestModle.ini'))
    
32. Mode = f.get('Default', 'OAuthTestMode')
    
33. ip = socket.gethostbyname(socket.gethostname())
    
34. # 微信后台类
    
35. class WechatServer:
    
36. 
    
37.     # 初始化
    
38.     def __init__(self, url='http://localhost:90'):
    
39.         self.url = url
    
40.     # 发送相应请求到指定服务器
    
41.     def send_data(self, data=None, method=None):
    
42.         if method == "GET":
    
43.             return requests.get(self.url, params=data)
    
44.         elif method == "POST":
    
45.             return requests.post(self.url, data=data)
    
46.         else:
    
47.             raise Exception(u'Error: the method must be given')
    
48. 
    
49. # 接收客户端的文本消息，再将消息推送到公众号
    
50. @app.route('/receive/message', methods=['POST'])
    
51. def receive_message():
    
52.     # 接收POST请求的表单数据
    
53.     data = request.form
    
54.     if not data:
    
55.         return jsonify({"errcode": 43002, "errmsg": "require POST request"})
    
56.     ToUserName = data.get('ToUserName', '')
    
57.     FromUserName = data.get('FromUserName', '')
    
58.     CreateTime = int(data.get('CreateTime', ''))
    
59.     Content = data.get('Content', '')
    
60.     MsgId = data.get('MsgId', '')
    
61.     MsgType = data.get('MsgType', '')
    
62.     m = Msg(ToUserName=ToUserName, FromUserName=FromUserName, CreateTime=CreateTime, Content=Content, MsgId=MsgId, MsgType=MsgType)
    
63.     # 把数据插入到消息表中
    
64.     init_db()
    
65.     db_session.add(m)
    
66.     db_session.commit()
    
67.     # 获取公众号url
    
68.     url = db_session.query(Public_Num).filter_by(appnum=ToUserName).first().appurl
    
69.     xml = '<xml>\n<ToUserName><![CDATA[{a}]]></ToUserName>\n<FromUserName><![CDATA[{b}]]></FromUserName> \n<CreateTime>{c}</CreateTime>\n<Content><![CDATA[{d}]]></Content>\n<MsgId><![CDATA[{e}]]></MsgId>\n<MsgType>{f}</MsgType>\n</xml>'.format(a=ToUserName, b=FromUserName, c=CreateTime, d=Content, e=MsgId, f=MsgType)
    
70.    # 向公众号推送XML数据
    
71.     WechatServer(url=url).send_data(data=xml, method='POST')
    
72.     return jsonify({'success': True})
    
73. 
    
74. # 接收公众号服务器发来的xml数据并把数据发送到相应的用户
    
75. @app.route('/send/message', methods=['POST', 'GET'])
    
76. def send_message():
    
77.     ToUserName, FromUserName, CreateTime, Content, MsgType = '', '', '', '', ''
    
78.     if request.method == 'POST':
    
79.         data = request.form
    
80.         if not data:
    
81.             return jsonify({"errcode": 43002, "errmsg": "require POST request"})
    
82.         ToUserName = data.get('ToUserName', '')
    
83.         FromUserName = data.get('FromUserName', '')
    
84.         CreateTime = int(data.get('CreateTime', ''))
    
85.         Content = data.get('Content', '')
    
86.         MsgType = data.get('MsgType', '')
    
87.     elif request.method == 'GET':
    
88.         qs = request.query_string
    
89.         if not qs:
    
90.             return jsonify({"errcode": 43001, "errmsg": "require GET request"})
    
91.         data = urlparse.parse_qs(qs)
    
92.         ToUserName = data.get('ToUserName', '')[0]
    
93.         FromUserName = data.get('FromUserName', '')[0]
    
94.         CreateTime = int(data.get('CreateTime', '')[0])
    
95.         Content = data.get('Content', '')[0]
    
96.         MsgType = data.get('MsgType', '')[0]
    
97.     m = Msg(ToUserName=ToUserName, FromUserName=FromUserName, CreateTime=CreateTime, Content=Content, MsgType=MsgType)
    
98.     # 把数据插入到消息表中
    
99.     init_db()
    
100.     db_session.add(m)
     
101.     db_session.commit()
     
102.    # 把公众号回复的消息发送给客户端
     
103.     WechatServer(url='http://localhost:93/receivepublicinfo').send_data(data={'Content': Content, 'CreateTime': CreateTime, 'ToUserName': ToUserName, 'FromUserName': FromUserName, 'MsgType': MsgType}, method='POST')
     
104.     return jsonify({'success': True})
     
105. if __name__ == '__main__':
     
106.     http_server = WSGIServer((ip, 90), app)
     
107.     http_server.serve_forever()

复制代码

  好了，今天先讲这些吧。如果有什么错误，请指出来！谢谢。有什么问题也可以提出来我们一起探讨！
  
  

PatrickTse

谢谢大神分享！