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

sg_wechat_enterprise/we_api/utils.py

@@ -0,0 +1,149 @@
+# -*- coding: utf-8 -*-
+"""
+    wechatpy.utils
+    ~~~~~~~~~~~~~~~
+
+    This module provides some useful utilities.
+
+    :copyright: (c) 2014 by messense.
+    :license: MIT, see LICENSE for more details.
+"""
+from __future__ import absolute_import, unicode_literals
+import six
+import six.moves.urllib.parse as urlparse
+import sys
+import string
+import random
+import hashlib
+
+try:
+    '''Use simplejson if we can, fallback to json otherwise.'''
+    import simplejson as json
+except ImportError:
+    import json  # NOQA
+
+
+class ObjectDict(dict):
+    """Makes a dictionary behave like an object, with attribute-style access.
+    """
+
+    def __getattr__(self, key):
+        if key in self:
+            return self[key]
+        return None
+
+    def __setattr__(self, key, value):
+        self[key] = value
+
+
+class WeChatSigner(object):
+    """WeChat data signer"""
+
+    def __init__(self, delimiter=b''):
+        self._data = []
+        self._delimiter = to_binary(delimiter)
+
+    def add_data(self, *args):
+        """Add data to signer"""
+        for data in args:
+            self._data.append(to_binary(data))
+
+    @property
+    def signature(self):
+        """Get data signature"""
+        self._data.sort()
+        str_to_sign = self._delimiter.join(self._data)
+        return hashlib.sha1(str_to_sign).hexdigest()
+
+
+def check_signature(token, signature, timestamp, nonce):
+    """Check WeChat callback signature, raises InvalidSignatureException
+    if check failed.
+
+    :param token: WeChat callback token
+    :param signature: WeChat callback signature sent by WeChat server
+    :param timestamp: WeChat callback timestamp sent by WeChat server
+    :param nonce: WeChat callback nonce sent by WeChat sever
+    """
+    signer = WeChatSigner()
+    signer.add_data(token, timestamp, nonce)
+    if signer.signature != signature:
+        from wechatpy.exceptions import InvalidSignatureException
+
+        raise InvalidSignatureException()
+
+
+def to_text(value, encoding='utf-8'):
+    """Convert value to unicode, default encoding is utf-8
+
+    :param value: Value to be converted
+    :param encoding: Desired encoding
+    """
+    if not value:
+        return ''
+    if isinstance(value, six.text_type):
+        return value
+    if isinstance(value, six.binary_type):
+        return value.decode(encoding)
+    return six.text_type(value)
+
+
+def to_binary(value, encoding='utf-8'):
+    """Convert value to binary string, default encoding is utf-8
+
+    :param value: Value to be converted
+    :param encoding: Desired encoding
+    """
+    if not value:
+        return b''
+    if isinstance(value, six.binary_type):
+        return value
+    if isinstance(value, six.text_type):
+        return value.encode(encoding)
+    return six.binary_type(value)
+
+
+def timezone(zone):
+    """Try to get timezone using pytz or python-dateutil
+
+    :param zone: timezone str
+    :return: timezone tzinfo or None
+    """
+    try:
+        import pytz
+        return pytz.timezone(zone)
+    except ImportError:
+        pass
+    try:
+        from dateutil.tz import gettz
+        return gettz(zone)
+    except ImportError:
+        return None
+
+
+def random_string(length=16):
+    rule = string.ascii_letters + string.digits
+    rand_list = random.sample(rule, length)
+    return ''.join(rand_list)
+
+
+def get_querystring(uri):
+    """Get Qeruystring information from uri.
+
+    :param uri: uri
+    :return: querystring info or {}
+    """
+    parts = urlparse.urlsplit(uri)
+    if sys.version_info[:2] == (2, 6):
+        query = parts.path
+        if query.startswith('?'):
+            query = query[1:]
+    else:
+        query = parts.query
+    return urlparse.parse_qs(query)
+
+
+def byte2int(c):
+    if six.PY2:
+        return ord(c)
+    return c