Python+Selenium自动化学习笔记

1、环境搭建

1.1、安装selenium模块文件

pip install selenium

1.2、安装ChromeDriver驱动

【1】下载安装Chrome浏览器。

【2】查看Chrome版本。

【4】环境变量配置。

  • 在Windows下,下载完成后,将ChromeDriver的可执行文件chromedriver.exe文件拖到Python的Scripts目录下。

1.3、安装GeckoDriver驱动(Firefox浏览器)

【1】下载GeckoDriver驱动。

【2】环境变量配置

  • 在Windows下,直接将geckodriver.exe文件拖到Python的Scripts目录下。

2、基本使用方法

2.1、初始化浏览器驱动

# 0、导入模块文件
from selenium import webdriver

# 1、初始化浏览器为chrome浏览器
browser = webdriver.Chrome()

# 2、初始化浏览器为firefox浏览器
browser = webdriver.Firefox()

# 3、初始化浏览器为无界面的chrome浏览器
option = webdriver.ChromeOptions()
option.add_argument("headless")
driver = webdriver.Chrome(chrome_options=option)

2.2、设置浏览器大小

  • maximize_window():设置浏览器大小为全屏
  • set_window_size(500,500):设置浏览器分辨率为:500×500
from selenium import webdriver
import time  

browser = webdriver.Chrome()
browser.maximize_window()   #设置浏览器大小:全屏
browser.get('https://www.baidu.com')  
time.sleep(2)

browser.set_window_size(500,500)  # 分辨率 1280*800  
time.sleep(2)

browser.set_window_size(1024,768)  # 分辨率 1024*768  
time.sleep(2)

2.3、刷新浏览器页面

  • refresh()方法:刷新浏览器页面
from selenium import webdriver
import time  

browser = webdriver.Chrome()
browser.maximize_window()   #设置浏览器大小:全屏
browser.get('https://www.baidu.com')  
time.sleep(1)

try:
    browser.refresh()  #刷新页面
    print('刷新页面')
except Exception as e:
    print('test fail')

#输出内容:刷新页面

2.4、浏览器后退和前进

  • back():后退
  • forward():前进
from selenium import webdriver
import time  

browser = webdriver.Chrome()
browser.maximize_window()   #设置浏览器大小:全屏
browser.get('https://www.baidu.com')  

#点击打开百度新闻页面
browser.find_element_by_link_text('新闻').click()
time.sleep(2)

browser.back()  #后退:回到百度首页
time.sleep(2)

browser.forward()  #前进:前进到百度新闻页面
time.sleep(2)

3、获取页面信息

3.1、页面title

  • title:获取当前页面的标题显示的字段
from selenium import webdriver
import time  

browser = webdriver.Chrome()
browser.get('https://www.baidu.com')  

#打印网页标题
print(browser.title)
#输出内容:百度一下,你就知道

3.2、页面URL

  • current_url:获取当前页面的URL
from selenium import webdriver
import time  

browser = webdriver.Chrome()
browser.get('https://www.baidu.com')  

#打印网页标题
print(browser.current_url)
#输出内容:https://www.baidu.com/

3.3、浏览器版本号

  • capabilities['version']):打印浏览器version的值
from selenium import webdriver
import time  

browser = webdriver.Chrome()
browser.get('https://www.baidu.com')  

#打印网页标题
print(browser.capabilities['version'])
#输出内容:67.0.3396.87

3.4、元素尺寸

  • size:返回元素的尺寸
from selenium import webdriver
import time  

browser = webdriver.Chrome()
browser.get('https://www.baidu.com')  

#定位输入框
input_box = browser.find_element_by_id('kw')
#打印输入框尺寸
print(input_box.size)
#输出内容:{'height': 22, 'width': 500}

3.5、元素的文本

  • text:返回元素的文本信息
from selenium import webdriver
import time  

browser = webdriver.Chrome()
browser.get('https://www.baidu.com')  

