-
Notifications
You must be signed in to change notification settings - Fork 1
/
wechat.py
62 lines (52 loc) · 1.82 KB
/
wechat.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import json
import sys
import os
import simplejson
import urllib.request
import configparser
def gettoken(corpid,corpsecret):
gettoken_url = 'https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=' + corpid + '&corpsecret=' + corpsecret
try:
token_file = urllib.request.urlopen(gettoken_url)
except urllib.request.HTTPError as e:
print(e.code)
print(e.read().decode("utf8"))
sys.exit()
token_data = token_file.read().decode('utf-8')
print(token_data)
token_json = json.loads(token_data)
token_json.keys()
token = token_json['access_token']
return token
def senddata(touser, subject, content):
try:
config = configparser.RawConfigParser()
config_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'etc', 'config.ini')
config.read(config_file)
corpid = config['DEFAULT']['Corpid']
corpsecret = config['DEFAULT']['Corpsecret']
except:
print("======\n获取corp信息失败\n======")
return
access_token = gettoken(corpid,corpsecret)
send_url = "https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=" + access_token
send_values = {
"touser":touser,
#"toparty":"1",
"msgtype":"text",
"agentid":"1000003",
"text":{
"content":subject + '\n' + content
},
"safe":"0"
}
#send_data = json.dumps(send_values, ensure_ascii=False)
send_data = simplejson.dumps(send_values, ensure_ascii=False).encode('utf-8')
send_request = urllib.request.Request(send_url, send_data)
response = json.loads(urllib.request.urlopen(send_request).read())
print(str(response))
if __name__ == '__main__':
user = str(sys.argv[1])
subject = str(sys.argv[2])
content = str(sys.argv[3])
senddata(user, subject,content)