selenium+firefox+js实现动态设置firefox浏览器代理IP

from selenium import webdriver
import time,random
from redis import ConnectionPool, StrictRedis


REDIS_IPPool_ARGS = {
   'host': '192.168.2.99',
   'port': 6379,
   'db': 14    # 储存的IP格式：http://xxx.xxx.xxx.xxx:xxx
}

pool_ip = ConnectionPool(**REDIS_IPPool_ARGS)
conn_ip = StrictRedis(connection_pool=pool_ip)

def get_ip():
    # 随机取IP
    ip = conn_ip.lindex("proxies", random.choice(range(0, conn_ip.llen("proxies")))).decode('utf-8')
    return ip.split(':')[1][2:], ip.split(':')[-1]

ip_addr, ip_port = get_ip()
print(ip_addr, ip_port)

js_base='''var pf = Components.classes["@mozilla.org/preferences-service;1"].getService(Components.interfaces.nsIPrefBranch);

    pf.setIntPref("network.proxy.type", 1);

    pf.setCharPref("network.proxy.http", "{0}");

    pf.setIntPref("network.proxy.http_port", {1});

    pf.setCharPref("network.proxy.ssl", "{2}");

    pf.setIntPref("network.proxy.ssl_port", {3});'''

driver = webdriver.Firefox()
driver.get("about:config")    # 如果不请求这个页面，js中会抛出Components.classes未定义的异常
driver.execute_script(js_base.format(ip_addr,ip_port,ip_addr,ip_port))

# 测试ip是否设置成功
driver.get('https://www.baidu.com')
driver.find_element_by_xpath('//input[@id="kw"]').send_keys('ip')
driver.find_element_by_xpath('//input[@id="su"]').click()
time.sleep(6)

ip_addr, ip_port = get_ip()
print(ip_addr, ip_port)

driver.get("about:config")
driver.execute_script(js_base.format(ip_addr,ip_port,ip_addr,ip_port))    # 切换IP

# 测试ip是否设置成功
driver.get('https://www.baidu.com')
driver.find_element_by_xpath('//input[@id="kw"]').send_keys('ip')
driver.find_element_by_xpath('//input[@id="su"]').click()
time.sleep(10)

driver.quit()