using contextlib instead of try-catch

This commit is contained in:
k4yt3x
2019-08-25 23:03:37 -04:00
parent 6521e071ce
commit e10315f89e
2 changed files with 5 additions and 7 deletions

View File

@@ -27,6 +27,7 @@ Installation Details:
# built-in imports
import argparse
import contextlib
import json
import os
import pathlib
@@ -274,10 +275,8 @@ def download(url, save_path, chunk_size=4096):
file_name = None
if 'content-disposition' in stream.headers:
disposition = stream.headers['content-disposition']
try:
with contextlib.suppress(IndexError):
file_name = re.findall("filename=(.+)", disposition)[0].strip('"')
except IndexError:
pass
if file_name is None:
# output_file = f'{save_path}\\{stream.url.split("/")[-1]}'