#定位备案元素
recordcode = browser.find_element_by_id('jgwab')
#打印备案元素信息
print(recordcode.text)
#输出内容:京公网安备11000002000001号

3.6、元素属性值

  • get_attribute('')方法
  • get_attribute('href'):获取href属性值
  • get_attribute('id'):获取id属性值
# coding=utf-8
import time
from selenium import webdriver


driver = webdriver.Chrome()
driver.maximize_window()
driver.implicitly_wait(6)
driver.get("https://www.baidu.com")
time.sleep(1)

for link in driver.find_elements_by_xpath("//*[@href]"):
    print (link.get_attribute('href'))
driver.quit()

4、定位页面元素

4.1、ID定位元素

from selenium import webdriver
# 设置浏览器
browser = webdriver.Chrome()
#设置浏览器大小:全屏
browser.maximize_window()
#打开百度首页
browser.get('https://www.baidu.com/')
#定位百度搜索输入框之前,先分析下它的html结构
#<input type="text" class="s_ipt nobg_s_fm_hover" name="wd" id="kw" maxlength="100" autocomplete="off">
#发现它的 id="kw" ,接下来我们就通过id进行定位
try:
    browser.find_element_by_id('kw').send_keys('哈哈')
    print('test post:id')
except Exception as e:
    print('test fail')

#输出内容:test post:id

4.2、name定位元素

from selenium import webdriver
browser = webdriver.Chrome()
browser.maximize_window()

#打开百度首页
browser.get('https://www.baidu.com/')
#搜索框的html结构:<input type="text" class="s_ipt nobg_s_fm_hover" name="wd" id="kw" maxlength="100" autocomplete="off">
# 根据name属性定位
try:
    browser.find_element_by_name('wd').send_keys('哈哈')
    print('test post:name')
except Exception as e:
    print('test fail')

#输出内容:test post:name

4.3、class定位元素

from selenium import webdriver
browser = webdriver.Chrome()
browser.maximize_window()

#打开百度首页
browser.get('https://www.baidu.com/')
#搜索框的html结构:<input type="text" class="s_ipt nobg_s_fm_hover" name="wd" id="kw" maxlength="100" autocomplete="off">
# 根据class_name属性定位
try:
    browser.find_element_by_class_name('s_ipt').send_keys('哈哈')
    print('test post:class_name')
except Exception as e:
    print('test fail')

#输出内容:test post:class_name

4.4、tag_name定位元素

from selenium import webdriver
browser = webdriver.Chrome()
browser.maximize_window()

#打开百度首页
browser.get('https://www.baidu.com/')
#搜索框的html结构:<input type="text" class="s_ipt nobg_s_fm_hover" name="wd" id="kw" maxlength="100" autocomplete="off">
# 根据tag_name属性定位
try:
    browser.find_element_by_tag_name('form')
    print('test post:tag_name')
except Exception as e:
    print('test fail')

#输出内容:test post:tag_name
  • link_text:根据跳转链接上面的文字来定位元素。
from selenium import webdriver
browser = webdriver.Chrome()
browser.maximize_window()

#打开百度首页
browser.get('https://www.baidu.com/')
# 根据link_text属性定位元素“新闻”,然后点击按钮
try:
    browser.find_element_by_link_text('新闻').click()
    print('test post:tag_name')
except Exception as e:
    print('test fail')

#输出内容:test post:link_text
  • 和link_text定位元素差不多,partial_link_text是通过文字信息中的部分字段来定位元素。
from selenium import webdriver
browser = webdriver.Chrome()
browser.maximize_window()

#打开百度首页
browser.get('https://www.baidu.com/')
# 根据partial_link_text属性定位元素“新闻”,然后点击按钮
try:
    browser.find_element_by_partial_link_text('闻').click()
    print('test post:tag_name')
except Exception as e:
    print('test fail')

#输出内容:test post:partial_link_text

4.7、xpath定位元素

from selenium import webdriver
browser = webdriver.Chrome()
browser.maximize_window()

