def test_get_token(self):
wework = WeWork()
token = wework.get_contact_token()
assert token != None
token = wework.get_contact_token()
assert token != None
token = wework.get_contact_token()
assert token != None
def list(self, *id):
if (id.count == 0):
params = {'access_token': WeWork.get_accesstoken_contact()}
else:
params = {
'access_token': WeWork.get_accesstoken_contact(),
'id': id
}
self.printResponse(
requests.request("GET",
self.list_url,
params=params,
proxies=self.proxies,
verify=False).json())
return (requests.request("GET", self.list_url, params=params).json())
def send_text(self, msg=None, users=[], parties=[], tags=[]):
self.json_data = requests.post(
self._send_url,
params={
"access_token": WeWork.get_app_token()
},
# 需要设置UTF8编码
headers={
'content-type': 'application/json; charset=utf-8'
},
json={
"touser": "******".join(users),
"toparty": "|".join(parties),
"totag": "|".join(tags),
"msgtype": "text",
"agentid": WeWork.agent_id,
"text": {
"content": msg
},
"safe": 0,
"enable_id_trans": 0
},
proxies=self.proxies,
verify=False).json()
self.verbose(self.json_data)
def list(self, id):
resp = requests.get(self.list_url,
params={
"access_token": WeWork.get_token(),
"id": id
}).json()
return resp
def delete(self, id):
r = requests.get(self.delete_url,
params={
"access_token": WeWork.get_token(),
"id": id
}).json()
self.verbose(r)
return r
def request(self, method, url, params: dict, json, data):
#todo: 封装从外部读取api、或者api默认追加的一些参数
self.json_data = requests.request(
method,
url=url,
params=params.update({"access_token": WeWork.get_app_token()}))
self.verbose(self.json_data)
return self.json_data
def create(self, name, name_en, parentid, order):
data = {"name": name, "name_en": name_en, "parentid": parentid, "order": order,"access_token": WeWork.get_token()}
params = {"access_token": WeWork.get_token()}
# 上传中有中文，需要设置UTF-8编码
headers = {'content-type': 'application/json; charset=utf-8'}
self.do_request.add_headers(headers)
self.json_data = self.do_request.send(self.create_url, method="post", data=data, params=params).json()
self.verbose(self.json_data)
return self.json_data
def creat(self, name, parentid, order):
data = {"name": name, "parentid": parentid, "order": order, "id": None}
r = requests.post(self.create_url,
json=data,
params={
"access_token": WeWork.get_token()
}).json()
self.verbose(r)
return r
def delete(self, id):
self.json_data = requests.get(self.delete_url,
params={
"access_token":
WeWork.get_addr_token(),
"id": id
}).json()
self.verbose(self.json_data)
return self.json_data
def update(self, id, name, parentid, order):
data = {"id": id, "name": name, "parentid": parentid, "order": order}
self.json_data = requests.post(self.update_url,
json=data,
params={
"access_token":
WeWork.get_addr_token()
}).json()
self.verbose(self.json_data)
return self.json_data
def creat(self, name, parentid, order):
r = requests.post(self.creat_url,
params={
"access_token": WeWork.get_token()
},
json={
"name": name,
"parentid": parentid,
"order": order,
"id": ""
}).json()
return r
def create(self, name: str, parentid, order):
self.json_data = requests.post(
self.create_url,
params={"access_token": WeWork.get_contact_token()},
# 需要设置UTF8编码
headers={'content-type': 'application/json; charset=utf-8'},
json={
"name": name, "parentid": parentid, "order": order, "id": None
},
# proxies=proxies,
verify=False
).json()
self.verbose(self.json_data)
return self.json_data
def send_text(self, tousers,parties,tags):
send_json={
"touser" : "|".join(tousers),
"toparty" : "|".join(parties),
"totag" : "|".join(tags),
"msgtype" : "text",
"agentid" : WeWork.agent_id,
"text" : {
"content" : "你的快递已到，请携带工卡前往邮件中心领取。\n出发前可查看<a href=\"http://work.weixin.qq.com\">邮件中心视频实况</a>，聪明避开排队。"
},
"safe":0,
"enable_id_trans": 0
}
params = {'access_token': WeWork.get_accesstoken_app()}
#print(WeWork.get_accesstoken_app())
return requests.post(url=self._send_url,json=send_json,params=params).json()
def upload_long_source(self):
long_content = {
"Content-Disposition": "form-data",
"name": "fieldNameHere",
"filename": "20180103195745.png",
"Content-Type": "image/png",
"Content-Length": 220
}
self.json_data = requests.post(self._long_url,
params={
"access_token":
WeWork.get_app_token()
},
json=long_content,
data="file").json()
self.verbose(self.json_data)
def create(self, name, parentid, order, id):
proxy_addr = {
'http': 'http://127.0.0.1:7788',
'https': 'https://127.0.0.1:7788'
}
data = {"name": name, "parentid": parentid, "order": order, "id": id}
self.json_data = requests.post(
self.create_url,
params={
"access_token": WeWork.get_addr_token()
},
#设置一个UTF8编码
headers={
'content-type': 'application/json; charset=utf-8'
},
json=data,
#proxies=proxy_addr,
verify=False).json()
self.verbose(self.json_data)
return self.json_data
def send_text(self, msg=None, users=[], parties=[], tags=[]):
content = {
"touser": "******".join(users),
"toparty": "|".join(parties),
"totag": "|".join(tags),
"msgtype": "text",
"agentid": WeWork.agentid,
"text": {
"content": msg
},
"safe": 0,
"enable_id_trans": 0
}
self.json_data = requests.post(
self._send_url,
params={
"access_token": WeWork.get_app_token()
},
json=content,
# proxies= self.proxies,
verify=False).json()
self.verbose(self.json_data)
def upload_tmp_source(self, type):
tmp_content = {
"Content-Type": "multipart/form-data",
"boundary": "-------------------------acebdf13572468",
"Content-Length": 220,
"Content-Disposition": "form-data",
"name": "media",
"filename": "wework.txt",
"filelength": 6,
"Content-Type": "application/octet-stream"
}
# self.request(post, self._tmp_url,
# params={"access_token": WeWork.get_app_token()},
# json=content, data=type)
self.json_data = requests.post(self._tmp_url,
params={
"access_token":
WeWork.get_app_token()
},
json=tmp_content).json()
self.verbose(self.json_data)
def list(self, id):
self.json_data = requests.get(self.list_url, params={"access_token": WeWork.get_contact_token(), "id": id}).json()
self.verbose(self.json_data)
return self.json_data
def setup_class(cls):
cls.membertools = MemberTools()
# 传入secrect key，返回对应的token值
cls.token = WeWork.get_token(cls.membertools.secret)
def setup_class(cls):
cls.token = WeWork.get_token()
def list(self,id):
params = {'access_token': WeWork.get_token(), 'id': id}
self.json_data = self.do_request.send(self.list_url, method="get", params=params).json()
self.verbose(self.json_data)
return self.json_data
def test_get_tokne(self):
wework = WeWork()
r = wework.get_token('oKPwlYZ2bMunvnI4q6ZDr94YYZc_a4OxH2KM3oEh82k')
def test_get_token(self):
r = WeWork.get_access_token(WeWork.secret)
assert r["errcode"] == 0
def test_get_accesstoken(self):
wework = WeWork()
assert wework.get_accesstoken_contact() != None
def delete(self,id):
params = {"access_token": WeWork.get_token(), "id": id}
self.json_data = self.do_request.send(self.delete_url, method="get", params=params).json()
return self.json_data
def test_get_token(self):
wework = WeWork()
token = wework.get_access_token()
assert token != None
