diff --git a/src/you_get/common.py b/src/you_get/common.py index e2ab59a22ed0fa6cff36f69f769355764bde641f..4a833aea5030a4c7de337212b72c237fb37703b4 100644 --- a/src/you_get/common.py +++ b/src/you_get/common.py @@ -133,9 +133,12 @@ def url_info(url, faker = False): ext = mapping[type] else: type = None - filename = parse.unquote(r1(r'filename="?(.+)"?', headers['content-disposition'])) - if len(filename.split('.')) > 1: - ext = filename.split('.')[-1] + if headers['content-disposition']: + filename = parse.unquote(r1(r'filename="?(.+)"?', headers['content-disposition'])) + if len(filename.split('.')) > 1: + ext = filename.split('.')[-1] + else: + ext = None else: ext = None