#打开百度首页
browser.get('https://www.baidu.com/')
# 根据xpath定位元素
try:
    browser.find_element_by_xpath('//*[@id="kw"]').send_keys('哈哈')
    print('test post:xpath')
except Exception as e:
    print('test fail')

#输出内容:test post:xpath

4.8、CSS定位页面元素

from selenium import webdriver
browser = webdriver.Chrome()
browser.maximize_window()

#打开百度首页
browser.get('https://www.baidu.com/')
# 根据css_selector定位元素
try:
    browser.find_element_by_css_selector('#kw').send_keys('哈哈')
    print('test post:xpath')
except Exception as e:
    print('test fail')

#输出内容:test post:css_selector

4.9、By定位

  • 除了使用上面的方法外,还可以利用find_element()方法,通过By来定位元素。
  • 使用之前需要导入By类:
#导入By类
from selenium.webdriver.common.by import By
  • 那么上面的方法还可以改写为:
browser.find_element(By.ID,'kw')
browser.find_element(By.NAME,'wd')
browser.find_element(By.CLASS_NAME,'s_ipt')
browser.find_element(By.TAG_NAME,'form')
browser.find_element(By.LINK_TEXT,'新闻')
browser.find_element(By.PARTIAL_LINK_TEXT,'闻')
browser.find_element(By.XPATH,'//*[@id="kw"]')
browser.find_element(By.CSS_SELECTOR,'#kw')

4.10、定位一组元素

  • 定位一组元素和定位某个元素差不多,唯一区别就是在单词element后面增加了一个s表示复数:
find_elements_by_id('')
find_elements_by_id_by_class_name('')
find_elements_by_tag_name('')
find_elements_by_link_text('')
find_elements_by_partial_link_text('')
find_elements_by_xpath('')
find_elements_by_css_selector('')

5、页面操作

5.1、输入内容:send_keys()

#定位输入框
input_box = browser.find_element_by_id('kw')
#输入内容
input_box.send_keys('关键字')

5.2、点击按钮:click()

#定位搜索按钮
button = browser.find_element_by_id('su')
button.click()

5.3、清空输入框:clear()

#定位输入框
input_box = browser.find_element_by_id('kw')
#清空输入框
input_box.clear()

5.4、模拟'回车'操作:submit()

  • submit():通常应用于提交表单;例如:搜索框输入内容后的回车操作。
from selenium import webdriver

browser = webdriver.Chrome()
browser.maximize_window()
browser.get('https://www.baidu.com')

# 定位输入框
input_box = browser.find_element_by_id('kw')
# 输入关键词
input_box.send_keys('关键字')
# 模拟回车操作
input_box.submit()

5.5、单选框选中

from selenium import webdriver
import time
driver = webdriver.Chrome()
driver.maximize_window()
driver.get('http://news.baidu.com/')
time.sleep(1)

try:
    #定位搜索框下面的新闻全文和新闻标题单选项
    radio_button = driver.find_elements_by_css_selector(".search-radios>input")
    # 定位的一组元素后,返回的是一个列表,这里就循环选中对应的选项
    for i in radio_button:
        i.click()
    print ('选中单选框.')
except Exception as e:
    print ('fail',format(e))

5.5、复选框选中

from selenium import webdriver
import time
driver = webdriver.Chrome()
driver.maximize_window()
driver.get('https://passport.baidu.com/v2/?reg&tt=1529850358050&overseas=undefined&gid=02B2650-6D11-430E-A19D-529F69B9F1CE&tpl=xw&u=http%3A%2F%2Fnews.baidu.com%2F')
time.sleep(1)

try:
    #定位 阅读并接受《百度用户协议》及《百度隐私权保护声明》 的多选框
    checkbox = driver.find_elements_by_css_selector("#TANGRAM__PSP_3__isAgree")
    # 定位的一组元素后,返回的是一个列表,这里就循环选中对应的选项
    for i in checkbox:
        i.click()
    print ('选中多选框')
except Exception as e:
    print ('fail',format(e))

