Fix crash on missing 'content-type' header

master
trivernis 4 years ago
parent 212c7575bd
commit 2be171a09e

@ -51,9 +51,12 @@ def request_loop(client: Client, urls: [str], fm: FileManager, method: str = 'GE
try:
req = client.request(url, method=method, data=body, verify=verify)
if req.status_code == 200:
extension = mimetypes.guess_extension(req.headers['content-type'].split(';')[0])
content_type = 'text/plain'
if 'content-type' in req.headers.keys():
content_type = req.headers['content-type'].split(';')[0]
extension = mimetypes.guess_extension(content_type)
print('[+] Request to %s succeeded: mime: %s, timing: %ss' %
(url, req.headers['content-type'], req.elapsed.total_seconds()))
(url, content_type, req.elapsed.total_seconds()))
f_name = time.strftime('%d-%m-%y_%H-%M-%S') + extension
with fm.get_file(d, f_name) as f:
f.write(req.text)

Loading…
Cancel
Save