Ramhm · February 12, 2021 14:59 · Jul 12, 2018 · Jul 12, 2018
diff --git a/proxy.py b/proxy.py
@@ -76,7 +76,6 @@ def proxy_chrome(PROXY_HOST,PROXY_PORT,PROXY_USER,PROXY_PASS):
     co.add_argument('--disable-gpu')
     #disable infobars
     co.add_argument('--disable-infobars')
-    #disable image load in the session
     co.add_experimental_option("excludeSwitches",["ignore-certificate-errors"])
     #location of chromedriver, please change it according to your project.
     chromedriver = os.getcwd()+'/Chromedriver/chromedriver'

diff --git a/proxy.py b/proxy.py
@@ -0,0 +1,86 @@
+from selenium import webdriver
+from selenium.webdriver.common.proxy import Proxy
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
+from selenium.webdriver.chrome.options import Options
+import zipfile,os
+
+def proxy_chrome(PROXY_HOST,PROXY_PORT,PROXY_USER,PROXY_PASS):
+    manifest_json = """
+            {
+                "version": "1.0.0",
+                "manifest_version": 2,
+                "name": "Chrome Proxy",
+                "permissions": [
+                    "proxy",
+                    "tabs",
+                    "unlimitedStorage",
+                    "storage",
+                    "<all_urls>",
+                    "webRequest",
+                    "webRequestBlocking"
+                ],
+                "background": {
+                    "scripts": ["background.js"]
+                },
+                "minimum_chrome_version":"22.0.0"
+            }
+            """
+
+    background_js = """
+    var config = {
+            mode: "fixed_servers",
+            rules: {
+              singleProxy: {
+                scheme: "http",
+                host: "%(host)s",
+                port: parseInt(%(port)d)
+              },
+              bypassList: ["foobar.com"]
+            }
+          };
+
+    chrome.proxy.settings.set({value: config, scope: "regular"}, function() {});
+
+    function callbackFn(details) {
+        return {
+            authCredentials: {
+                username: "%(user)s",
+                password: "%(pass)s"
+            }
+        };
+    }
+
+    chrome.webRequest.onAuthRequired.addListener(
+                callbackFn,
+                {urls: ["<all_urls>"]},
+                ['blocking']
+    );
+        """ % {
+            "host": PROXY_HOST,
+            "port": PROXY_PORT,
+            "user": PROXY_USER,
+            "pass": PROXY_PASS,
+        }
+
+
+    pluginfile = 'extension/proxy_auth_plugin.zip'
+
+    with zipfile.ZipFile(pluginfile, 'w') as zp:
+        zp.writestr("manifest.json", manifest_json)
+        zp.writestr("background.js", background_js)
+
+    co = Options()
+    #extension support is not possible in incognito mode for now
+    #co.add_argument('--incognito')
+    co.add_argument('--disable-gpu')
+    #disable infobars
+    co.add_argument('--disable-infobars')
+    #disable image load in the session
+    co.add_experimental_option("excludeSwitches",["ignore-certificate-errors"])
+    #location of chromedriver, please change it according to your project.
+    chromedriver = os.getcwd()+'/Chromedriver/chromedriver'
+    co.add_extension(pluginfile)
+    driver = webdriver.Chrome(chromedriver,chrome_options=co)
+    #return the driver with added proxy configuration.
+    return driver