1. 程式人生 > 實用技巧 >20_代理方式開啟瀏覽器

20_代理方式開啟瀏覽器

無需代理

        # 無代理情況
        cls.driver = selenium.webdriver.Chrome()
        cls.driver.get("http://daido.sitetest1.com/")
        cls.driver.maximize_window()

需要代理,代理沒有密碼

        # 有代理,代理無密碼
        # chromeOptions = webdriver.ChromeOptions()  # 設定代理
        # chromeOptions.add_argument("--proxy-server=http://192.168.116.141:808")
        # # cls.driver=webdriver.Chrome(chrome_options=chromeOptions)
        # cls.driver = webdriver.Chrome()  #不適用代理
        # cls.driver.maximize_window()
        # cls.driver.get("http://platform1.comitbpm.com/")  # rakuten

有代理,代理需要密碼

   # 有代理的情況,且代理需要密碼
    # proxy = 'test001:[email protected]:3128'
    # cls.driver = Function.Getproxy.get_driver(proxy)
    # cls.driver.get("http://platform1.comitbpm.com/")
    # cls.driver.implicitly_wait(10)  # 隱式等待10秒

此時需要使用單獨的driver,預設的不可以

下面是配置圖,和code

import string
import zipfile
from selenium import webdriver


def create_proxy_auth_extension(proxy_host, proxy_port, proxy_username, proxy_password, scheme='http',
                                plugin_path=None):
    if plugin_path is None:
        plugin_path = r'{}_{}@http-dyn.dobel.com_9020.zip'.format(proxy_username, proxy_password)

    manifest_json = """
    {
        "version": "1.0.0",
        "manifest_version": 2,
        "name": "Dobel Proxy",
        "permissions": [
            "proxy",
            "tabs",
            "unlimitedStorage",
            "storage",
            "<all_urls>",
            "webRequest",
            "webRequestBlocking"
        ],
        "background": {
            "scripts": ["background.js"]
        },
        "minimum_chrome_version":"22.0.0"
    }
    """

    background_js = string.Template(
        """
        var config = {
            mode: "fixed_servers",
            rules: {
                singleProxy: {
                    scheme: "${scheme}",
                    host: "${host}",
                    port: parseInt(${port})
                },
                bypassList: ["foobar.com"]
            }
          };

        chrome.proxy.settings.set({value: config, scope: "regular"}, function() {});

        function callbackFn(details) {
            return {
                authCredentials: {
                    username: "${username}",
                    password: "${password}"
                }
            };
        }

        chrome.webRequest.onAuthRequired.addListener(
            callbackFn,
            {urls: ["<all_urls>"]},
            ['blocking']
        );
        """
    ).substitute(
        host=proxy_host,
        port=proxy_port,
        username=proxy_username,
        password=proxy_password,
        scheme=scheme,
    )

    with zipfile.ZipFile(plugin_path, 'w') as zp:
        zp.writestr("manifest.json", manifest_json)
        zp.writestr("background.js", background_js)

    return plugin_path


def from_proxy_get_daili(proxy):
    # proxy是這種格式 user:pass@ip:port
    user_pass_str, ip_port_str = proxy.split('@')
    proxy_host, proxy_port = ip_port_str.split(':')
    proxy_user, proxy_pass = user_pass_str.split(':')
    return proxy_host, proxy_port, proxy_user, proxy_pass


def get_driver(proxy):
    proxy_host, proxy_port, proxy_user, proxy_pass = from_proxy_get_daili(proxy)
    proxy_auth_plugin_path = create_proxy_auth_extension(
        proxy_host=proxy_host,
        proxy_port=proxy_port,
        proxy_username=proxy_user,
        proxy_password=proxy_pass)
    option = webdriver.ChromeOptions()
    option.add_extension(proxy_auth_plugin_path)
    driver = webdriver.Chrome(chrome_options=option)
    return driver