nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

correctly set the ext field

Browse source
This commit is contained in:
Romain Bignon 2011-05-07 09:24:44 +02:00
commit 0061d598e0
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
weboob/backends/youporn/pages/video.py
View file

@ -34,14 +34,19 @@ class VideoPage(PornPage):
if video is None: if video is None:
video = YoupornVideo(self.group_dict['id']) video = YoupornVideo(self.group_dict['id'])
video.title = self.get_title() video.title = self.get_title()
video.url = self.get_url() video.url, video.ext = self.get_url()
self.set_details(video) self.set_details(video)
return video return video
def get_url(self): def get_url(self):
download_div = self.parser.select(self.document.getroot(), '#download', 1) download_div = self.parser.select(self.document.getroot(), '#download', 1)
a = self.parser.select(download_div, 'a', 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): def get_title(self):
element = self.parser.select(self.document.getroot(), '#videoArea h1', 1) element = self.parser.select(self.document.getroot(), '#videoArea h1', 1)