[skip e2e] Add new func of get_token on utils

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
test_wt2
wangting0128 2021-12-20 12:09:47 +08:00
parent 71c595e696
commit 5cefe60489
1 changed files with 13 additions and 0 deletions

View File

@ -3,6 +3,7 @@ import time
import logging
import string
import random
import requests
import json
import os
from yaml.representer import SafeRepresenter
@ -263,3 +264,15 @@ def read_json_file(file_name):
with open(file_name) as f:
file_dict = json.load(f)
return file_dict
def get_token(url):
""" get the request token and return the value """
rep = requests.get(url)
data = json.loads(rep.text)
if 'token' in data:
token = data['token']
else:
token = ''
print("Can not get token.")
return token