zl程序教程

您现在的位置是:首页 >  Python

当前栏目

Python自动化测试Selenium+chrome连接HTTP代理(账密+白名单)

2023-02-18 16:31:23 时间

Selenium 有很多功能, 但其核心是 web 浏览器自动化的一个工具集,它允许用户模拟终端用户执行的常见活动;将文本输入到字段中,选择下拉值和复选框,并单击文档中的链接。 它还提供许多其他控件,比如鼠标移动、任意 JavaScript 执行等等。

虽然 Selenium 主要用于网站的前端测试,但其核心是浏览器用户代理库。本次来说说,Python使用Selenium调用Chrome浏览器并通过HTTP代理进行自动化测试:

白名单模式代码示例:

```python
from selenium import webdriver
from selenium.webdriver.chrome.service import Service

targetURL = "http://myip.ipip.net" #访问的目标站点
proxyAddr = "您的代理IP:端口号" 

if __name__ == '__main__':
  browser_location = r".\Chrome\chrome.exe" #指定浏览器路径位置
  driver_location = r".\Chrome\chromedriver.exe" #指定Driver路径位置

  option = webdriver.ChromeOptions()
  option.binary_location = browser_location #设置浏览器位置
  option.add_argument("--start-maximized") #窗口最大化运行
  option.add_argument('--proxy-server=%(server)s' % {"server": proxyAddr})

  driver = webdriver.Chrome(service=Service(driver_location), options=option)
  driver.get(targetURL)
  print(driver.page_source)
```

运行结果:

账密模式代码如下:

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
import string
import zipfile
targetURL = "http://d.qg.net/ip" #访问的目标站点
proxyHost = "您的代理IP"
proxyPort = "端口号" 
authKey = "请改成您的Key" 
password = "请改成您的AuthPwd"
# 账密模式
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'./{}_{}_qgnet_proxyauth_plugin.zip'.format(proxy_username, proxy_password)
    manifest_json = """
        {
            "version": "1.0.0",
            "manifest_version": 2,
            "name": "QG.NET Proxy",
            "permissions": [
                "proxy",
                "tabs",
                "unlimitedStorage",
                "storage",
                "",
                "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: ["localhost"]
            }
          };
        chrome.proxy.settings.set({value: config, scope: "regular"}, function() {});
        function callbackFn(details) {
            return {
                authCredentials: {
                    username: "${username}",
                    password: "${password}"
                }
            };
        }
        chrome.webRequest.onAuthRequired.addListener(
            callbackFn,
            {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
if __name__ == '__main__':
    # browser_location = r"C:\Users\Administrator\Desktop\Chrome\chrome.exe"  # 指定浏览器路径位置
    driver_location = r"C:\Users\Administrator\Desktop\Chrome\chromedriver.exe"  # 指定Driver路径位置
    proxy_auth_plugin_path = create_proxy_auth_extension(
        proxy_host=proxyHost,
        proxy_port=proxyPort,
        proxy_username=authKey,
        proxy_password=password)
    option = webdriver.ChromeOptions()
    # option.binary_location = browser_location #设置浏览器位置
    option.add_argument("--start-maximized") #窗口最大化运行
    option.add_extension(proxy_auth_plugin_path) #添加proxy插件
    driver = webdriver.Chrome(service=Service(driver_location), options=option)
    driver.get(targetURL)
    print(driver.page_source)

返回结果如下: