什么是Selenium库:
自动化测试工具,支持多种浏览器。支持的浏览器包括IE(7, 8, 9, 10, 11),Mozilla Firefox,Safari,Google Chrome,Opera等。
爬虫中主要用来解决JavaScript渲染的问题。用于驱动浏览器,并且给予浏览器动作。
安装Selenium库:pip3 install selenium
Selcnium库的使用详解:
在使用之前我们需要安装webDriver驱动,具体安装方式,自行百度,切记版本对应。
基本使用:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 基本用法 from selenium import webdriver from selenium.webdriver.common.by import By from selenium.webdriver.common.keys import Keys from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.support.wait import WebDriverWait browser = webdriver.Chrome() try : browser.get( " http://www.baidu.com " ) input = browser.find_element_by_id( ' kw ' ) input.send_keys( ' Python ' ) input.send_keys(Keys.ENTER) wait = WebDriverWait(browser, 10 ) wait.until(EC.presence_of_element_located((By.ID, ' content_left ' ))) print (browser.current_url) print (browser.get_cookies()) print (browser.page_source) finally : browser.close()
如果这段代码可以运行,说明你的webDriver版本正确(需要安装Google浏览器)
运行结果:
声明浏览器对象:
刚才我们说了Selenium支持多浏览器,下面我看下分别怎么进行声明
# !/usr/bin/env python # -*- coding: utf-8 -*- # 声明浏览器对象 from selenium import webdriver browser = webdriver.Chrome() browser = webdriver.Safari() browser = webdriver.Edge() browser = webdriver.Firefox() browser = webdriver.PhantomJS()
我这里没有安装那些浏览器,就不给大家运行代码了,建议使用Chrome浏览器(Google谷歌浏览器)
访问页面:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 访问页面 from selenium import webdriver browser = webdriver.Chrome() browser.get( " http://baidu.com " ) print (browser.page_source) browser.close()
运行结果:
查找元素:
单个元素:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 查找元素,单个元素 from selenium import webdriver browser = webdriver.Chrome() browser.get( " http://taobao.com " ) input_first = browser.find_element_by_id( ' q ' ) input_second = browser.find_element_by_css_selector( ' #q ' ) input_three = browser.find_element_by_xpath( ' //*[@id="q"] ' ) print (input_first) print (input_second) print (input_three) browser.close()
运行结果:
-
find_element_by_name
-
find_element_by_xpath
-
find_element_by_link_text
-
find_element_by_partial_link_text
-
find_element_by_tag_name
-
find_element_by_class_name
-
find_element_by_css_selector
这些都为查找方式
也可以用通用方式来查找:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 查找元素,单个元素 from selenium import webdriver from selenium.webdriver.common.by import By browser = webdriver.Chrome() browser.get( " http://taobao.com " ) input_first = browser.find_element(By.ID, ' q ' ) print (input_first) browser.close()
运行结果:
多个元素:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 查找元素,多个元素 from selenium import webdriver from selenium.webdriver.common.by import By browser = webdriver.Chrome() browser.get( " http://taobao.com " ) input_first = browser.find_elements_by_css_selector( ' .service-bd li ' ) for i in input_first: print (i) browser.close()
运行结果:
还有很多方法和find_elment用法完全一致,返回一个列表数据。
元素交互操作:
对获取的元素调用交互方法:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 元素交互操作 from selenium import webdriver from selenium.webdriver.common.by import By browser = webdriver.Chrome() browser.get( " http://baidu.com " ) input_first = browser.find_element(By.ID, ' kw ' ) input_first.send_keys( ' python从入坑到放弃 ' ) button = browser.find_element_by_class_name( ' bg s_btn ' ) button.click()
运行代码我们会看到打开Chrome浏览器,并且输入要搜索的内容,然后点击搜索按钮。更多操作访问地址:https://selenium-python.readthedocs.io/api.html#module-selenium.webdriver.remote.webelement
交互操作:
将动作附加到动作链中串行执行
# !/usr/bin/env python # -*- coding: utf-8 -*- # 交互操作 from selenium import webdriver from selenium.webdriver import ActionChains browser = webdriver.Chrome() url = ' https://www.runoob.com/try/try.php?filename=jqueryui-api-droppable ' browser.get(url) browser.switch_to.frame( ' iframeResult ' ) source = browser.find_element_by_id( ' draggable ' ) target = browser.find_element_by_id( ' droppable ' ) actions = ActionChains(browser) actions.drag_and_drop(source, target) actions.perform()
运行代码我们会看到内部的滑块进行了拖拽操作。更多详细的操作可以访问:https://selenium-python.readthedocs.io/api.html#module-selenium.webdriver.common.action_chains
执行Javascript: ⭐️ ⭐️ ⭐️ ⭐️ ⭐️
# !/usr/bin/env python # -*- coding: utf-8 -*- # 执行javascript from selenium import webdriver browser = webdriver.Chrome() browser.get( ' https://www.zhihu.com/explore ' ) browser.execute_script( ' window.scrollTo(0,document.body.scrollHeight) ' ) browser.execute_script( ' alert("弹出") ' )
运行代码我们可以看到,滚动条被下拉,并且给予了弹出框。
获取元素信息:
获取属性:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 获取元素信息:获取属性 from selenium import webdriver browser = webdriver.Chrome() url = " http://www.zhihu.com/explore " browser.get(url) logo = browser.find_element_by_id( ' zh-top-link-logo ' ) print (logo) print (logo.get_attribute( ' class ' ))
运行结果:
获取文本值:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 获取文本值 from selenium import webdriver browser = webdriver.Chrome() url = " http://www.zhihu.com/explore " browser.get(url) question = browser.find_element_by_class_name( ' zu-top-add-question ' ) print (question.text)
运行结果:
获取ID,位置,标签名,大小:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 获取ID,位置,标签名,大小 from selenium import webdriver browser = webdriver.Chrome() url = " http://www.zhihu.com/explore " browser.get(url) question = browser.find_element_by_class_name( ' zu-top-add-question ' ) print (question.id) print (question.location) print (question.tag_name) print (question.size)
运行结果:
Frame:
# !/usr/bin/env python # -*- coding: utf-8 -*- # Frame from selenium import webdriver from selenium.common.exceptions import NoSuchElementException browser = webdriver.Chrome() url = ' https://www.runoob.com/try/try.php?filename=jqueryui-api-droppable ' browser.get(url) browser.switch_to.frame( ' iframeResult ' ) source = browser.find_element_by_id( ' draggable ' ) print (source) try : logo = browser.find_element_by_class_name( ' logo ' ) except NoSuchElementException: print ( " NO LOGO " ) browser.switch_to.parent_frame() logo = browser.find_element_by_class_name( ' logo ' ) print (logo) print (logo.text)
运行结果:
等待:
隐式等待 :
当使用了隐式等待执行测试的时候,如果WebDriver没有在DOM中找到元素,将继续等待,超出设定时间则抛出找不到元素的异常,换句话来说,当元素或查找元素没有立即出现的时候,隐式等待将等待一段时间再查找DOM,默认时间是0
# !/usr/bin/env python # -*- coding: utf-8 -*- # 隐式等待 from selenium import webdriver browser = webdriver.Chrome() url = " http://www.zhihu.com/explore " browser.get(url) input = browser.find_element_by_class_name( ' zu-top-add-question ' ) print (input)
运行结果:
显示等待:比较常用
# !/usr/bin/env python # -*- coding: utf-8 -*- # 显示等待 from selenium import webdriver from selenium.webdriver.common.by import By from selenium.webdriver.common.keys import Keys from selenium.webdriver.support import expected_conditions as EC from selenium.webdriver.support.wait import WebDriverWait browser = webdriver.Chrome() browser.get( " http://www.taobao.com " ) wait = WebDriverWait(browser, 10 ) wait.until(EC.presence_of_element_located((By.ID, ' q ' ))) button = wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR, ' .btn-search ' ))) print (input,button)
- title_is 标题是某内容
-
title_contains 标题包含某内容
-
presence_of_element_located 元素加载出,传入定位元祖,如(By.ID,'p')
-
visibility_of_element_located 元素可见,传入定位元祖
-
visibility_of 可见,传入元素对象
-
presence_of_all_elements_located 所有元素加载出
-
text_to_be_present_in_element 某个元素文本包含某文字
-
text_to_be_present_in_element_value 某个元素值包含某文字
-
frame_to_be_available_and_switch_to_it 加载并切换
-
invisibility_of_element_located 元素不可见
-
element_to_be_clickable 元素可点击
-
staleness_of 判断一个元素是否仍在DOM,可判断页面是否已经刷新
-
element_to_be_selected 元素可选择,传元素对象
-
element_located_to_be_selected 元素可以选择,传入定位元祖
-
element_selection_state_to_be 传入元素对象以及状态,相等返回True,否则返回False
-
element_located_selection_state_to_be 传入定位元祖以及状态,相等返回True,否则返回False
-
alert_is_present 是否出现Alert
详细内容,可以阅读官方地址:https://selenium-python.readthedocs.io/api.html#module-selenium.webdriver.support.expected_conditions
前进和后退:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 前进和后退 from selenium import webdriver browser = webdriver.Chrome() browser.get( " http://www.taobao.com " ) browser.get( " http://www.baidu.com " ) browser.get( " http://www.zhihu.com " ) browser.back() browser.forward()
运行代码我们会看到优先大家taobao.com然后打开baidu.com,最后打开zhihu.com,然后执行退回动作和前进动作
Cookies:
# !/usr/bin/env python # -*- coding: utf-8 -*- # Cookies from selenium import webdriver browser = webdriver.Chrome() browser.get( " http://www.zhihu.com " ) print (browser.get_cookies()) browser.add_cookie({ ' name ' : ' admin ' , ' domain ' : ' www.zhihu.com ' , ' value ' : ' cxiaocai ' }) print (browser.get_cookies()) browser.delete_all_cookies() print (browser.get_cookies())
运行结果:
选项卡管理:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 选项卡管理 from selenium import webdriver browser = webdriver.Chrome() browser.get( " http://www.baidu.com " ) browser.execute_script( ' window.open() ' ) print (browser.window_handles) browser.switch_to.window(browser.window_handles[ 1 ]) browser.get( ' http://www.taobao.com ' ) browser.switch_to.window(browser.window_handles[0]) browser.get( ' http://www.zhihu.com ' )
也可以使用浏览器的快捷方式的操作键位来打开窗口(不建议这样使用,建议使用上面的方式来管理选项卡)
异常处理:
# !/usr/bin/env python # -*- coding: utf-8 -*- # 异常处理 from selenium import webdriver from selenium.common.exceptions import TimeoutException,NoSuchElementException browser = webdriver.Chrome() try : browser.get( " http://www.baidu.com " ) except TimeoutException: print ( " 请求超时 " ) try : browser.find_element_by_id( ' hello ' ) except NoSuchElementException: print ( " NoSuchElementException " )
运行结果:
由于异常处理比较复杂,异常也有很多,在这里不在一一列举了,建议大家去官网查看,地址:https://selenium-python.readthedocs.io/api.html#module-selenium.common.exceptions
上述代码地址:https://gitee.com/dwyui/senlenium.git
到这里Selenium库的使用就说完了,python用于爬虫的库就说了这么多,前面的urllib,Requests,BeautfuliSoup,PyQuery还有今天的Selenium库,明天开始直接讲解真实案例,最近我会整理几个简单的小爬虫案例。