Niko_Public/backend/wechat.py

# 微信公众号相关接口
import urllib.parse
import requests
import json
import time
import pytz

from backend.settings import WECHAT_ID_SECRET
from backend.models import WeChatAccessToken, JsapiAccessToken
from datetime import datetime, timedelta
from backend.functions import HttpRequest, get_nonce_str, SHA1, service_log


# wechat access token management
class AccessTokenManagement:

    def __init__(self, application_name):
        self.application = application_name     # application_name用于存在本地数据库，和WX客户端没有关系

    def request_new_token(self):
        url = 'https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=' + WECHAT_ID_SECRET['appID']\
              + '&secret=' + WECHAT_ID_SECRET['AppSecret']
        res = urllib.request.urlopen(url)
        values = eval(res.read())
        wechat_token = WeChatAccessToken.search_by_application(self.application)
        if wechat_token is None:
            wechat_token = WeChatAccessToken()
        wechat_token.application = self.application
        try:
            wechat_token.accessTokenKey = values['access_token']
            wechat_token.save()
        except Exception as e:
            print(e)
            service_log("微信获取access token错误")
            service_log(values['errmsg'])

    def get_token(self):
        wechat_token = WeChatAccessToken.search_by_application(self.application)
        if wechat_token is None:
            self.request_new_token()
            wechat_token = WeChatAccessToken.search_by_application(self.application)
        utc = pytz.UTC
        if datetime.utcnow().replace(tzinfo=utc) > (wechat_token.requestTime + timedelta(seconds=7200 - 300)).replace(tzinfo=utc):
            self.request_new_token()
            wechat_token = WeChatAccessToken.search_by_application(self.application)
        return wechat_token.accessTokenKey


# wechat jsapi token management
# 参考文档 https://mp.weixin.qq.com/wiki?action=doc&id=mp1421141115&t=0.9173760134966377#62
class JsapiTokenManagement:

    def __init__(self, application_name):
        self.application = application_name     # application_name用于存在本地数据库，和WX客户端没有关系

    def request_new_token(self):
        atm = AccessTokenManagement()
        access_token = atm.get_token()
        url = 'https://api.weixin.qq.com/cgi-bin/ticket/getticket?access_token=' + access_token + '&type=jsapi'
        res = urllib.request.urlopen(url)
        values = eval(res.read())
        jsapi_token = JsapiAccessToken.search_by_application(self.application)
        if jsapi_token is None:
            jsapi_token = JsapiAccessToken()
        jsapi_token.application = self.application
        jsapi_token.accessTokenKey = values['ticket']
        jsapi_token.save()

    def get_token(self):
        jsapi_token = JsapiAccessToken.search_by_application(self.application)
        if jsapi_token is None:
            self.request_new_token()
            jsapi_token = JsapiAccessToken.search_by_application(self.application)
        if datetime.now() > jsapi_token.requestTime + timedelta(seconds=7200 - 300):
            self.request_new_token()
            jsapi_token = JsapiAccessToken.search_by_application(self.application)
        return jsapi_token.accessTokenKey


# 微信微信网页授权
class WeChatOA2:
    def __init__(self):
        self.base_url = 'https://open.weixin.qq.com/connect/oauth2/authorize?'

    # state只接受a-zA-Z0-9
    def snsapi_base(self, state):
        final_url = self.base_url + 'appid=' + WECHAT_ID_SECRET['appID']
        final_url = final_url + '&redirect_uri=' + urllib.parse.quote(WECHAT_ID_SECRET['redirect_uri'])
        final_url = final_url + '&response_type=code'
        final_url = final_url + '&scope=snsapi_base'
        final_url = final_url + '&state=' + state
        final_url = final_url + '#wechat_redirect'
        print(final_url)
        return final_url

    def snsapi_userinfo(self, state):
        final_url = self.base_url + 'appid=' + WECHAT_ID_SECRET['appID']
        final_url = final_url + '&redirect_uri=' + urllib.parse.quote(WECHAT_ID_SECRET['redirect_uri'])
        final_url = final_url + '&response_type=code'
        final_url = final_url + '&scope=snsapi_userinfo'
        final_url = final_url + '&state=' + state
        final_url = final_url + '#wechat_redirect'
        print(final_url)
        return final_url


# 微信模板消息发送
# 开发参考页面 https://mp.weixin.qq.com/wiki?t=resource/res_main&id=mp1433751277
class WeChatMsg:
    def __init__(self, openid, template_id, application_name, url=None, miniprogram=None, pagepath=None):
        self.POST_content = {
            'touser': openid,
            'template_id': template_id,
            'url': url,  # 模板跳转链接
            'miniprogram': miniprogram,
            'appid': WECHAT_ID_SECRET['appID'],
            'pagepath': pagepath,
        }
        self.application_name = application_name
        self.first = {}
        self.remark = {}
        self.keywords = {}

    def set_first(self, value, color='#000000'):
        self.first = {'first': {'value': value, 'color': color}}

    def set_remark(self, value, color='#000000'):
        self.remark = {'remark': {'value': value, 'color': color}}

    def add_keyboard(self, value, color='#000000'):
        l = len(self.keywords)
        new_keyboard = {'keyword{}'.format(l + 1): {'value': value, 'color': color}}
        # 从字典中取值 https://blog.csdn.net/weixin_43069755/article/details/85005682
        self.keywords = {**self.keywords, **new_keyboard}
        return self.keywords

    def submit(self):
        data = {'data': {**self.first, **self.keywords, **self.remark}}
        self.POST_content = {**self.POST_content, **data}
        POST_data = json.dumps(self.POST_content)
        atm = AccessTokenManagement(application_name=self.application_name)
        access_token = atm.get_token()
        url = 'https://api.weixin.qq.com/cgi-bin/message/template/send?access_token={}'.format(access_token)
        resp = requests.post(url, data=POST_data.encode('utf-8'), headers={'Content-Type': 'text/xml'})
        msg = resp.text.encode('ISO-8859-1').decode('utf-8')
        return msg


# JSAPI 调用
# https://mp.weixin.qq.com/wiki?action=doc&id=mp1421141115&t=0.9173760134966377#62
class JSAPI:
    def __init__(self, url):
        self.data = {'timestamp': str(int(time.time())), 'noncestr': get_nonce_str(), 'url': url}

    def get_sign(self):
        # 计算签名
        jtm = JsapiTokenManagement()
        self.data['jsapi_ticket'] = jtm.get_token()
        keys, paras = sorted(self.data), []
        paras = ['{}={}'.format(key, self.data[key]) for key in keys]  # and kwargs[key] != '']
        stringA = '&'.join(paras)

        stringSignTemp = stringA  # + '&key=' + WECHAT_ID_SECRET['wx_mch_api_key']
        sign = SHA1(stringSignTemp)
        self.data['signature'] = sign

    def get_parameter(self):
        self.get_sign()
        self.data['appId'] = WECHAT_ID_SECRET['appID']
        return self.data