Merge pull request #10 from halfcrazy/add-wechatpy

Add wechatpy

Author: picococoo

.travis.yml

@@ -8,7 +8,6 @@ services:
 
 env:
   - TOX_ENV=py27
-  - TOX_ENV=pypy
 
 install:
   - pip install flake8

requirements.txt

@@ -1,16 +1,28 @@
 backports-abc==0.4
 backports.ssl-match-hostname==3.5.0.1
 certifi==2016.2.28
+cffi==1.6.0
 click==6.6
+cryptography==1.3.2
+enum34==1.1.4
 flake8==2.5.4
+idna==2.1
+ipaddress==1.0.16
 mccabe==0.4.0
+optionaldict==0.1.1
 pep8==1.7.0
 py==1.4.31
+pyasn1==0.1.9
+pycparser==2.14
 pyflakes==1.0.0
 PyJWT==1.4.0
 pymongo==3.2.2
 pytest==2.9.1
 pytest-tornado==0.4.5
+python-dateutil==2.5.3
+requests==2.10.0
 singledispatch==3.4.0.3
 six==1.10.0
 tornado==4.3
+wechatpy==1.2.9
+xmltodict==0.10.1

setup.py

@@ -30,7 +30,6 @@
         'Programming Language :: Python :: 2',
         'Programming Language :: Python :: 2.7',
         'Programming Language :: Python :: Implementation :: CPython',
-        'Programming Language :: Python :: Implementation :: PyPy',
     ],
 
     packages=find_packages('.', exclude=('tests*')),

spirit/app.py

@@ -13,7 +13,7 @@ def __init__(self, config):
             cookie_secret=config.cookie_secret,
             template_path=config.template_path,
             static_path=config.static_path,
-            xsrf_cookies=True,
+            xsrf_cookies=config.xsrf_cookies,
         )
 
         self.db = getattr(

spirit/config/_base.py

@@ -17,6 +17,7 @@ def load_config(mode):
 class Config(object):
     def __init__(self):
         self.DEBUG = True
+        self.xsrf_cookies = True
         self.cookie_secret = base64.b64encode(
             uuid.uuid4().bytes + uuid.uuid4().bytes)
         self.template_path = os.path.abspath('../templates/')

spirit/config/development.py

@@ -3,7 +3,9 @@
 
 
 class DevelopmentConfig(Config):
-    pass
+    def __init__(self):
+        Config.__init__(self)
+        self.xsrf_cookies = False
 
 
 config = DevelopmentConfig()

spirit/controllers/wechat.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# import tornado.web
+from wechatpy import create_reply, parse_message
+from wechatpy.utils import check_signature
+from wechatpy.exceptions import InvalidSignatureException
+
+from ._base import BaseHandler
+
+
+class InterfaceHandler(BaseHandler):
+    def get(self):
+        signature = self.get_argument('signature')
+        timestamp = self.get_argument('timestamp')
+        nonce = self.get_argument('nonce')
+        echostr = self.get_argument('echostr')
+
+        token = 'spirit'
+
+        try:
+            check_signature(token, signature, timestamp, nonce)
+        except InvalidSignatureException:
+            # 处理异常情况或忽略
+            pass
+        self.write(echostr)
+
+    def post(self):
+        data = self.request.body
+        msg = parse_message(data)
+
+        if msg.type == 'text':
+            reply = create_reply(msg.content, msg)
+        else:
+            reply = create_reply('Sorry, can not handle this for now', msg)
+        self.write(reply.render())

spirit/routers.py

@@ -1,7 +1,8 @@
 # -*- coding: utf-8 -*-
-from .controllers import site
+from .controllers import site, wechat
 
 
 routers = [
-    (r"/", site.IndexHandler)
+    (r"/", site.IndexHandler),
+    (r"/wx_interface", wechat.InterfaceHandler)
 ]

tox.ini

@@ -4,7 +4,7 @@
 # and then run "tox" from this directory.
 
 [tox]
-envlist = py27, pypy
+envlist = py27
 
 [base]
 deps =