Selenium基本使用
Selenium
简介
Selenium是一个Web的自动化测试工具，最初是为网站自动化测试而开发的，类型像我们玩游戏用的按键精灵，可以按指定的命令自动操作，不同是Selenium 可以直接运行在浏览器上，它支持所有主流的浏览器（包括PhantomJS这些无界面的浏览器，但是现在谷歌的无头比较火，下面展示谷歌的无头）
Selenium 可以根据我们的指令，让浏览器自动加载页面，获取需要的数据，甚至页面截屏，或者判断网站上某些动作是否发生。
Selenium 自己不带浏览器，不支持浏览器的功能，它需要与第三方浏览器结合在一起才能使用，本质是python通过代码，借助于浏览器驱动，操作浏览器。我们可以安装PhantomJS、谷歌无头(推荐)，火狐无头···
👉无头安装参考：https://www.cnblogs.com/Neeo/articles/13949854.html#phantomjs
安装
- 安装：
pip install selenium - Selenium 官方参考文档：http://selenium-python.readthedocs.io/index.html
这里使用谷歌无头
对比无头浏览器的优势这里就不再对比了，可以自行百度这些无头的优劣，安装谷歌无头需要安装对应版本的驱动，把驱动放在项目下；
👉谷歌驱动：https://registry.npmmirror.com/binary.html?path=chromedriver/
基本使用
from selenium import webdriver
import time
# 用代码打开一个浏览器
bro=webdriver.Chrome(executable_path='./chromedriver.exe')
# 在地址栏输入地址
bro.get('https://www.baidu.com')
# 找到输入框
search=bro.find_element_by_id('kw')
# 模拟输入框输入内容
search.send_keys("美女")
# 找到百度一下按钮
button=bro.find_element_by_id('su')
# 点击一下按钮
button.click()
time.sleep(2)
# 打印当前页面的html内容
print(bro.page_source)
with open('baidu.html','w',encoding='utf-8') as f:
f.write(bro.page_source) # 包含redner+ajax
bro.close()
无头浏览器#
在做爬虫的时候，不希望显示的打开浏览器，但是selenium必须要用浏览器，让浏览器不显示，后台运行，完成爬虫；
demo
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
# 得到一个配置对象
chrome_options = Options()
chrome_options.add_argument('window-size=1920x3000') #指定浏览器分辨率
chrome_options.add_argument('--disable-gpu') # 谷歌文档提到需要加上这个属性来规避bug
chrome_options.add_argument('--hide-scrollbars') # 隐藏滚动条, 应对一些特殊页面
chrome_options.add_argument('blinfk-settings=imagesEnabled=alse') # 不加载图片, 提升速度
chrome_options.add_argument('--headless') # 浏览器不提供可视化页面. linux下如果系统不支持可视化不加这条会启动失败
bro=webdriver.Chrome(executable_path='./chromedriver.exe',options=chrome_options)
bro.get('http://www.cnblogs.com')
print(bro.page_source)
bro.close()
获取元素位置，属性，大小#
一般用于破解验证码
属性：location、tag_name、size
print(tag.id) # id，但是不是标签的id，selenium提供的一个id
print(tag.location) # 位置
print(tag.tag_name) # 标签名
print(tag.size) # 标签的大小
demo
from selenium import webdriver
import time
from PIL import Image
bro=webdriver.Chrome(executable_path='./chromedriver.exe') #
# 在地址栏输入地址
bro.get('https://www.jd.com/')
# 找到图片
img=bro.find_element_by_css_selector('a.logo_tit_lk')
# print(img.location) # 图片位置 {'x': 105, 'y': 41}
# print(img.size) # 图片大小 通过位置和大小可以唯一确定这张图，通过截图可以把图截出来
# print(img.id) # selenium提供的id号，忽略
# print(img.tag_name) # a
location=img.location
size=img.size
bro.save_screenshot('./main.png') # 把整个页面保存成图片
# pillow抠图，把图标抠出来
# 第一个参数 开始截图的x坐标
# 第二个参数 开始截图的y坐标
# 第三个参数 结束截图的x坐标
# 第四个参数 结束截图的y坐标
img_tu = (int(location['x']), int(location['y']), int(location['x'] + size['width']), int(location['y'] + size['height']))
# #使用pillow打开截图
img=Image.open('./main.png')
#从截图中按照位置扣除验证码
code_img=img.crop(img_tu)
# 把扣出来的图，保存到本地
code_img.save('./code.png')
bro.close()
## 补充：标签位置和大小:size和location
# 一般用来扣验证码图片：可能会由于分辨率问题导致扣出的图不一致---》通过修改分辨率--》实现正确抠图
# 验证码是img---》src--》自己加载就能拿到验证码，保存到本地即可(requests)-->更简单
等待元素加载的方式#
- 强制等待：设置固定休眠时间，单位为秒。 由python的
time包提供, 导入 time 包后就可以使用。缺点：不智能，使用太多的sleep会影响脚本运行速度。
- 显示等待：每个标签都要写等待的逻辑操作，建议参考显示等待
- 隐式等待：由webdriver提供的方法，一旦设置，这个隐式等待会在WebDriver对象实例的整个生命周期起作用，它不针对某一个元素，是全局元素等待，即在定位元素时，需要等待页面全部元素加载完成，才会执行下一个语句。如果超出了设置时间的则抛出异常。
强制等待#
import time
time.sleep(5)
隐士等待#
from selenium import webdriver
driver=webdriver.Chrome(executable_path='./chromedriver.exe')
driver.implicitly_wait(10) #隐式等待10秒
# 取这个标签，如果取不到就等待，直到标签加载完成或10s到了
显示等待#
需要通过
from selenium.webdriver.support.wait import WebDriverWait导入模块
WebDriverWait(driver,timeout,poll_frequency=0.5,ignored_exceptions=None)
- driver：浏览器驱动
- timeout：最长超时时间，默认以秒为单位
- poll_frequency：检测的间隔步长，默认为0.5s
- ignored_exceptions：超时后的抛出的异常信息，默认抛出NoSuchElementExeception异常。
与until()或者until_not()方法结合使用#
WebDriverWait(driver,10).until(method，message="")
调用该方法提供的驱动程序作为参数，直到返回值为True
WebDriverWait(driver,10).until_not(method，message="")
调用该方法提供的驱动程序作为参数，直到返回值为False
在设置时间（10s）内，等待后面的条件发生。如果超过设置时间未发生，则抛出异常。在等待期间，每隔一定时间（默认0.5秒)，调用until或until_not里的方法，直到它返回True或False.
元素操作demo#
元素操作方法
- find_element_by_id ：通过id找
- find_element_by_link_text ：通过a标签文字
- find_element_by_partial_link_text ：通过a标签文字模糊查找
- find_element_by_tag_name ：通过标签名找
- find_element_by_class_name ：通过类名找
- find_element_by_name ：通过name属性找
- find_element_by_css_selector ：css选择器
- find_element_by_xpath ：xpath选择器
from selenium import webdriver
import time
bro=webdriver.Chrome(executable_path='./chromedriver') # mac linux
bro.get('https://www.baidu.com/')
bro.implicitly_wait(10) # 隐式等待
# 查找a标签文本内容是登陆的
login_a=bro.find_element_by_link_text('登录')
# login_a=bro.find_element_by_link_id('s-top-loginbtn')
# 点击a标签
login_a.click()
# 找到账号登陆，点击
login_pwd_btn=bro.find_element_by_id('TANGRAM__PSP_11__changePwdCodeItem')
login_pwd_btn.click()
# 找到用户名的输入框和密码的输入框，输入用户名密码
username=bro.find_element_by_name('userName')
pwd=bro.find_element_by_css_selector('#TANGRAM__PSP_11__password')
username.send_keys("换成你的账号")
pwd.send_keys('换成你的密码')
time.sleep(3)
username.clear()
submit=bro.find_element_by_id('TANGRAM__PSP_11__submit') # 提交按钮
submit.click()
# 弹出验证码识别--》可以手动点击
# 登陆成功
time.sleep(5)
bro.close()
执行js#
selenium也可以操作js代码，主要有以下这几种情况，第一种是操控页面滑动，第二种是使用当前页面中得一些变量，执行页面中得函数
from selenium import webdriver
import time
bro=webdriver.Chrome(executable_path='./chromedriver.exe')
bro.get('https://www.pearvideo.com/category_9')
bro.implicitly_wait(10) # 隐式等待
# bro.execute_script("alert('hello')")
# 第一种情况，控制操作页面滑动
# bro.execute_script('window.scrollBy(0, document.body.scrollHeight)') # 向下移动
# time.sleep(1)
# bro.execute_script('window.scrollBy(0, document.body.scrollHeight)')
# time.sleep(1)
# bro.execute_script('window.scrollBy(0, document.body.scrollHeight)')
'''第二种情况：使用当前页面中的一些变量,执行页面中的函数'''
bro.execute_script('alert(md5_vm_test())')
time.sleep(5)
bro.close()
切换选项卡#
import time
from selenium import webdriver
browser=webdriver.Chrome(executable_path='./chromedriver.exe')
browser.get('https://www.baidu.com')
# 打开选项卡
browser.execute_script('window.open()')
print(browser.window_handles) #获取所有的选项卡
browser.switch_to.window(browser.window_handles[1])
browser.get('https://www.taobao.com')
time.sleep(2)
browser.switch_to.window(browser.window_handles[0])
browser.get('https://www.sina.com.cn')
browser.close() # 关闭当前选项卡
browser.quit() # 退出浏览器
模拟前进后退#
import time
from selenium import webdriver
browser=webdriver.Chrome(executable_path='./chromedriver')
browser.get('https://www.baidu.com')
browser.get('https://www.taobao.com')
browser.get('http://www.sina.com.cn/')
browser.back()
time.sleep(2)
browser.forward()
browser.close()
异常处理#
from selenium import webdriver
from selenium.common.exceptions import TimeoutException,NoSuchElementException,NoSuchFrameException
try:
browser=webdriver.Chrome()
browser.get('http://www.runoob.com/try/try.php?filename=jqueryui-api-droppable')
browser.switch_to.frame('iframssseResult')
except TimeoutException as e:
print(e)
except NoSuchFrameException as e:
print(e)
finally:
browser.close()
selenium登录cnblogs获取cookie
'''
先使用selenium 半自动登录到cnblogs，然后取出cookie存到本地，下次使用selenium 访问cnblogs只需加载之前的cookie即可变成了登陆状态
'''
from selenium import webdriver
import json
import time
bro=webdriver.Chrome(executable_path='./chromedriver')
# 登陆取cookie的过程
# try:
# bro.get('http://www.cnblogs.com')
# bro.implicitly_wait(10) # 隐士等待
# submit_a=bro.find_element_by_link_text('登录')
# submit_a.click() # 点击
# username=bro.find_element_by_id('mat-input-0') # 找到输入框
# password=bro.find_element_by_id('mat-input-1')
# username.send_keys('输入账号')
# password.send_keys('输入密码') # 手动输入
#
#
#
# # submit=bro.find_element_by_class_name('mat-button-wrapper')
# # submit.click()
# input() # 手动输入密码，点击登录，验证码通过，再敲回车，继续往下走
# # 弹出验证码---》不好破--->手动操作---》
# # 登陆成功了
# # 把cookie保存到本地
# # print(bro.get_cookies())
#
# with open('cnblogs.json','w',encoding='utf-8') as f:
# json.dump(bro.get_cookies(),f)
#
#
# except Exception as e:
# print(e)
# finally:
# bro.close()
# 访问写入cookie
try:
bro.get('http://www.cnblogs.com')
bro.implicitly_wait(10)
# 写入本地的cookie
with open('cnblogs.json','r',encoding='utf-8') as f:
cookie_dic=json.load(f)
# 写入到浏览器
for item in cookie_dic: # 设置cookie必须用字典，cookie的json文件是列表，所以用循环往里放
bro.add_cookie(item)
bro.refresh() # 刷新一下浏览器
time.sleep(2)
except Exception as e:
print(e)
finally:
bro.close()
打码平台使用
这里使用超老鹰作为示例，来使用，破解验证码；当然还有云打码；
import requests
from hashlib import md5
class ChaojiyingClient():
def __init__(self, username, password, soft_id):
self.username = username
password = password.encode('utf8')
self.password = md5(password).hexdigest()
self.soft_id = soft_id
self.base_params = {
'user': self.username,
'pass2': self.password,
'softid': self.soft_id,
}
self.headers = {
'Connection': 'Keep-Alive',
'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
}
def PostPic(self, im, codetype):
"""
im: 图片字节
codetype: 题目类型 参考 http://www.chaojiying.com/price.html
"""
params = {
'codetype': codetype,
}
params.update(self.base_params)
files = {'userfile': ('ccc.jpg', im)}
r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files, headers=self.headers)
return r.json()
def PostPic_base64(self, base64_str, codetype):
"""
im: 图片字节
codetype: 题目类型 参考 http://www.chaojiying.com/price.html
"""
params = {
'codetype': codetype,
'file_base64':base64_str
}
params.update(self.base_params)
r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, headers=self.headers)
return r.json()
def ReportError(self, im_id):
"""
im_id:报错题目的图片ID
"""
params = {
'id': im_id,
}
params.update(self.base_params)
r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
return r.json()
if __name__ == '__main__':
chaojiying = ChaojiyingClient('账号', '密码', '软件id') #用户中心>>软件ID 生成一个替换 96001
im = open('./b.png', 'rb').read() #本地图片文件路径 来替换 a.jpg 有时WIN系统须要//
print(chaojiying.PostPic(im, 6001)) #1902 验证码类型 官方网站>>价格体系 3.4+版 print 后要加()
#print chaojiying.PostPic(base64_str, 1902) #此处为传入 base64代码
xpath使用
# 常用
'''
. # 选取当前节点。
.. # 选取当前节点的父节点。
/ # 表示当前路径
// # 表示任意路径，子子孙孙
nodename # a img p 节点名字
—举例
//div # //div 在当前html的任意路径下找div
/div # 只找本层的div
* # 任意标签
@href # 取这个标签的属性
/text() # 获取标签的文本
'''
doc='''
<html>
<head>
<base href='http://example.com/' />
<title>Example website</title>
</head>
<body>
<div id='images'>
<a href='image1.html'>Name: My image 1 <br /><img src='image1_thumb.jpg' /></a>
<a href='image2.html'>Name: My image 2 <br /><img src='image2_thumb.jpg' /></a>
<a href='image3.html'>Name: My image 3 <br /><img src='image3_thumb.jpg' /></a>
<a href='image4.html'>Name: My image 4 <br /><img src='image4_thumb.jpg' /></a>
<a href='image5.html' class='li li-item' name='items'>Name: My image 5 <br /><img src='image5_thumb.jpg' /></a>
<a href='image6.html' name='items'><span><h5>test</h5></span>Name: My image 6 <br /><img src='image6_thumb.jpg' /></a>
</div>
</body>
</html>
'''
from lxml import etree
html=etree.HTML(doc)
# html=etree.parse('search.html',etree.HTMLParser())
# 1 所有节点
# a=html.xpath('//*')
# 2 指定节点（结果为列表）
# a=html.xpath('//head')
# 3 子节点，子孙节点
# a=html.xpath('//div/a')
# a=html.xpath('//body/a') #无数据
# a=html.xpath('//body//a')
# 4 父节点
# a=html.xpath('//body//a[@href="image1.html"]/..')
# a=html.xpath('//body//a[1]/..')
# 也可以这样
# a=html.xpath('//body//a[1]/parent::*')
# 5 属性匹配
# a=html.xpath('//body//a[@href="image1.html"]')
# 6 文本获取
# a=html.xpath('//body//a[@href="image1.html"]/text()')
# 7 属性获取
# a=html.xpath('//body//a/@href')
# # 注意从1 开始取（不是从0）
# a=html.xpath('//body//a[1]/@href')
# 8 属性多值匹配
# a 标签有多个class类，直接匹配就不可以了，需要用contains
# a=html.xpath('//body//a[@class="li"]')
# a=html.xpath('//body//a[contains(@class,"li")]')
# a=html.xpath('//body//a[contains(@class,"li")]/text()')
# 9 多属性匹配
# a=html.xpath('//body//a[contains(@class,"li") or @name="items"]')
# a=html.xpath('//body//a[contains(@class,"li") and @name="items"]/text()')
# # a=html.xpath('//body//a[contains(@class,"li")]/text()')
# 10 按序选择
# a=html.xpath('//a[2]/text()')
# a=html.xpath('//a[2]/@href')
# 取最后一个
# a=html.xpath('//a[last()]/@href')
# 位置小于3的
# a=html.xpath('//a[position()<3]/@href')
# 倒数第二个
# a=html.xpath('//a[last()-2]/@href')
# 11 节点轴选择
# ancestor：祖先节点
# 使用了* 获取所有祖先节点
# a=html.xpath('//a/ancestor::*')
# # 获取祖先节点中的div
# a=html.xpath('//a/ancestor::div')
# attribute：属性值
# a=html.xpath('//a[1]/attribute::*')
# child：直接子节点
# a=html.xpath('//a[1]/child::*')
# descendant：所有子孙节点
# a=html.xpath('//a[6]/descendant::*')
# following:当前节点之后所有节点
# a=html.xpath('//a[1]/following::*')
# a=html.xpath('//a[1]/following::*[1]/@href')
# following-sibling:当前节点之后同级节点
# a=html.xpath('//a[1]/following-sibling::*')
# a=html.xpath('//a[1]/following-sibling::a')
# a=html.xpath('//a[1]/following-sibling::*[2]')
# a=html.xpath('//a[1]/following-sibling::*[2]/@href')
# print(a)
# 中级大招---》复制
# //*[@id="maincontent"]/div[5]/table/tbody/tr[2]/td[2]
爬取京东商品
这里爬取精品内衣作为示例：
from selenium import webdriver
from selenium.webdriver.common.keys import Keys # 键盘按键操作
import time
def get_goods(driver):
try:
# 找到所有类名叫gl-item的标签
goods = driver.find_elements_by_class_name('gl-item')
for good in goods:
detail_url = good.find_element_by_tag_name('a').get_attribute('href')
p_name = good.find_element_by_css_selector('.p-name em').text.replace('\n', '')
price = good.find_element_by_css_selector('.p-price i').text
p_commit = good.find_element_by_css_selector('.p-commit a').text
img=good.find_element_by_css_selector('div.p-img img').get_attribute('src')
if not img:
img='http:'+good.find_element_by_css_selector('div.p-img img').get_attribute('data-lazy-img')
msg = '''
商品 : %s
链接 : %s
图片 : %s
价钱 ：%s
评论 ：%s
''' % (p_name, detail_url,img, price, p_commit)
print(msg, end='\n\n')
button = driver.find_element_by_partial_link_text('下一页')
button.click()
time.sleep(1)
get_goods(driver)
except Exception:
pass
def spider(url, keyword):
driver = webdriver.Chrome(executable_path='./chromedriver.exe')
driver.get(url)
driver.implicitly_wait(3) # 使用隐式等待
try:
input_tag = driver.find_element_by_id('key')
input_tag.send_keys(keyword)
input_tag.send_keys(Keys.ENTER) # 敲回车
get_goods(driver)
finally:
driver.close()
if __name__ == '__main__':
spider('https://www.jd.com/', keyword='精品内衣')
