From: Jaime Marquínez Ferrándiz Date: Fri, 12 Jul 2013 19:52:59 +0000 (+0200) Subject: Use determine_ext when saving the thumbnail X-Git-Url: https://git.cielonegro.org/gitweb.cgi?a=commitdiff_plain;h=cbdbb7666540ba07ab7e4a3a7bc34759bf0ca6d9;p=youtube-dl.git Use determine_ext when saving the thumbnail Urls that contain a query produced filenames with wrong extensions --- diff --git a/youtube_dl/YoutubeDL.py b/youtube_dl/YoutubeDL.py index e24706115..cc5f76157 100644 --- a/youtube_dl/YoutubeDL.py +++ b/youtube_dl/YoutubeDL.py @@ -530,9 +530,7 @@ class YoutubeDL(object): if self.params.get('writethumbnail', False): if 'thumbnail' in info_dict: - thumb_format = info_dict['thumbnail'].rpartition(u'/')[2].rpartition(u'.')[2] - if not thumb_format: - thumb_format = 'jpg' + thumb_format = determine_ext(info_dict['thumbnail'], u'jpg') thumb_filename = filename.rpartition('.')[0] + u'.' + thumb_format self.to_screen(u'[%s] %s: Downloading thumbnail ...' % (info_dict['extractor'], info_dict['id'])) diff --git a/youtube_dl/utils.py b/youtube_dl/utils.py index 532ca585d..cf2ea654e 100644 --- a/youtube_dl/utils.py +++ b/youtube_dl/utils.py @@ -650,12 +650,12 @@ def unified_strdate(date_str): pass return upload_date -def determine_ext(url): +def determine_ext(url, default_ext=u'unknown_video'): guess = url.partition(u'?')[0].rpartition(u'.')[2] if re.match(r'^[A-Za-z0-9]+$', guess): return guess else: - return u'unknown_video' + return default_ext def date_from_str(date_str): """