From f23a9ab4afa02854cd275d7fc0ec14a5f0a7bcc9 Mon Sep 17 00:00:00 2001 From: Julien Veyssier Date: Tue, 26 Feb 2013 00:37:53 +0100 Subject: [PATCH] [tvsub] correction when empty result --- modules/tvsubtitles/pages.py | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/modules/tvsubtitles/pages.py b/modules/tvsubtitles/pages.py index ed08f0d1..15df1da8 100644 --- a/modules/tvsubtitles/pages.py +++ b/modules/tvsubtitles/pages.py @@ -48,16 +48,17 @@ class SearchPage(BasePage): """ Page which contains results as a list of series """ def iter_subtitles(self,language): - list_result = self.parser.select(self.document.getroot(),'div.left_articles ul',1) - li_result = self.parser.select(list_result,'li') - for line in li_result: - if len(self.parser.select(line,'img[alt=%s]'%language)) > 0: - link = self.parser.select(line,'a',1) - href = link.attrib.get('href','') - self.browser.location("http://%s%s"%(self.browser.DOMAIN,href)) - assert self.browser.is_on_page(SeriePage) - for subtitle in self.browser.page.iter_subtitles(language): - yield subtitle + list_result = self.parser.select(self.document.getroot(),'div.left_articles ul') + if len(list_result) > 0: + li_result = self.parser.select(list_result[0],'li') + for line in li_result: + if len(self.parser.select(line,'img[alt=%s]'%language)) > 0: + link = self.parser.select(line,'a',1) + href = link.attrib.get('href','') + self.browser.location("http://%s%s"%(self.browser.DOMAIN,href)) + assert self.browser.is_on_page(SeriePage) + for subtitle in self.browser.page.iter_subtitles(language): + yield subtitle class SeriePage(BasePage):