51Testing软件测试论坛

标题: Chrome代理、UA和Cookie如何在Selenium设置？ [打印本页]

作者: lsekfe 时间: 2023-5-8 13:16
标题: Chrome代理、UA和Cookie如何在Selenium设置？
　一、selenium设置Chrome User-Agent+代理
　　先上代码，这里我直接用的UA轮子，随机取出一个来设置。
　　from selenium import webdriver
　　from selenium.webdriver.chrome.options import Options
　　from fake_useragent import UserAgent
　　headers = {'User-Agent': UserAgent().random} #从UA轮子里随机抽选一个用作配置，也可以自己手动设置
　　print(headers) #自己手动设置，格式像这样（看输出）
　　proxy = "127.0.0.1:8080" #这里是我自己本地开的一个代理服务用作测试
　　ops = Options()
　　ops.add_argument('User-Agent={}'.format(headers['User-Agent'])) #配置为自己设置的UA
　　ops.add_argument('--proxy-server=http://%s' % proxy) #添加代理格式为ip:port
　　browser = webdriver.Chrome(options=ops)
　　browser.get('http://192.168.31.130/dvwa/index.php')

　　输出：
　　{'User-Agent': 'Mozilla/5.0 (X11; NetBSD) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/27.0.1453.116 Safari/537.36'}

　　可在burpsuit上看到代理的流量说明代理成功，并且UA也已设置为自己配置的UA。

　　二、selenium设置Chrome Cookie
　　登录后可看到Cookie长这样：

　　错误示例
　　这估计很多人都踩了这个坑：
　　browser.delete_all_cookies()  # 删除原来的cookie
　　browser.add_cookie({'think_lang': 'zh-cn', 'PHPSESSID': '6be4ba542aeecd9bef7967269add2835'})
　　browser.get('http://192.168.31.130/dvwa/index.php')

　　然后就会报这个错。
　　InvalidArgumentException: Message: invalid argument: missing 'name'
　　正确示例
　　我这样写是为了方便大家更清楚当Cookie有多个 name 和 value 时该如何设置。
　　cookie1 = {'name': 'security', 'value': 'impossible'}
　　cookie2 = {'name': 'PHPSESSID', 'value': 'n5u6pc2lul897cdca7pek6s713'}
　　browser.delete_all_cookies()  # 删除原来的cookie
　　browser.add_cookie(cookie1)
　　browser.add_cookie(cookie2)
　　browser.get('http://192.168.31.130/dvwa/index.php')

　　当然你也可以用一个for循环显得更专业（doge）。
　　cookie = {
　　 'security': 'impossible',
　　 'PHPSESSID': 'n5u6pc2lul897cdca7pek6s713'
　　}
　　browser.delete_all_cookies()  # 删除原来的cookie
　　for name, value in cookie.items():
　　 browser.add_cookie({'name':name,'value':value})
　　browser.get('http://192.168.31.130/dvwa/index.php')

　　完整代码
　　from selenium import webdriver
　　from selenium.webdriver.chrome.options import Options
　　from fake_useragent import UserAgent
　　headers = {'User-Agent': UserAgent().random} #从UA轮子里随机抽选一个用作配置，也可以自己手动设置
　　print(headers) #自己手动设置，格式像这样（看输出）
　　proxy = "127.0.0.1:8080" #这里是我自己本地开的一个代理服务用作测试
　　ops = Options()
　　ops.add_argument('User-Agent={}'.format(headers['User-Agent'])) #配置为自己设置的UA
　　ops.add_argument('--proxy-server=http://%s' % proxy) #添加代理格式为ip:port
　　browser = webdriver.Chrome(options=ops)
　　browser.get('http://192.168.31.130/dvwa/index.php')
　　cookie = {
　　 'security': 'impossible',
　　 'PHPSESSID': 'n5u6pc2lul897cdca7pek6s713'
　　}
　　browser.delete_all_cookies()  # 删除原来的cookie
　　for name, value in cookie.items():
　　 browser.add_cookie({'name':name,'value':value})
　　browser.get('http://192.168.31.130/dvwa/index.php')

　　这里运行完如果想退出selenium驱动的浏览器记得用browser.quit()，手动叉掉的话进程可能没关掉。
　　browser.quit()

　　代码功能：代理并设置自定义UA然后用Cookie自动登录成功。

欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/)