1.声明浏览器对象
- from selenium import webdriver
-
- browser = webdriver.Chrome()
- # browser = webdriver.Firefox()
- # browser = webdriver.Edge()
2.访问页面
- from selenium import webdriver
-
- browser = webdriver.Chrome()
- url = 'http://www.baidu.com'
- browser.get(url)
- print(browser.page_source)
- browser.close() #使用完要关闭浏览器
3.节点查找
3.1 单个节点
从淘宝首页提取搜索框
- from selenium import webdriver
- from selenium.webdriver.common.by import By
-
- browser = webdriver.Chrome()
- url = 'http://www.taobao.com'
- browser.get(url)
- input_first = browser.find_element_by_id('q') #根据id获取
- input_second = browser.find_elements_by_css_selector('#q') #根据css选择器获取
- input_third = browser.find_element_by_xpath('//*[@id="q"]') #根据xpath获取
- input_forth = browser.find_element(By.ID,'q') #find_element()
- print(input_first,input_second,input_third,input_forth)
- browser.close()
运行结果为WebElement类型,<selenium.webdriver.remote.webelement.WebElement (session="47f594f34cc9868f68ab3653e1d542a6">
find_element()方法只查询到第一个节点
3.2 多个节点
查询左侧导航栏
- from selenium import webdriver
- from selenium.webdriver.common.by import By
-
- url = 'http://www.taobao.com'
- browser = webdriver.Chrome()
- browser.get(url)
- lis1 = browser.find_elements_by_css_selector('.service-bd li') #class属性为service-bd的ul下的所有li标签,返回结果为列表
- lis2 = browser.find_elements(By.CSS_SELECTOR,'.service-bd li')
- print(len(lis1),lis1)
- print(len(lis2),lis2)
- browser.close()
3.3 节点交互
Selenium可以驱动浏览器来执行一些操作,常用方法有:
在输入框中输入要查找的内容进行搜索
- from selenium import webdriver
- import time
-
- url = 'http://www.taobao.com'
- browser = webdriver.Chrome()
- browser.get(url)
- input = browser.find_element_by_id('q')
- input.send_keys('电脑')
- time.sleep(3)
- input.clear()
- input.send_keys('笔记本电脑')
- button = browser.find_elements_by_css_selector('.btn-search tb-bg')
- button.click()
- browser.close()
3.4 执行JavaScript
对于某些操作selenium API没有提供,如下拉进度条,可以直接模拟运行javascript,使用execute_script()方法。
- from selenium import webdriver
- import time
-
- url = 'http://www.taobao.com'
- browser = webdriver.Chrome()
- browser.get(url)
- browser.execute_script('window.scrollTo(0,document.body.scrollHeight)')
- time.sleep(5)
- browser.execute_script('alert("To Buttom")')
4.获取节点信息
page_source属性可以获取网页源代码,然后可以使用解析库获取信息。
selenium选择节点,返回的WebElement类型,也有对应的属性和方法来获取节点信息。
4.1 获取属性
get_attribute()方法获取属性。
- input = browser.find_element_by_id('q')
- print(input.get_attribute('name'))
4.2 获取文本值
text属性可获取文本值。
- lis = browser.find_elements_by_css_selector('.service-bd li')
- for li in lis:
- print(li.text)
4.3 获取id、位置、标签名和大小
id属性获取节点id;location属性获取该节点在页面中的相对位置;tag_name属性获取标签名;size属性获取节点宽高。
5.切换Frame
iframe即子Frame,相当于页面的子页面,结构和外部网页的结构完全一致,selenium打开页面后,默认在父级的Frame里,不能获取到在子Frame里的节点,需要使用switch_to.frame()方法来切换Frame。
- from selenium import webdriver
- from selenium.common.exceptions import NoSuchElementException
- browser = webdriver.Chrome()
- url = 'https://www.cdsy.xyz/try/try.php?filename=jqueryui-api-droppable'
- browser.get(url)
- browser.switch_to.frame('iframeResult') #切换到子Frame
- try:
- logo1 = browser.find_element_by_class_name('logo')
- except NoSuchElementException:
- print('no logo')
- browser.switch_to.parent_frame() #切换到父级Frame
- logo = browser.find_element_by_class_name('logo')
- print(logo)
- print(logo.text)
- browser.close()
6.延时等待
selenium中get()方法会在网页框架加载结束后结束执行,此时如果获取page_source,可能不是浏览器完全加载的页面,所以需要延时等待一段时间,确保节点全部加载。
6.1 隐式等待
使用隐式等待,如果selenium在DOM中没有找到节点,会继续等待,超出设定时间后,会抛出找不到节点的异常,规定了一个固定时间,默认时间为0。用implicitly_wait()实现隐式等待。
- from selenium import webdriver
- browser = webdriver.Chrome()
- browser.implicitly_wait(10)
- browser.get('http://www.baidu.com')
- input = browser.find_element_by_id('kw')
- print(input)
- browser.close()
6.2显示等待
显示等待指定一个最长等待时间,如果在规定时间内没有加载出来,会抛出超时异常。
- from selenium import webdriver
- from selenium.webdriver.common.by import By
- from selenium.webdriver.support.ui import WebDriverWait
- from selenium.webdriver.support import expected_conditions as EC
-
- browser = webdriver.Chrome()
- url = 'http://www.taobao.com'
- browser.get(url)
- wait = WebDriverWait(browser,10) #引入WebDriverWait对象,指定最长等待时间
- input = wait.until(EC.presence_of_element_located((By.ID,'q'))) #调用until()方法,传入等待条件expected_conditions
- button = wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR,'.btn-search')))
- print(input,button)
- browser.close()
7.前进和后退
back()后退;forward()前进。
- from selenium import webdriver
- import time
-
- browser = webdriver.Chrome()
- browser.get('http://www.baidu.com')
- browser.get('http://taobao.com')
- browser.get('https://www.cnblogs.com/')
- browser.back()
- time.sleep(3)
- browser.forward()
- time.sleep(3)
- browser.back()
- time.sleep(3)
- browser.close()
8.选项卡操作
- from selenium import webdriver
- import time
-
- 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://taobao.com') #在第二个选项卡打开淘宝页面
- time.sleep(3)
- browser.switch_to.window(browser.window_handles[0]) #切换到第一个选项卡
- browser.get('https://www.cnblogs.com/') #在第一个选项卡打开博客页面
更多资料见参考文档:https://selenium-python.readthedocs.io/api.html
9.案例--使用selenium抓取淘宝商品
目标:
利用selenium抓取淘宝商品并用pyquery解析得到商品的图片、名称、价格、购买人数、店铺名称和店铺所在地信息,并保存到MongoDB中。