5.6、操作下拉框:Select模块(index)

  • selenium提供了Select模块,可以直接根据属性或索引定位。

【1】导入select方法:

from selenium.webdriver.support.select import Select

【2】Select模块中的定位方法。


'''1、三种选择某一选项项的方法'''

select_by_index()  			# 通过索引定位;注意:index索引是从“0”开始。
select_by_value()  			# 通过value值定位,value标签的属性值。
select_by_visible_text() 	# 通过文本值定位,即显示在下拉框的值。

'''2、三种返回options信息的方法'''

options                  	# 返回select元素所有的options
all_selected_options     	# 返回select元素中所有已选中的选项
first_selected_options   	# 返回select元素中选中的第一个选项                  


'''3、四种取消选中项的方法'''

deselect_all             	# 取消全部的已选择项
deselect_by_index        	# 取消已选中的索引项
deselect_by_value        	# 取消已选中的value值
deselect_by_visible_text 	# 取消已选中的文本值

【3】使用实例

'''这个是我百度首页搜索设置里面的一个下拉框html代码'''
html = '''
<select name="NR" id="nr">
    <option value="10" selected="">每页显示10条</option>
    <option value="20">每页显示20条</option>
    <option value="50">每页显示50条</option>
</select>
'''

# coding=utf-8
from selenium import webdriver
from selenium.webdriver.support.select import Select

# 初始化
driver = webdriver.Chrome()
......

'''如果我们需要选择:50条,那么可以用下面的三种方法
    先选中输入框元素,然后在选择对应选项'''
# 根据索引选择
Select(driver.find_element_by_name("NR")).select_by_index("2")
# 根据value值选择
Select(driver.find_element_by_name("NR")).select_by_value("50")
# 根据文本值选择
Select(driver.find_element_by_name("NR")).select_by_visible_text("每页显示50条")

driver.quit()

5.7、多个浏览器窗口切换

切换浏览器窗口的时候将会用到的三个方法:

  • current_window_handle:获取当前窗口的句柄。
  • window_handles:返回当前浏览器的所有窗口的句柄。
  • switch_to_window():用于切换到对应的窗口。
from selenium import webdriver
from selenium.webdriver.common.keys import Keys

driver = webdriver.Chrome()
driver.maximize_window()
driver.implicitly_wait(10)
driver.get("http://www.baidu.com/")

# 获取当前窗口句柄
seacher_windows = driver.current_window_handle

# 跳转新页面
driver.find_element_by_xpath('//*[@id="block1_server"]/div[2]/ul/li[1]/div/div[1]/a[3]').click()

all_handles = driver.window_handles
#回到首页
for handle in all_handles:
	if handle != seacher_windows:
		driver.switch_to.window(seacher_windows)

6、模拟鼠标操作:ActionChains类

【1】使用之前需要引入 ActionChains 类。

from selenium.webdriver.common.action_chains import ActionChains

【2】ActionChains类提供的鼠标常用方法:

  • perform():执行所有 ActionChains 中存储的行为。
  • context_click():右击
  • double_click():双击
  • drag_and_drop():拖到
  • move_to_element():鼠标悬停

6.1、鼠标右击实例

from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains  # 引入 ActionChains 类

browser = webdriver.Chrome()
browser.get('https://www.baidu.com')  


# 定位到要右击的元素
right_click = browser.find_element_by_link_text('新闻')

# 对定位到的元素执行鼠标右键操作
#ActionChains(driver):调用ActionChains()类,并将浏览器驱动browser作为参数传入
#context_click(right_click):模拟鼠标双击,需要传入指定元素定位作为参数
#perform():执行ActionChains()中储存的所有操作,可以看做是执行之前一系列的操作
try:
    ActionChains(browser).context_click(right_click).perform()
    print('成功右击')
except Exception as e:
    print('fail')
#输出内容:成功双击

注意:

  • ActionChains(driver):调用ActionChains()类,并将浏览器驱动browser作为参数传入
  • context_click(right_click):模拟鼠标双击,需要传入指定元素定位作为参数
  • perform():执行ActionChains()中储存的所有操作,可以看做是执行之前一系列的操作

6.2、鼠标右击:context_click()

#   鼠标右击
# 定位到要右击的元素
right_click  = browser.find_element_by_id("xx")

# 对定位到的元素执行右击操作
ActionChains(browser).move_to_element(right_click ).perform()

6.3、鼠标双击:double_click()

# 定位到要右击的元素
double_click = browser.find_element_by_id('xx')

# 对定位到的元素执行鼠标右键操作
ActionChains(browser).context_click(double_click).perform()

6.4、鼠标拖动:drag_and_drop(source,target)

  • source:开始位置;需要拖动的元素
  • target:结束位置;拖到后需要放置的目的地元素
# 开始位置:定位到元素的原位置
source = driver.find_element_by_id("xx")

# 结束位置:定位到元素要移动到的目标位置
target = driver.find_element_by_id("xx")

# 执行元素的拖放操作
ActionChains(driver).drag_and_drop(source,target).perform()

6.5、鼠标悬停:move_to_element()

# 定位到要悬停的元素
move = driver.find_element_by_id("xx")

# 对定位到的元素执行悬停操作
ActionChains(driver).move_to_element(move).perform()

7、模拟键盘操作

  • selenium中的Keys()类提供了大部分的键盘操作方法;通过send_keys()方法来模拟键盘上的按键。

【1】导入键盘类 Keys()

from selenium.webdriver.common.keys import Keys

7.1、常用的键盘操作:

  • send_keys(Keys.BACK_SPACE):删除键(BackSpace)
  • send_keys(Keys.SPACE):空格键(Space)
  • send_keys(Keys.TAB):制表键(TAB)
  • send_keys(Keys.ESCAPE):回退键(ESCAPE)
  • send_keys(Keys.ENTER):回车键(ENTER)
  • send_keys(Keys.CONTRL,'a'):全选(Ctrl+A)
  • send_keys(Keys.CONTRL,'c'):复制(Ctrl+C)
  • send_keys(Keys.CONTRL,'x'):剪切(Ctrl+X)
  • send_keys(Keys.CONTRL,'v'):粘贴(Ctrl+V)
  • send_keys(Keys.F1):键盘F1
  • .....
  • send_keys(Keys.F12):键盘F12

7.2、键盘操作实例

from selenium import webdriver  
from selenium.webdriver.common.keys import Keys  

driver=webdriver.Chrome()  
driver.get("http://www.baidu.com")  
#输入关键词内容
driver.find_element_by_id("kw").send_keys("selenium")  
#删除键
driver.find_element_by_id("kw").send_keys(Keys.BACK_SPACE)  
#空格键 
driver.find_element_by_id("kw").send_keys(Keys.SPACE)  
#输入内容
driver.find_element_by_id("kw").send_keys("教程")  
#全选(Ctrl+A)  
driver.find_element_by_id("kw").send_keys(Keys.CONTROL,'a') 
#剪切(Ctrl+X)  
driver.find_element_by_id("kw").send_keys(Keys.CONTROL,'x') 
#粘贴(Ctrl+V)  
driver.find_element_by_id("kw").send_keys(Keys.CONTROL,'v')  
#回车键  
driver.find_element_by_id("kw").send_keys(Keys.ENTER)  

8、设置页面等待加载条件

如果遇到使用ajax加载的网页,页面元素可能不是同时加载出来的,这个时候,就需要我们通过设置一个等待条件,等待页面元素加载完成,避免出现因为元素未加载导致的错误的出现。

WebDriver提供了两种等待类型:显示等待、隐式等待。

8.1、显示等待:WebDriverWait()类+ExpectedCondition类

  • 显示等待:设置一个等待时间和一个条件,在规定时间内,每隔一段时间查看下条件是否成立,如果成立那么程序就继续执行,否则就提示一个超时异常(TimeoutException)。
  • 通常情况下WebDriverWait类会结合ExpectedCondition类一起使用。

