WechatRequest

Complexity

Total Complexity

7

Size/Duplication

Total Lines	31
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	6	1
A	refreshtoken()	0	4	2
A	get()	0	8	2
A	post()	0	6	2

import requests

from lib.wechatutil.gettoken import gettoken


class WechatRequest(object):

    defaultheader = {'content-type': 'application/json; charset=utf8'}

    def __init__(self, corpId, corpSecret):
        self.corpID = corpId
        self.corpSecret = corpSecret
        self.token = ''
        self.session = requests.session()
        self.refreshtoken()

    def get(self, url, customparams, customheader=None):
        header = self.defaultheader
        params = {"access_token": self.token}
        params.update(customparams)
        if customheader is not None:
            header.update(customheader)
        response = self.session.get(url, params=params, headers=header)
        return response.json()

    def post(self, url, data, customheader=None):
        header = self.defaultheader
        if customheader is not None:
            header.update(customheader)
        response = self.session.post(url + "?access_token=" + self.token, data=data, headers=header)
        return response.json()

    def refreshtoken(self):
        rc, self.token, errmsg = gettoken(self.corpID, self.corpSecret, self.session)
        if rc != 0:
            raise ValueError(errmsg)