little factorization

This commit is contained in:
smurail 2014-08-01 10:54:31 +02:00 committed by Romain Bignon
commit 07294ac0f6

View file

@ -24,9 +24,8 @@ from weboob.capabilities.bank import CapBank, AccountNotFound
from weboob.tools.backend import BaseBackend, BackendConfig from weboob.tools.backend import BaseBackend, BackendConfig
from weboob.tools.value import ValueBackendPassword, Value from weboob.tools.value import ValueBackendPassword, Value
from .browser import LCLBrowser from .browser import LCLBrowser, LCLProBrowser
from .enterprise.browser import LCLEnterpriseBrowser from .enterprise.browser import LCLEnterpriseBrowser
from .browser import LCLProBrowser
__all__ = ['LCLBackend'] __all__ = ['LCLBackend']
@ -48,19 +47,17 @@ class LCLBackend(BaseBackend, CapBank):
BROWSER = LCLBrowser BROWSER = LCLBrowser
def create_default_browser(self): def create_default_browser(self):
website = self.config['website'].get() # assume all `website` option choices are defined here
if website == 'ent': browsers = {'par': LCLBrowser,
self.BROWSER = LCLEnterpriseBrowser 'pro': LCLProBrowser,
return self.create_browser(self.config['login'].get(), 'ent': LCLEnterpriseBrowser}
self.config['password'].get())
elif website == 'pro': website_value = self.config['website']
self.BROWSER = LCLProBrowser self.BROWSER = browsers.get(website_value.get(),
return self.create_browser(self.config['login'].get(), browsers[website_value.default])
self.config['password'].get())
else: return self.create_browser(self.config['login'].get(),
self.BROWSER = LCLBrowser self.config['password'].get())
return self.create_browser(self.config['login'].get(),
self.config['password'].get())
def deinit(self): def deinit(self):
# don't need to logout if the browser hasn't been used. # don't need to logout if the browser hasn't been used.