sgpe: fix crash when there isn't any card

This commit is contained in:
Romain Bignon 2013-07-28 21:30:13 +02:00
commit 482a7c3c6c

View file

@ -73,8 +73,13 @@ class SGPEBrowser(BaseBrowser):
def cards(self): def cards(self):
doc = self.get_document(self.openurl('/Pgn/NavigationServlet?PageID=CartesFutures&MenuID=%sOPF&Classeur=1&NumeroPage=1&PageDetail=1' % self.MENUID)) doc = self.get_document(self.openurl('/Pgn/NavigationServlet?PageID=CartesFutures&MenuID=%sOPF&Classeur=1&NumeroPage=1&PageDetail=1' % self.MENUID))
try:
url = doc.xpath('//iframe[@name="cartes"]')[0].attrib['src'] url = doc.xpath('//iframe[@name="cartes"]')[0].attrib['src']
except IndexError:
return False
else:
self.location(url) self.location(url)
return True
def history(self, _id, page=1): def history(self, _id, page=1):
if page > 1: if page > 1:
@ -94,8 +99,7 @@ class SGPEBrowser(BaseBrowser):
for acc in self.page.get_list(): for acc in self.page.get_list():
yield acc yield acc
self.cards() if self.cards():
assert self.is_on_page(CardsPage) assert self.is_on_page(CardsPage)
for acc in self.page.get_list(): for acc in self.page.get_list():
yield acc yield acc