river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-26 12:57:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

重构

Browse Source
This commit is contained in:
Jing Ling
2020-01-07 01:34:43 +08:00
parent c16bcada89
commit ae8a7a65d0
2 changed files with 26 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/common/request.py
+23 -25
View File
@@ -141,33 +141,31 @@ def get_title(markup):
def request_callback(future, index, datas):
try:
result = future.result()
except Exception as e:
logger.log('TRACE', e.args)
datas[index]['reason'] = str(e.args)
result = future.result()
if isinstance(result, Exception):
logger.log('TRACE', result.args)
datas[index]['reason'] = str(result.args)
datas[index]['valid'] = 0
else:
if isinstance(result, tuple):
resp, text = result
datas[index]['reason'] = resp.reason
datas[index]['status'] = resp.status
if resp.status == 400 or resp.status >= 500:
datas[index]['valid'] = 0
else:
datas[index]['valid'] = 1
headers = resp.headers
banner = str({'Server': headers.get('Server'),
'Via': headers.get('Via'),
'X-Powered-By': headers.get('X-Powered-By')})
datas[index]['banner'] = banner[1:-1]
title = get_title(text).strip()
datas[index]['title'] = utils.remove_string(title)
datas[index]['header'] = str(dict(headers))[1:-1]
datas[index]['response'] = utils.remove_string(text)
else:
datas[index]['reason'] = 'Something error'
if isinstance(result, tuple):
resp, text = result
datas[index]['reason'] = resp.reason
datas[index]['status'] = resp.status
if resp.status == 400 or resp.status >= 500:
datas[index]['valid'] = 0
else:
datas[index]['valid'] = 1
headers = resp.headers
banner = str({'Server': headers.get('Server'),
'Via': headers.get('Via'),
'X-Powered-By': headers.get('X-Powered-By')})
datas[index]['banner'] = banner[1:-1]
title = get_title(text).strip()
datas[index]['title'] = utils.remove_string(title)
datas[index]['header'] = str(dict(headers))[1:-1]
datas[index]['response'] = utils.remove_string(text)
else:
datas[index]['reason'] = str(result)
datas[index]['valid'] = 0
def get_connector():