util.py

#encoding:utf-8
import urllib2
import json
import requests
from datetime import datetime, timedelta

from django.conf import settings


APPID = settings.WX_APPID
APPSECRET = settings.WX_APPSECRET


class _AccessToken(str):
    def __init__(self):
        super(_AccessToken, self).__init__()
        self.last_refresh = None
        self._token = ''

    def __str__(self):
        return self.token

    __repr__ = __str__

    @property
    def token(self):
        if not self._token:
            self._refresh()

        elif datetime.now() - self.last_refresh > timedelta(seconds=7130):
            self._refresh()

        return self._token

    def _refresh(self):
        url = 'https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=%s&secret=%s' \
              % (APPID, APPSECRET)

        self.last_refresh = datetime.now()
        response_data = json.loads(urllib2.urlopen(url).read())

        try:
            self._token = response_data['access_token']
        except KeyError:
            self._token = ''


ACCESSON_TOKEN = _AccessToken()

menu_data = {
    "button": [
        {
            "name": "",
            "sub_button": [
                {
                    "type": "click",
                    "name": "",
                    "key": ""
                }
            ]
        },
        {
            "name": "",
            "sub_button": [
                {
                    "type": "click",
                    "name": "",
                    "key": "DEL",
                },
                {
                    "type": "click",
                    "name": "",
                    "key": "NORMAL",
                },
                {
                    "type": "click",
                    "name": "",
                    "key": "",
                }
            ]
        },
        {
            "name": "功能",
            "sub_button": [
                {
                    "type": "click",
                    "name": "",
                    "key": "",
                },
                {
                    "type": "click",
                    "name": "",
                    "key": "",
                },
                {
                    "type": "click",
                    "name": "",
                    "key": "",
                }
            ]
        }]
}


def update_menu():
    menu_data = data
    requests.post('https://api.weixin.qq.com/cgi-bin/menu/create?access_token=%s' % ACCESSON_TOKEN,
                  data=json.dumps(menu_data, ensure_ascii=False, encoding='utf-8'))