def crest_data():
uinfo_list = []
for i in range(910, 920):
data = user_login(i)
login_token = data.get('token')
uid = data.get('user').get('uid')
nickname = data.get('user').get('nickname')
avatar = data.get('user').get('avatar')
uinfo = {'token': login_token, 'uid': uid, 'nickname': nickname, 'avatar': avatar}
uinfo_list.append(uinfo)
DataUtils.set_data(uinfo_list, constant.uinfo_api_pressure_file)
print(uinfo_list)
def downdload_gift_icon():
file_path = BASE_DIR + os.sep + 'data_fj' + os.sep + 'giftdata'
FileUtils.clear_file(file_path)
dataList = DataUtils.get_yaml_data('gift_list.yaml')
# print(dataList)
for i in range(len(dataList)):
iconUrl = dataList[i].get('icon')
fileName = file_path + os.sep + 'gift' + '_' + str(dataList[i].get(
'gift_id')) + '_' + dataList[i].get('name') + '.png'
DownLoadThread(iconUrl, fileName).start()
def send_req(type, url, params):
'''调试用'''
type = type.lower()
header = DataUtils.get_user_info_auto()
header.update(FanjiaoConfig.signature_fj(params, type))
requests.DEFAULT_RETRIES = 5
s = requests.session()
s.keep_alive = False
if 'get'.__eq__(type):
r = requests.get(url, headers=header, params=params, timeout=10)
else:
r = requests.post(url, headers=header, json=params, timeout=10)
LogUtils.print_response(r, isformart=True)
return r
def get_curl_command1(type, *data):
'''
:param type: 接口类型 get post
:param data: url ,params
:return:
'''
datas: dict = DataUtils.get_yaml_data('curl_model.yaml')
# print(datas)
typedata: str = datas.get(type)
if type == 'post':
url, params = data
command = typedata.format(url, params)
else:
url = data[0]
command = typedata.format(url)
print(command)
return command
def send_pre_req(req, url_data):
'''
发送请求的方法
压测脚本调用
:param req: 请求client
:param url_data: 接口的参数url等信息
:return: 响应response
'''
# 压测接口调用
api = url_data.get('url')
url = constant.BASE_URL + api
params = url_data.get('params')
type = url_data.get('type').lower()
header = DataUtils.get_user_info_pressure()
header.update(FanjiaoConfig.signature_fj(params, type))
if 'get'.__eq__(type):
r = req.get(url, headers=header, params=params, name=api)
else:
r = req.post(url, headers=header, json=params, name=api)
LogUtils.print_response(r, url_data.get('comment'))
return r
def send_auto_request(params_data):
'''
发送请求的方法
接口自动化脚本调用
:param params_data: 接口的参数url等信息
:return: 响应response
'''
url = constant.BASE_URL + params_data.get('url')
params = params_data.get('params')
type = params_data.get('type').lower()
header = DataUtils.get_user_info_auto()
header.update(FanjiaoConfig.signature_fj(params, type))
requests.DEFAULT_RETRIES = 5
s = requests.session()
s.keep_alive = False
if 'get'.__eq__(type):
r = requests.get(url, headers=header, params=params, timeout=10)
else:
r = requests.post(url, headers=header, json=params, timeout=10)
LogUtils.print_response(r, params_data.get('comment'), isformart=True)
return r
def setUp(self) -> None:
self.url_data = DataUtils.get_yaml_data('t07_account_api_data.yaml')
def setUp(self) -> None:
self.url_data = DataUtils.get_yaml_data('t06_home_api_data.yaml')
def setUp(self) -> None:
self.url_data = DataUtils.get_yaml_data('t05_album_info_api_data.yaml')
def setUp(self) -> None:
self.url_data = DataUtils.get_yaml_data('t08_comment_api_data.yaml')