This commit is contained in:
Mort Yao 2013-02-14 23:51:40 +01:00
parent 106c85e2e5
commit 0967d731af

View File

@ -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