/Python/Selenium proxy with authentication

Selenium proxy with authentication

import os
import zipfile

from selenium import webdriver

def initiate(proxy):
    proxy = proxy.replace("http://", "").replace("https://", "")
    PROXY_HOST = proxy.split("@")[1].split(":")[0]          #host
    PROXY_PORT = int(proxy.split("@")[1].split(":")[1])     #port
    PROXY_USER = proxy.split("@")[0].split(":")[0]          #username
    PROXY_PASS = proxy.split("@")[0].split(":")[1]          #password

    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: "%s",
                port: parseInt(%s)
            },
            bypassList: ["localhost"]
            }
        };

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

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

    chrome.webRequest.onAuthRequired.addListener(
                callbackFn,
                {urls: ["<all_urls>"]},
                ['blocking']
    );
    """ % (PROXY_HOST, PROXY_PORT, PROXY_USER, PROXY_PASS)

    path = os.path.dirname(os.path.abspath(__file__))
    chrome_options = webdriver.ChromeOptions()
    pluginfile = 'proxy_auth_plugin.zip'

    with zipfile.ZipFile(pluginfile, 'w') as zp:
        zp.writestr("manifest.json", manifest_json)
        zp.writestr("background.js", background_js)
    chrome_options.add_extension(pluginfile)
        os.path.join(path, 'chromedriver'),
        chrome_options=chrome_options)
    return driver

def main():
    driver = initiate("username:password@host:port")
    #driver.get('https://www.google.com/search?q=my+ip+address')
    driver.get('https://api.ipify.org/?format=json')

if __name__ == '__main__':
    main()

Source: stackoverflow.com

Similar codes

0 vote129 views1 answer
Generator comprehension python
Python
1 vote137 views1 answer
How to aggregate multiple columns in pyspark
Python
1 vote158 views1 answer
Install xgboost
Python
39 votes140 views6 answers
How to use a class in python
Python
3 votes93 views1 answer
How to present if statement in dfd
Python
0 vote130 views1 answer
Requirements.txt dev python
Python
15 votes127 views1 answer
Python dict to kwargs
Python
0 vote151 views1 answer
Padnas check if string is in list of strings
Python
0 vote126 views1 answer
Filter field set in django formds
Python
0 vote203 views1 answer
Django float validator
Python Django