ability to set a proxy address

This commit is contained in:
Romain Bignon 2010-05-11 15:12:16 +02:00
commit 99c64d4849

View file

@ -129,7 +129,7 @@ class BaseBrowser(mechanize.Browser):
default_features.remove('_robots')
def __init__(self, username=None, password=None, firefox_cookies=None,
parser=None, history=NoHistory()):
parser=None, history=NoHistory(), proxy=None):
"""
Constructor of Browser.
@ -146,6 +146,10 @@ class BaseBrowser(mechanize.Browser):
['User-agent', self.USER_AGENT]
]
# Use a proxy
if proxy:
self.set_proxies({"http": proxy})
# Share cookies with firefox
if firefox_cookies and HAVE_COOKIES:
self._cookie = FirefoxCookieJar(self.DOMAIN, firefox_cookies)