-
Notifications
You must be signed in to change notification settings - Fork 18
Expand file tree
/
Copy pathGetToken.py
More file actions
53 lines (44 loc) · 1.5 KB
/
Copy pathGetToken.py
File metadata and controls
53 lines (44 loc) · 1.5 KB
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
# -*- coding:utf-8 -*-
'''
Created on 2017年3月29日
@author: Jin
'''
import json
import requests
import xlrd
# import xlwt
from xlutils.copy import copy
Testdata = xlrd.open_workbook('C:/Jin/workpase/ApiTest/src/testdata.xls')#打开测试数据
table = Testdata.sheets()[0]#选择sheet
hurl = table.cell(7,1).value#读取URL
def test_get_token():
'''登陆'''
husername = table.cell(3,1).value
hpassword = table.cell(4,1).value
hotp = table.cell(5,1).value
hcontent_type = table.cell(6,1).value
hdata = {
"username":husername,
"password":hpassword,
"otp":hotp}
headers = {'content-type': hcontent_type
}
r = requests.post(hurl+'/login', data=json.dumps(hdata), headers=headers)
hjson = json.loads(r.text)#获取并处理返回的json数据
hcode = str(hjson['code'])
print('请求返回状态为：'+hcode)
if hcode == table.cell(9,1).value:
token = hjson['data']['token']#获取token
print('当前token为：'+token)
#将获取的TOKEN保存到testdata中
oldWb = xlrd.open_workbook('C:/Jin/workpase/ApiTest/src/testdata.xls',formatting_info=True)
newWb = copy(oldWb)
newWs = newWb.get_sheet(0)
newWs.write(8, 1, token)
print ("Token写入成功")
newWb.save('C:/Jin/workpase/ApiTest/src/testdata.xls')
print ("TestdDate保存成功")
else:
print('登陆失败，程序退出')
exit()
# test_get_token() #