From 7891ce61184eea477c3a9210c3a3be298363a0fb Mon Sep 17 00:00:00 2001 From: Romain Bignon Date: Wed, 25 Mar 2015 01:21:28 +0100 Subject: [PATCH] fix crash if dump_state() occurs without a current page --- weboob/browser/browsers.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/weboob/browser/browsers.py b/weboob/browser/browsers.py index 08642b58..cb912518 100644 --- a/weboob/browser/browsers.py +++ b/weboob/browser/browsers.py @@ -762,7 +762,8 @@ class LoginBrowser(PagesBrowser): self.session.cookies.clear() def load_state(self, state): - super(PagesBrowser, self).load_state(state) + super(LoginBrowser, self).load_state(state) + if 'url' in state: try: self.location(state['url']) @@ -770,11 +771,11 @@ class LoginBrowser(PagesBrowser): pass def dump_state(self): - if not self.page.logged: + if not self.page or not self.page.logged: return {} + state = super(LoginBrowser, self).dump_state() - if self.page: - state['url'] = self.page.url + state['url'] = self.page.url return state