Add some WeChat JS SDK related APIs

tests/fixtures/ticket_getticket.json

@@ -0,0 +1,6 @@
+{
+    "errcode":0,
+    "errmsg":"ok",
+    "ticket":"bxLdikRXVbTPdHSM05e5u5sUoXNKd8-41ZO3MhKoyN5OfkWITDGgnr2fwJ0m9E8NYzWKVZvdVtaUgWvsdshFKA",
+    "expires_in":7200
+}

tests/test_client.py

@@ -399,3 +399,28 @@ def test_datacube_get_upstream_msg_dist_month(self):
                 '2014-12-07'
             )
             self.assertEqual(1, len(result))
+
+    def test_jsapi_get_ticket(self):
+        with HTTMock(wechat_api_mock):
+            result = self.client.jsapi.get_ticket()
+            self.assertEqual(
+                'bxLdikRXVbTPdHSM05e5u5sUoXNKd8-41ZO3MhKoyN5OfkWITDGgnr2fwJ0m9E8NYzWKVZvdVtaUgWvsdshFKA',  # NOQA
+                result['ticket']
+            )
+            self.assertEqual(7200, result['expires_in'])
+
+    def test_jsapi_get_jsapi_signature(self):
+        noncestr = 'Wm3WZYTPz0wzccnW'
+        ticket = 'sM4AOVdWfPE4DxkXGEs8VMCPGGVi4C3VM0P37wVUCFvkVAy_90u5h9nbSlYy3-Sl-HhTdfl2fzFy1AOcHKP7qg'  # NOQA
+        timestamp = 1414587457
+        url = 'http://mp.weixin.qq.com?params=value'
+        signature = self.client.jsapi.get_jsapi_signature(
+            noncestr,
+            ticket,
+            timestamp,
+            url
+        )
+        self.assertEqual(
+            '0f9de62fce790f9a083d5c99e95740ceb90c27ed',
+            signature
+        )

wechatpy/client/__init__.py

@@ -34,6 +34,7 @@ def __init__(self, appid, secret, access_token=None):
         self.merchant = api.WeChatMerchant(weak_self)
         self.customservice = api.WeChatCustomService(weak_self)
         self.datacube = api.WeChatDataCube(weak_self)
+        self.jsapi = api.WeChatJSAPI(weak_self)
 
     def fetch_access_token(self):
         """

wechatpy/client/api/__init__.py

@@ -12,3 +12,4 @@
 from .merchant import WeChatMerchant  # NOQA
 from .customservice import WeChatCustomService  # NOQA
 from .datacube import WeChatDataCube  # NOQA
+from .jsapi import WeChatJSAPI  # NOQA

wechatpy/client/api/jsapi.py

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+"""
+    wechatpy.client.js
+    ~~~~~~~~~~~~~~~~~~~~
+
+    This module provides some APIs for JS SDK
+
+    :copyright: (c) 2014 by messense.
+    :license: MIT, see LICENSE for more details.
+"""
+from __future__ import absolute_import, unicode_literals
+
+from wechatpy.utils import WeChatSigner
+from .base import BaseWeChatAPI
+
+
+class WeChatJSAPI(BaseWeChatAPI):
+
+    def get_ticket(self, type='jsapi'):
+        """
+        获取微信 JS-SDK ticket
+
+        :return: 返回的 JSON 数据包
+        """
+        return self._get(
+            'ticket/getticket',
+            params={'type': type}
+        )
+
+    def get_jsapi_signature(self, noncestr, ticket, timestamp, url):
+        data = [
+            'noncestr={noncestr}'.format(noncestr=noncestr),
+            'jsapi_ticket={ticket}'.format(ticket=ticket),
+            'timestamp={timestamp}'.format(timestamp=timestamp),
+            'url={url}'.format(url=url),
+        ]
+        signer = WeChatSigner(delimiter=b'&')
+        signer.add_data(*data)
+        return signer.signature

wechatpy/utils.py

@@ -38,8 +38,9 @@ def __setitem__(self, key, value, dict_setitem=dict.__setitem__):
 class WeChatSigner(object):
     """WeChat data signer"""
 
-    def __init__(self):
+    def __init__(self, delimiter=b''):
         self._data = []
+        self._delimiter = to_binary(delimiter)
 
     def add_data(self, *args):
         """Add data to signer"""
@@ -50,7 +51,7 @@ def add_data(self, *args):
     def signature(self):
         """Get data signature"""
         self._data.sort()
-        str_to_sign = b''.join(self._data)
+        str_to_sign = self._delimiter.join(self._data)
         return hashlib.sha1(str_to_sign).hexdigest()