Ensure filename is always set and report any errors while trying to get the filename from a url.

This commit is contained in:
John Schember 2011-03-12 14:40:41 -05:00
parent 77a9964fac
commit 5de6672807

View File

@ -519,6 +519,7 @@ def get_download_filename(url, cookie_file=None):
cj.load(cookie_file)
br.set_cookiejar(cj)
try:
with closing(br.open(url)) as r:
disposition = r.info().get('Content-disposition', '')
for p in disposition.split(';'):
@ -534,6 +535,10 @@ def get_download_filename(url, cookie_file=None):
filename = filename[:-1]
filename = urllib2_unquote(filename)
break
except:
import traceback
traceback.print_exc()
if not filename:
filename = r.geturl().split('/')[-1]