升级WxPusher包接口数据

util/wxpusher/__init__.py

@@ -0,0 +1,18 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Init for WxPusher.
+
+File: __init__.py
+Author: huxuan
+Email: i(at)huxuan.org
+"""
+from .exceptions import WxPusherException
+from .exceptions import WxPusherNoneTokenException
+from .wxpusher import WxPusher
+
+__all__ = [
+    'WxPusher',
+    'WxPusherException',
+    'WxPusherNoneTokenException',
+]

util/wxpusher/exceptions.py

@@ -0,0 +1,17 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+WxPusher Exceptions.
+
+File: exceptions.py
+Author: huxuan
+Email: i(at)huxuan.org
+"""
+
+
+class WxPusherException(Exception):
+    """WxPusher specific base exception."""
+
+
+class WxPusherNoneTokenException(WxPusherException):
+    """Raised when both token and default token are None."""

util/wxpusher/wxpusher.py

@@ -0,0 +1,71 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+WxPusher Python SDK.
+
+File: wxpusher.py
+Author: huxuan
+Email: i(at)huxuan.org
+"""
+import requests
+
+from . import exceptions
+
+BASEURL = 'http://wxpusher.zjiecode.com/api'
+
+
+class WxPusher():
+    """WxPusher Python SDK."""
+
+    default_token = None
+
+    @classmethod
+    def send_message(cls, content, summary, **kwargs):
+        """Send Message."""
+        payload = {
+            'appToken': cls._get_token(kwargs.get('token')),
+            'summary': summary,
+            'content': content,
+            'contentType': kwargs.get('content_type', 1),
+            'topicIds': kwargs.get('topic_ids', []),
+            'uids': kwargs.get('uids', []),
+            'url': kwargs.get('url'),
+        }
+        url = f'{BASEURL}/send/message'
+        return requests.post(url, json=payload).json()
+
+    @classmethod
+    def query_message(cls, message_id):
+        """Query message status."""
+        url = f'{BASEURL}/send/query/{message_id}'
+        return requests.get(url).json()
+
+    @classmethod
+    def create_qrcode(cls, extra, valid_time=1800, token=None):
+        """Create qrcode with extra callback information."""
+        payload = {
+            'appToken': cls._get_token(token),
+            'extra': extra,
+            'validTime': valid_time,
+        }
+        url = f'{BASEURL}/fun/create/qrcode'
+        return requests.post(url, json=payload).json()
+
+    @classmethod
+    def query_user(cls, page, page_size, token=None):
+        """Query users."""
+        payload = {
+            'appToken': cls._get_token(token),
+            'page': page,
+            'pageSize': page_size,
+        }
+        url = f'{BASEURL}/fun/wxuser'
+        return requests.get(url, params=payload).json()
+
+    @classmethod
+    def _get_token(cls, token=None):
+        """Get token with validation."""
+        token = token or cls.default_token
+        if not token:
+            raise exceptions.WxPusherNoneTokenException()
+        return token