【实例】

from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

driver = webdriver.Chrome()
driver.get('https://www.baidu.com')
# 设置浏览器:driver  等待时间:20s
wait = WebDriverWait(driver, 20)
# 设置判断条件:等待id='kw'的元素加载完成
input_box = wait.until(EC.presence_of_element_located((By.ID, 'kw')))
# 在关键词输入:关键词
input_box.send_keys('关键词')

8.1.1、WebDriverWait的具体参数和方法:

WebDriverWait(driver,timeout,poll_frequency=0.5,ignored_exceptions=None)
	driver: 浏览器驱动
	timeout: 超时时间,等待的最长时间(同时要考虑隐性等待时间)
	poll_frequency: 每次检测的间隔时间,默认是0.5秒
	ignored_exceptions:超时后的异常信息,默认情况下抛出NoSuchElementException异常


until(method,message='')
    method: 在等待期间,每隔一段时间调用这个传入的方法,直到返回值不是False
    message: 如果超时,抛出TimeoutException,将message传入异常


until_not(method,message='')
	until_not 与until相反,until是当某元素出现或什么条件成立则继续执行,
    until_not是当某元素消失或什么条件不成立则继续执行,参数也相同。
    method
    message

8.1.2、设置页面加载判断条件:ExpectedCondition

  • ExpectedCondition中可使用的判断条件:
from selenium.webdriver.support import expected_conditions as EC


# 判断标题是否和预期的一致
title_is
# 判断标题中是否包含预期的字符串
title_contains


# 判断指定元素是否加载出来
presence_of_element_located
# 判断所有元素是否加载完成
presence_of_all_elements_located


# 判断某个元素是否可见. 可见代表元素非隐藏,并且元素的宽和高都不等于0,传入参数是元组类型的locator
visibility_of_element_located
# 判断元素是否可见,传入参数是定位后的元素WebElement
visibility_of
# 判断某个元素是否不可见,或是否不存在于DOM树
invisibility_of_element_located


# 判断元素的 text 是否包含预期字符串
text_to_be_present_in_element
# 判断元素的 value 是否包含预期字符串
text_to_be_present_in_element_value


#判断frame是否可切入,可传入locator元组或者直接传入定位方式:id、name、index或WebElement
frame_to_be_available_and_switch_to_it


#判断是否有alert出现
alert_is_present


#判断元素是否可点击
element_to_be_clickable


# 判断元素是否被选中,一般用在下拉列表,传入WebElement对象
element_to_be_selected
# 判断元素是否被选中
element_located_to_be_selected
# 判断元素的选中状态是否和预期一致,传入参数:定位后的元素,相等返回True,否则返回False
element_selection_state_to_be
# 判断元素的选中状态是否和预期一致,传入参数:元素的定位,相等返回True,否则返回False
element_located_selection_state_to_be


#判断一个元素是否仍在DOM中,传入WebElement对象,可以判断页面是否刷新了
staleness_of

【调用方法如下】

WebDriverWait(driver, 超时时长, 调用频率, 忽略异常).until(可执行方法, 超时时返回的信息)

8.2、隐式等待:implicitly_wait()

  • implicitly_wait(xx):设置等待时间为xx秒,等待元素加载完成,如果到了时间元素没有加载出,就抛出一个NoSuchElementException的错误。
  • 注意:隐性等待对整个driver的周期都起作用,所以只要设置一次即可。
from selenium import webdriver

driver = webdriver.Chrome()
driver.implicitly_wait(30)  # 隐性等待,最长等30秒

driver.get('https://www.baidu.com')

print(driver.current_url)
print(driver.title)

8.3、强制等待:sleep()

  • 强制等待:不管浏览器元素是否加载完成,程序都得等待3秒,3秒一到,继续执行下面的代码。
from selenium import webdriver
from time import sleep

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

sleep(3)  # 强制等待3秒再执行下一步

print(driver.title)
原文地址:https://www.cnblogs.com/jasontang369/p/9616208.html