-
Notifications
You must be signed in to change notification settings - Fork 4
/
runapi.py
136 lines (130 loc) · 6.2 KB
/
runapi.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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
# -*- coding: UTF-8 -*-
import os
import requests as req
import json,sys,time,random
global access_token_list
app_num=os.getenv('APP_NUM')
if app_num == '':
app_num = '1'
access_token_list=['wangziyingwen']*int(app_num)
#配置选项，自由选择
config_list = {'每次轮数':3,
'是否启动随机时间':'Y','延时范围起始':100,'结束':400,
'是否开启随机api顺序':'Y',
'是否开启各api延时':'Y','api延时范围开始':2,'api延时结束':5,
'是否开启各账号延时':'Y','账号延时范围开始':60,'账号延时结束':120,
}
#'是否开启备用应用':'N','是否开启测试':'N'
api_list = [r'https://graph.microsoft.com/v1.0/me/',
r'https://graph.microsoft.com/v1.0/users',
r'https://graph.microsoft.com/v1.0/me/people',
r'https://graph.microsoft.com/v1.0/groups',
r'https://graph.microsoft.com/v1.0/me/contacts',
r'https://graph.microsoft.com/v1.0/me/drive/root',
r'https://graph.microsoft.com/v1.0/me/drive/root/children',
r'https://graph.microsoft.com/v1.0/drive/root',
r'https://graph.microsoft.com/v1.0/me/drive',
r'https://graph.microsoft.com/v1.0/me/drive/recent',
r'https://graph.microsoft.com/v1.0/me/drive/sharedWithMe',
r'https://graph.microsoft.com/v1.0/me/calendars',
r'https://graph.microsoft.com/v1.0/me/events',
r'https://graph.microsoft.com/v1.0/sites/root',
r'https://graph.microsoft.com/v1.0/sites/root/sites',
r'https://graph.microsoft.com/v1.0/sites/root/drives',
r'https://graph.microsoft.com/v1.0/sites/root/columns',
r'https://graph.microsoft.com/v1.0/me/onenote/notebooks',
r'https://graph.microsoft.com/v1.0/me/onenote/sections',
r'https://graph.microsoft.com/v1.0/me/onenote/pages',
r'https://graph.microsoft.com/v1.0/me/messages',
r'https://graph.microsoft.com/v1.0/me/mailFolders',
r'https://graph.microsoft.com/v1.0/me/outlook/masterCategories',
r'https://graph.microsoft.com/v1.0/me/mailFolders/Inbox/messages/delta',
r'https://graph.microsoft.com/v1.0/me/mailFolders/inbox/messageRules',
r"https://graph.microsoft.com/v1.0/me/messages?$filter=importance eq 'high'",
r'https://graph.microsoft.com/v1.0/me/messages?$search="hello world"',
r'https://graph.microsoft.com/beta/me/messages?$select=internetMessageHeaders&$top',
]
#微软refresh_token获取
def getmstoken(ms_token):
headers={'Content-Type':'application/x-www-form-urlencoded'
}
data={'grant_type': 'refresh_token',
'refresh_token': ms_token,
'client_id':client_id,
'client_secret':client_secret,
'redirect_uri':'http://localhost:53682/'
}
html = req.post('https://login.microsoftonline.com/common/oauth2/v2.0/token',data=data,headers=headers)
jsontxt = json.loads(html.text)
refresh_token = jsontxt['refresh_token']
access_token = jsontxt['access_token']
return access_token
#调用函数
def runapi(apilist,a):
localtime = time.asctime( time.localtime(time.time()) )
access_token=access_token_list[a-1]
headers={
'Authorization':access_token,
'Content-Type':'application/json'
}
for a in range(len(apilist)):
try:
if req.get(api_list[apilist[a]],headers=headers).status_code == 200:
print('第'+str(apilist[a])+"号api调用成功")
if config_list['是否开启各api延时'] != 'N':
time.sleep(random.randint(config_list['api延时范围开始'],config_list['api延时结束']))
except:
print("pass")
pass
#一次性获取access_token，降低获取率
for a in range(1, int(app_num)+1):
if a == 1:
client_id=os.getenv('CLIENT_ID')
client_secret=os.getenv('CLIENT_SECRET')
ms_token=os.getenv('MS_TOKEN')
access_token_list[a-1]=getmstoken(ms_token)
else:
client_id=os.getenv('CLIENT_ID_'+str(a))
client_secret=os.getenv('CLIENT_SECRET_'+str(a))
ms_token=os.getenv('MS_TOKEN_'+str(a))
access_token_list[a-1]=getmstoken(ms_token)
#随机api序列
fixed_api=[0,1,5,6,20,21]
#保证抽取到outlook,onedrive的api
ex_api=[2,3,4,7,8,9,10,22,23,24,25,26,27,13,14,15,16,17,18,19,11,12]
#额外抽取填充的api
fixed_api.extend(random.sample(ex_api,6))
random.shuffle(fixed_api)
final_list=fixed_api
#实际运行
print('共'+str(config_list['每次轮数'])+'轮')
for c in range(1,config_list['每次轮数']+1):
if config_list['是否启动随机时间'] == 'Y':
time.sleep(random.randint(config_list['延时范围起始'],config_list['结束']))
for a in range(1, int(app_num)+1):
if config_list['是否开启各账号延时'] == 'Y':
time.sleep(random.randint(config_list['账号延时范围开始'],config_list['账号延时结束']))
if a==1:
client_id=os.getenv('CLIENT_ID')
client_secret=os.getenv('CLIENT_SECRET')
print('\n'+'应用/账号 '+str(a)+' 的第'+str(c)+'轮'+time.asctime(time.localtime(time.time()))+'\n')
if config_list['是否开启随机api顺序'] == 'Y':
print("已开启随机顺序,共12个api,自己数")
apilist=final_list
runapi(apilist,a)
else:
print("原版顺序,共10个api,自己数")
apilist=[5,9,8,1,20,24,23,6,21,22]
runapi(apilista,a)
else:
client_id=os.getenv('CLIENT_ID_'+str(a))
client_secret=os.getenv('CLIENT_SECRET_'+str(a))
print('\n'+'应用/账号 '+str(a)+' 的第'+str(c)+'轮'+time.asctime(time.localtime(time.time()))+'\n')
if config_list['是否开启随机api顺序'] == 'Y':
print("已开启随机顺序,共12个api,自己数")
apilist=final_list
runapi(apilist,a)
else:
print("原版顺序,共10个api,自己数")
apilist=[5,9,8,1,20,24,23,6,21,22]
runapi(apilist,a)