python websocket接口测试_python测试http、websocket接口

importurllib3importrequestsimportloggingimporttimefrom web_wss importwss

urllib3.disable_warnings()

time_log= time.strftime('%Y-%m-%d %H_%M_%S', time.localtime(time.time()))#logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(message)s')

logging.basicConfig(filename= 'result.log', filemode= 'w+', level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

deflogin():

list_uid=[]

list_head=[]

phone= phonelogin_header= {'User-Agent': 'test/1.0.0(build:test;ANDROID 9)','bndkey': 'test'}for line inphone:try:#requests.adapters.DEFAULT_RETRIES = 10

#s = requests.session()

#s.keep_alive = False

sessionRequest=requests.session()#登陆

app_login = {'account': '86,' +line,'accountKind': '1','passwd': 'test'}

r_login= sessionRequest.post(url=url_login, data=app_login, headers=login_header, verify=False, timeout=10)#print(r_login.url)

if '登录成功' inr_login.text:

logging.info(u'登陆成功:' + line + '-' + r_login.json()['data']['uid'])else:

logging.info(u'登陆失败:' +r_login.text)#请求头

uid = r_login.json()['data']['uid']

u_token= dict(r_login.headers)['utoken']

t_token= dict(r_login.headers)['ttoken']

head= {'User-Agent': 'test/1.0.0(build:test;ANDROID 7.1.1)','bndkey': 'test','utoken': u_token,'ttoken': t_token}

r_start= sessionRequest.post(url=url_open_take, data={'main': '1'}, headers=head, verify=False, timeout=10)if 'true' inr_start.text:

logging.info(u'接单开启成功: uid -' + str(uid) + '-' +r_start.text)else:

logging.info(u'接单开启失败: uid -' + str(uid) + '-' +r_start.text)

r_sell= requests.post(url=url_sell, data={'merchantDeposit': '1', 'playerDeposit': '1'}, headers=head, verify=False, timeout=10)if 'true' inr_sell.text:

logging.info(u'卖币开启成功: uid -' + str(uid) + '-' +r_sell.text)else:

logging.info(u'卖币开启失败: uid -' + str(uid) + '-' +r_sell.text)

list_uid.append(uid)

list_head.append(head)

logging.info(u'uid:' + str(uid) + '- utoken:' + str(u_token) + '- ttoken:' + str(t_token) + '\n')#logging.info(u'HEAD: ' + str(list_head) + '\n')

exceptBaseException as msg:

logging.info(u'---ERROR：' + str(msg) + '\n' + '\n' + '\n')returnlist_uid, list_headdefaccomplishOrder():

list_data=login()

list_uid=list_data[0]

list_head= list_data[1]for uid, head inzip(list_uid, list_head):try:

sessionRequest=requests.session()#list_order_id = []

#list_amount = []

try:if wss(uid) is notNone:#抢单

order_id_sec =wss(uid)

r_rob= sessionRequest.post(url=url_rob, data={'id': order_id_sec}, headers=head, verify=False, timeout=10)

logging.info(u'抢单:' +r_rob.text)#查询进行中订单

r_get_order = sessionRequest.get(url=url_run_order, headers=head, verify=False, timeout=10)if r_get_order.json()['data'] is notNone:

order_data=r_get_order.json()for i in range(len(r_get_order.json()['data'])):

order_id= order_data['data'][i]['id']

amount= order_data['data'][i]['amount']['reality']#list_order_id.append(order_id)

#list_amount.append(amount)

## logging.info('uid: ' + uid + ' - orderId: ' + order_id + '\n')

#return list_order_id, list_amount

#完成订单

r_pay = sessionRequest.post(url=url_pay, data={'id': order_id, 'amount': amount}, headers=head, verify=False, timeout=10)if 'true' inr_pay.text:

logging.info(u'完成订单: uid -' + uid + 'orderId:' + order_id + '\n')else:

logging.info(u'订单未完成: uid -' + uid +r_pay.text)else:

logging.info(u'暂时没有进行中订单:' + 'uid:' +uid)exceptBaseException as msg:

logging.info(u'---ERROR：' + str(msg) + '\n')exceptBaseException as msg:

logging.info(u'--- ERROR：' + str(msg) + '\n')if __name__ == '__main__':whileTrue:

accomplishOrder()