From 0061d598e0002be8fd2587a8e3d3c015ef25ae33 Mon Sep 17 00:00:00 2001 From: Romain Bignon Date: Sat, 7 May 2011 09:24:44 +0200 Subject: [PATCH] correctly set the ext field --- weboob/backends/youporn/pages/video.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/weboob/backends/youporn/pages/video.py b/weboob/backends/youporn/pages/video.py index 9391f908..03d1f163 100644 --- a/weboob/backends/youporn/pages/video.py +++ b/weboob/backends/youporn/pages/video.py @@ -34,14 +34,19 @@ class VideoPage(PornPage): if video is None: video = YoupornVideo(self.group_dict['id']) video.title = self.get_title() - video.url = self.get_url() + video.url, video.ext = self.get_url() self.set_details(video) return video def get_url(self): download_div = self.parser.select(self.document.getroot(), '#download', 1) a = self.parser.select(download_div, 'a', 1) - return a.attrib['href'] + m = re.match('^(\w+) - .*', a.text) + if m: + ext = m.group(1).lower() + else: + ext = 'flv' + return a.attrib['href'], ext def get_title(self): element = self.parser.select(self.document.getroot(), '#videoArea h1', 1)