Browser: move load/dump_state to LoginBrowser
It does not make sens to store a cookie for a non-logged browser.
This commit is contained in:
parent
c36f3bb040
commit
47fe42135c
1 changed files with 16 additions and 14 deletions
|
|
@ -634,20 +634,6 @@ class PagesBrowser(DomainBrowser):
|
||||||
for url in self._urls.itervalues():
|
for url in self._urls.itervalues():
|
||||||
url.browser = self
|
url.browser = self
|
||||||
|
|
||||||
def load_state(self, state):
|
|
||||||
super(PagesBrowser, self).load_state(state)
|
|
||||||
if 'url' in state:
|
|
||||||
try:
|
|
||||||
self.location(state['url'])
|
|
||||||
except requests.exceptions.HTTPError:
|
|
||||||
pass
|
|
||||||
|
|
||||||
def dump_state(self):
|
|
||||||
state = super(PagesBrowser, self).dump_state()
|
|
||||||
if self.page:
|
|
||||||
state['url'] = self.page.url
|
|
||||||
return state
|
|
||||||
|
|
||||||
def open(self, *args, **kwargs):
|
def open(self, *args, **kwargs):
|
||||||
"""
|
"""
|
||||||
Same method than
|
Same method than
|
||||||
|
|
@ -772,6 +758,22 @@ class LoginBrowser(PagesBrowser):
|
||||||
def do_logout(self):
|
def do_logout(self):
|
||||||
self.session.cookies.clear()
|
self.session.cookies.clear()
|
||||||
|
|
||||||
|
def load_state(self, state):
|
||||||
|
super(PagesBrowser, self).load_state(state)
|
||||||
|
if 'url' in state:
|
||||||
|
try:
|
||||||
|
self.location(state['url'])
|
||||||
|
except requests.exceptions.HTTPError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
def dump_state(self):
|
||||||
|
if not self.page.logged:
|
||||||
|
return {}
|
||||||
|
state = super(LoginBrowser, self).dump_state()
|
||||||
|
if self.page:
|
||||||
|
state['url'] = self.page.url
|
||||||
|
return state
|
||||||
|
|
||||||
|
|
||||||
class APIBrowser(DomainBrowser):
|
class APIBrowser(DomainBrowser):
|
||||||
def open(self, *args, **kwargs):
|
def open(self, *args, **kwargs):
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue