fix crash when there is no thumbnail (closes #704)

This commit is contained in:
Romain Bignon 2011-10-08 15:15:58 +02:00
commit b65f38da33

View file

@ -20,6 +20,7 @@
from datetime import datetime from datetime import datetime
from weboob.capabilities.base import NotAvailable
from weboob.tools.capabilities.thumbnail import Thumbnail from weboob.tools.capabilities.thumbnail import Thumbnail
from weboob.tools.browser import BasePage from weboob.tools.browser import BasePage
from .video import CanalplusVideo from .video import CanalplusVideo
@ -49,7 +50,11 @@ class VideoPage(BasePage):
video.description = infos.find('DESCRIPTION').text video.description = infos.find('DESCRIPTION').text
media = el.find('MEDIA') media = el.find('MEDIA')
video.thumbnail = Thumbnail(media.find('IMAGES').find('PETIT').text) url = media.find('IMAGES').find('PETIT').text
if url:
video.thumbnail = Thumbnail(url)
else:
video.thumbnail = NotAvailable
lastest_format = None lastest_format = None
for format in media.find('VIDEOS'): for format in media.find('VIDEOS'):
if format.text is None: if format.text is None:
@ -70,19 +75,19 @@ class VideoPage(BasePage):
def iter_results(self): def iter_results(self):
for vid in self.document.getchildren(): for vid in self.document.getchildren():
yield self.parse_video(vid) yield self.parse_video(vid)
def iter_channel(self): def iter_channel(self):
for vid in self.document.getchildren(): for vid in self.document.getchildren():
yield self.parse_video_channel(vid) yield self.parse_video_channel(vid)
def parse_video_channel(self,el): def parse_video_channel(self,el):
_id = el[0].text _id = el[0].text
video = CanalplusVideo(_id) video = CanalplusVideo(_id)
video.title = el[2][3][0].text video.title = el[2][3][0].text
video.date = datetime.now() video.date = datetime.now()
return video return video
def get_video(self, video, quality): def get_video(self, video, quality):
_id = self.group_dict['id'] _id = self.group_dict['id']
for vid in self.document.getchildren(): for vid in self.document.getchildren():