Add proxy support to browser2

2014-03-13 18:25:56 +01:00 · 2014-03-13 18:25:56 +01:00 · cafd6c8f05
commit cafd6c8f05
parent 0fe6358f59
1 changed files with 3 additions and 1 deletions
--- a/weboob/tools/browser2/browser.py
+++ b/weboob/tools/browser2/browser.py
@ -121,9 +121,11 @@ class BaseBrowser(object):
    TIMEOUT = 10.0
    REFRESH_MAX = 0.0

-    def __init__(self, logger=None):
+    def __init__(self, logger=None, proxy=None):
        self.logger = getLogger('browser', logger)
        self._setup_session(self.PROFILE)
+        if proxy is not None:
+            self.session.proxies = {'http': proxy, 'https': proxy}
        self.url = None
        self.response = None