Proxy example for splinter(0.7.4) on python2.7

splinter_proxy_example.py

# -*- coding: utf-8 -*-
"""
ProxyBrowser for splinter(0.7.4) on python2.7
Working on Chrome and Phantomjs
"""

from functools import partial

from selenium import webdriver
from splinter import Browser
from splinter.exceptions import DriverNotFoundError


class ProxyBrowser(object):
    def __new__(cls, driver='chrome', proxy=None, proxy_type='http', **kwargs):
        """
        >>> browser = ProxyBrowser(driver='chrome', proxy='61.132.74.89:1080', proxy_type='socks5')
        >>> browser.visit('http://wtfismyip.com')
        """
        browser = partial(Browser, driver)

        proxy_kwargs = cls.get_proxy_kwargs(driver, proxy, proxy_type)
        kwargs.update(proxy_kwargs)

        return browser(**kwargs)

    @classmethod
    def get_proxy_kwargs(cls, driver, proxy, proxy_type):
        """
        :param driver: chrome | phantomjs
        :param proxy: {address}:{port}
        :param proxy_type: http | socks5
        """
        proxy_server = '{0}://{1}'.format(proxy_type, proxy)
        print proxy_server
        if proxy:
            if driver == 'phantomjs':
                return {'service_args': ['--proxy={}'.format(proxy), '--proxy-type={}'.format(proxy_type)]}
            elif driver == 'chrome':
                chrome_options = webdriver.ChromeOptions()
                chrome_options.add_argument('--proxy-server={}'.format(proxy_server))
                return {'options': chrome_options}
            else:
                raise DriverNotFoundError('Not supported driver: {}'.format(driver))
        else:
            return {}