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

[tvsub] correction when empty result

Browse source
This commit is contained in:
Julien Veyssier 2013-02-26 00:37:53 +01:00
commit f23a9ab4af
1 changed files with 11 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

21
modules/tvsubtitles/pages.py
View file

@ -48,16 +48,17 @@ class SearchPage(BasePage):
""" Page which contains results as a list of series """ Page which contains results as a list of series
""" """
def iter_subtitles(self,language): def iter_subtitles(self,language):
list_result = self.parser.select(self.document.getroot(),'div.left_articles ul',1) list_result = self.parser.select(self.document.getroot(),'div.left_articles ul')
li_result = self.parser.select(list_result,'li') if len(list_result) > 0:
for line in li_result: li_result = self.parser.select(list_result[0],'li')
if len(self.parser.select(line,'img[alt=%s]'%language)) > 0: for line in li_result:
link = self.parser.select(line,'a',1) if len(self.parser.select(line,'img[alt=%s]'%language)) > 0:
href = link.attrib.get('href','') link = self.parser.select(line,'a',1)
self.browser.location("http://%s%s"%(self.browser.DOMAIN,href)) href = link.attrib.get('href','')
assert self.browser.is_on_page(SeriePage) self.browser.location("http://%s%s"%(self.browser.DOMAIN,href))
for subtitle in self.browser.page.iter_subtitles(language): assert self.browser.is_on_page(SeriePage)
yield subtitle for subtitle in self.browser.page.iter_subtitles(language):
yield subtitle
class SeriePage(BasePage): class SeriePage(BasePage):