优化工单模块，增加企微模块

sg_wechat_enterprise/we_api/crypto/__init__.py

@@ -0,0 +1,120 @@
+# -*- coding: utf-8 -*-
+"""
+    wechatpy.crypto
+    ~~~~~~~~~~~~~~~~
+
+    This module provides some crypto tools for WeChat and WeChat enterprise
+
+    :copyright: (c) 2014 by messense.
+    :license: MIT, see LICENSE for more details.
+"""
+from __future__ import absolute_import, unicode_literals
+import time
+import base64
+
+from wechatpy.utils import to_text, to_binary, WeChatSigner
+from wechatpy.exceptions import (
+    InvalidAppIdException,
+    InvalidSignatureException
+)
+from wechatpy.crypto.base import BasePrpCrypto
+
+
+def _get_signature(token, timestamp, nonce, encrypt):
+    signer = WeChatSigner()
+    signer.add_data(token, timestamp, nonce, encrypt)
+    return signer.signature
+
+
+class PrpCrypto(BasePrpCrypto):
+
+    def encrypt(self, text, app_id):
+        return self._encrypt(text, app_id)
+
+    def decrypt(self, text, app_id):
+        return self._decrypt(text, app_id, InvalidAppIdException)
+
+
+class BaseWeChatCrypto(object):
+
+    def __init__(self, token, encoding_aes_key, _id):
+        encoding_aes_key = to_binary(encoding_aes_key + '=')
+        self.key = base64.b64decode(encoding_aes_key)
+        assert len(self.key) == 32
+        self.token = token
+        self._id = _id
+
+    def _check_signature(self,
+                         signature,
+                         timestamp,
+                         nonce,
+                         echo_str,
+                         crypto_class=None):
+        _signature = _get_signature(self.token, timestamp, nonce, echo_str)
+        if _signature != signature:
+            raise InvalidSignatureException()
+        pc = crypto_class(self.key)
+        return pc.decrypt(echo_str, self._id)
+
+    def _encrypt_message(self,
+                         msg,
+                         nonce,
+                         timestamp=None,
+                         crypto_class=None):
+        from wechatpy.replies import BaseReply
+
+        xml = """<xml>
+<Encrypt><![CDATA[{encrypt}]]></Encrypt>
+<MsgSignature><![CDATA[{signature}]]></MsgSignature>
+<TimeStamp>{timestamp}</TimeStamp>
+<Nonce><![CDATA[{nonce}]]></Nonce>
+</xml>"""
+        if isinstance(msg, BaseReply):
+            msg = msg.render()
+        timestamp = timestamp or to_binary(int(time.time()))
+        pc = crypto_class(self.key)
+        encrypt = to_text(pc.encrypt(msg, self._id))
+        signature = _get_signature(self.token, timestamp, nonce, encrypt)
+        return to_text(xml.format(
+            encrypt=encrypt,
+            signature=signature,
+            timestamp=timestamp,
+            nonce=nonce
+        ))
+
+    def _decrypt_message(self,
+                         msg,
+                         signature,
+                         timestamp,
+                         nonce,
+                         crypto_class=None):
+        if not isinstance(msg, dict):
+            import xmltodict
+
+            msg = xmltodict.parse(to_text(msg))['xml']
+
+        encrypt = msg['Encrypt']
+        _signature = _get_signature(self.token, timestamp, nonce, encrypt)
+        if _signature != signature:
+            raise InvalidSignatureException()
+        pc = crypto_class(self.key)
+        return pc.decrypt(encrypt, self._id)
+
+
+class WeChatCrypto(BaseWeChatCrypto):
+
+    def __init__(self, token, encoding_aes_key, app_id):
+        super(WeChatCrypto, self).__init__(token, encoding_aes_key, app_id)
+        self.app_id = app_id
+
+    def encrypt_message(self, msg, nonce, timestamp=None):
+        return self._encrypt_message(msg, nonce, timestamp, PrpCrypto)
+
+    def decrypt_message(self, msg, signature, timestamp, nonce):
+        return self._decrypt_message(
+            msg,
+            signature,
+            timestamp,
+            nonce,
+            PrpCrypto
+        )