from selenium import webdriver
import string
import zipfile
import os


# 代理服务器
proxyHost = "http-dyn.abuyun.com"
proxyPort = "9020"

# 代理隧道验证信息 使用自己的
proxyUser = "H8S586W474G3FFFF"
proxyPass = "F516EB8****"

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 = os.getcwd()+r'/{}_{}@http-dyn.abuyun.com_9020.zip'.format(proxy_username, proxy_password)

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

    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)
    # print("bbbbbb")

    return plugin_path

#创建插件代理,如果文件已经创建,就不必再创建了
# proxy_auth_plugin_path = create_proxy_auth_extension(
#     proxy_host=proxyHost,
#     proxy_port=proxyPort,
#     proxy_username=proxyUser,
#     proxy_password=proxyPass)
# print("cccccccc")

option = webdriver.ChromeOptions()
#不显示UI界面,会报错
# option.add_argument('-headless')

option.add_argument("--start-maximized")
# option.add_extension(proxy_auth_plugin_path)
option.add_extension('./[email protected]_9020.zip')
# print("ddddddd")
driver = webdriver.Chrome(options=option,executable_path=r'F:\MyDatas\chromedriver.exe')
# print("eeeeeee")
driver.get('https://www.baidu.com/s?wd=ip')