Selenium快速入门之1：最常使用方法整理

Selenium最早是为了测试浏览器、网页的，但目前也被广泛应用于网络爬虫。本文整理Selenium最常见的使用方法。

一、创建webdriver，打开特定页面  

#打开浏览器、百度网站 driver = webdriver.Chrome() driver.get("https://baidu.com")

二、设置浏览器大小、设置页面大小  

driver = webdriver.Chrome() driver.get("https://baidu.com")

三、页面前进/后退、刷新、保存截屏

#后退 driver.back() #前进 driver.forward() #刷新 driver.refresh() #保存当前页面截屏 driver.save_screenshot("save_1.png")

四、关闭页面、退出浏览器

driver.close() driver.quit()

五、等待

#强制等待 2 秒 time.sleep(2) #隐性等待，最长等 30 秒 dr.implicitly_wait(30) #显式等待，每隔 0.5 秒进行一次检查，最长等 20 秒 WebDriverWait(dr, 20, 0.5).until(EC.presence_of_element_located((By.LINK_TEXT,u'首 页')))

六、打开多个页面

# 打开浏览器 driver=webdriver.Chrome() driver.get("http://baidu.com") # 停留两秒后打开搜狗搜索 driver=webdriver.Chrome() driver.get("http://baidu.com") #停留两秒后打开有道翻译 time.sleep(2) time.sleep(2) JS2='window.open("https://fanyi.youdao.com/");' driver.execute_script(JS2)

七、使用JS下滑到页面底部

driver=webdriver.Chrome() driver.get("https://top.baidu.com//") JS="window.scrollTo(10000,document.body.scrollHeight)" driver.execute_script(JS)

八、Cookie处理

#获取所有cookie cookies=driver.get_cookies() #获取key为BAIDUID的cookie cookie=driver.get_cookie("BAIDUID") #删除所有的cookie driver.delete_all_cookies()

九、元素定位

#根据ID定位 driver.find_element_by_id("kw") #根据NAME定位 driver.find_element_by_name("wd") #根据Xpath定位 driver.find_element_by_xpath(r'//*[@id="kw"]') #根据CSS选择器定位 driver.find_element_by_css_selector("#kw") #根据Link Text定位 driver.find_element_by_link_text("新闻")

关于xpath、CSS的使用，可参见文章：

解析html：Xpath快速了解使用、解析html：CSS Selector快速了解使用

十、文本框输入，按钮点击

driver.get("http://cn.bing.com/") keywords = 'SELENIUM爬虫' # id="sb_form_q"是 bing 搜索输入框，输入字符串"SELENIUM 自动化测试" driver.find_element_by_id("sb_form_q").send_keys(keywords) #点击搜索按钮 driver.find_element_by_id("sb_form_go").click()

十一、不同frame之间转换

#找到特定的frame并切换 Frame_name= driver.find_element_by_id(【Frame_name】) driver.switch_to_frame(Frame_name) #回到默认的frame driver.switch_to_default_content()

十二、使用JS修改页面标题、进行弹窗

#启动浏览器，打开页面 driver=webdriver.Chrome() driver.get("https://www.baidu.com/") #修改页面标题 time.sleep(5) JS1="document.title='xxxxxx';" driver.execute_script(JS1) #弹窗 time.sleep(5) JS2=r"alert($(document).attr('title'));" driver.execute_script(JS2) #退出 time.sleep(5) driver.quit()