From 25da89642abc5dd85786759f86bb75e7df4e6ee5 Mon Sep 17 00:00:00 2001 From: Laurent Bachelier Date: Fri, 3 Apr 2015 12:46:14 +0200 Subject: [PATCH] browser: Allow changing the Session class In a separate method, so arguments can be removed or added. --- weboob/browser/browsers.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/weboob/browser/browsers.py b/weboob/browser/browsers.py index 5a69058b..fe81e3a3 100644 --- a/weboob/browser/browsers.py +++ b/weboob/browser/browsers.py @@ -200,11 +200,14 @@ class Browser(object): else: self.logger.info(msg) + def _create_session(self): + return FuturesSession(max_workers=self.MAX_WORKERS, max_retries=self.MAX_RETRIES) + def _setup_session(self, profile): """ Set up a python-requests session for our usage. """ - session = FuturesSession(max_workers=self.MAX_WORKERS, max_retries=self.MAX_RETRIES) + session = self._create_session() session.proxies = self.PROXIES