Selenium webdriver的基本操作
常用类与方法回到顶部
from selenium import webdriver # 驱动浏览器
from selenium.webdriver import ActionChains # 鼠标的相关操作，比如滑动验证
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 # 等待事件，可以与EC连用
browser = webdriver.Chrome()
wait = WebDriverWait(browser, 10)
browser.get('https://www.baidu.com')
browser.maximize_window() # 窗口最大化
print(browser.current_url) # 获取当前页URL
print(browser.title) # 获取页面的title
print(browser.name) # 获取driver对象：chrome
print(browser.current_window_handle) # 获取当前窗口
print(browser.get_cookies()) # 获取cookies
print(browser.page_source) # 获取当前页面内容
browser.refresh() # 刷新
browser.save_screenshot('error.png') # 保存截图
browser.close() # 关闭当前窗口
browser.quit() # 退出浏览器驱动，关闭所有关联窗口
选择器回到顶部
from selenium import webdriver # 驱动浏览器
from selenium.webdriver.support.wait import WebDriverWait # 等待事件
browser = webdriver.Chrome()
wait = WebDriverWait(browser, 10)
browser.get('https://www.baidu.com')
browser.find_element_by_id('su') # id选择器
browser.find_element_by_class_name('xx') # 类选择器，1个
browser.find_elements_by_class_name('xx') # 类选择器，多个
browser.find_element_by_link_text('xxx') # 链接文本选择器
browser.find_element_by_xpath('xxxx') # xpath选择器
browser.find_element_by_tag_name('h1') # 标签选择器，获取1个
browser.find_elements_by_tag_name('h1') # 标签选择器，获取多个
browser.find_element_by_css_selector('xxx') # 样式选择器
在选择器的选用时，要记得获取一个还是多个，要区分开find_elements还是find_element。这个s容易忽略。
另外，上面的选择器，都可以写成下面的形式：
from selenium.webdriver.common.by import By # 选择器，以什么方式选择标签元素
browser.find_element(by=By.ID, value=None) # 选择器类型， value值
常用的选择器有：
By.ID
By.LINK_TEXT
By.CLASS_NAME
By.XPATH
By.CSS_SELECTOR
By.TAG_NAME
我们可以通过标签选择器来做：
import time
from selenium import webdriver # 驱动浏览器
from selenium.webdriver import ActionChains # 鼠标的相关操作，比如滑动验证
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()
wait = WebDriverWait(browser, 10)
browser.get('https://www.baidu.com')
input = browser.find_element_by_id('kw') # id选择器
input.send_keys('张开嘴') # 如果获取到的是个input框，就可以往里面填值
time.sleep(1)
input.clear() # 清空值
submit = browser.find_element(By.ID, 'su')
submit.click() # 点击事件
# 有的时候，click事件可能会有问题，我们使用键盘事件来处理
submit.send_keys(Keys.ENTER)
print(input.get_attribute('name')) # 获取name属性
print(input.text) # 获取内容，但这里的input没有text
print(input.tag_name)
print(input.get_property('name')) # 获取给定属性，用法类似get_attribute
print(browser.find_element_by_class_name('mnav').text) # 找个有text的
当获取了标签对象后，我们可以获取这个标签的属性、内容、css样式和子标签（子标签也支持这些操作），示例来自pythonav登录页：
import time
from selenium import webdriver
driver = webdriver.Chrome()
driver.implicitly_wait(10)
try:
driver.get('https://pythonav.com/login/')
<span class="hljs-comment"># 根据 class 属性获取 a 标签对象</span>
a_obj = driver.find_element_by_class_name(<span class="hljs-string">'navbar-brand'</span>)
<span class="hljs-comment"># 获取 a 标签的href属性</span>
print(a_obj.get_attribute(<span class="hljs-string">'href'</span>))
print(a_obj.text) <span class="hljs-comment"># 获取标签的文本内容</span>
print(a_obj.tag_name) <span class="hljs-comment"># 获取标签的名称</span>
<span class="hljs-comment"># 获取 css 样式</span>
print(a_obj.value_of_css_property(<span class="hljs-string">'padding'</span>))
print(a_obj.value_of_css_property(<span class="hljs-string">'font-size'</span>))
<span class="hljs-comment"># 如果这个标签有 submit 属性，那么就可以点击它了，比如我们点击form表单中有submit属性的标签</span>
driver.find_element_by_xpath(<span class="hljs-string">'//*[@id="fm"]/div[5]/div/input'</span>).submit()
<span class="hljs-comment"># 获取该标签下的子标签，或者子标签元素</span>
div_obj = driver.find_element_by_class_name(<span class="hljs-string">'col-xs-3'</span>)
print(div_obj.find_element_by_tag_name(<span class="hljs-string">'input'</span>))
print(div_obj.find_element_by_tag_name(<span class="hljs-string">'input'</span>).get_attribute(<span class="hljs-string">'value'</span>))
<span class="hljs-comment"># 获取验证码图片的大小</span>
print(driver.find_element_by_id(<span class="hljs-string">'image_code'</span>).size)
<span class="hljs-comment"># # 保存验证码图片</span>
driver.find_element_by_id(<span class="hljs-string">'image_code'</span>).screenshot(<span class="hljs-string">'./a.png'</span>)
except Exception as e:
print(e)
finally:
time.sleep(10)
driver.quit()
# 代码截止到2019/12/19号有效
需要注意的是：按钮提交时，除了click之外，还有一个submit，区别是click就是单纯的点一下。而submit是完成了表单提交，需要携带表单信息进行提交的。如果是button按钮的话，只能使用click，而不能使用submit。
Keys回到顶部
键盘事件有：
from selenium.webdriver.common.keys import Keys # 键盘相关
print(dir(Keys))
'''
[
'ADD', 'ALT', 'ARROW_DOWN', 'ARROW_LEFT', 'ARROW_RIGHT', 'ARROW_UP', 'BACKSPACE', 'BACK_SPACE', 'CANCEL', 'CLEAR', 'COMMAND', 'CONTROL', 'DECIMAL', 'DELETE', 'DIVIDE', 'DOWN', 'END', 'ENTER', 'EQUALS', 'ESCAPE', 'F1', 'F10', 'F11', 'F12', 'F2', 'F3', 'F4', 'F5', 'F6', 'F7', 'F8', 'F9', 'HELP', 'HOME', 'INSERT', 'LEFT', 'LEFT_ALT', 'LEFT_CONTROL', 'LEFT_SHIFT', 'META', 'MULTIPLY', 'NULL', 'NUMPAD0', 'NUMPAD1', 'NUMPAD2', 'NUMPAD3', 'NUMPAD4', 'NUMPAD5', 'NUMPAD6', 'NUMPAD7', 'NUMPAD8', 'NUMPAD9', 'PAGE_DOWN', 'PAGE_UP', 'PAUSE', 'RETURN', 'RIGHT', 'SEMICOLON', 'SEPARATOR', 'SHIFT', 'SPACE', 'SUBTRACT', 'TAB', 'UP'
]
'''
来个键盘事件示例：
import time
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
driver = webdriver.Chrome()
try:
driver.get(<span class="hljs-string">'https://pythonav.com/login/'</span>)
input_obj = driver.find_element_by_id(<span class="hljs-string">'id_username'</span>)
time.sleep(<span class="hljs-number">2</span>)
input_obj.send_keys(<span class="hljs-string">'selenium is cool'</span>)
time.sleep(<span class="hljs-number">2</span>)
input_obj.send_keys(Keys.CONTROL, <span class="hljs-string">'a'</span>)
time.sleep(<span class="hljs-number">2</span>)
input_obj.send_keys(Keys.DELETE)
time.sleep(<span class="hljs-number">3</span>)
input_obj.send_keys(<span class="hljs-string">'no 。。。。。'</span>)
time.sleep(<span class="hljs-number">3</span>)
pwd_obj = driver.find_element_by_id(<span class="hljs-string">'id_password'</span>)
pwd_obj.send_keys(<span class="hljs-string">'selenium is very cool'</span>)
pwd = pwd_obj.get_attribute(<span class="hljs-string">'value'</span>)
print(pwd)
js = <span class="hljs-string">"return $('#id_password').val()"</span>
res = driver.execute_script(js)
print(res)
js2 = <span class="hljs-string">"alert('{}')"</span>.format(res)
driver.execute_script(js2)
finally:
time.sleep(10)
driver.quit()
鼠标相关事件回到顶部
from selenium.webdriver import ActionChains # 鼠标的相关操作，比如滑动验证
print(dir(ActionChains))
'''
[
'click', 'click_and_hold', 'context_click', 'double_click', 'drag_and_drop', 'drag_and_drop_by_offset', 'key_down', 'key_up', 'move_by_offset', 'move_to_element', 'move_to_element_with_offset', 'pause', 'perform', 'release', 'reset_actions', 'send_keys', 'send_keys_to_element'
]
'''
示例：
import time
from selenium import webdriver
from selenium.webdriver.common.keys import Keys # 键盘事件
from selenium.webdriver import ActionChains # 鼠标事件
driver = webdriver.Chrome()
driver.get('https://www.baidu.com')
driver.find_element_by_id('kw').send_keys('听雨危楼')
driver.find_element_by_id('su').click()
time.sleep(2)
ActionChains(driver).click(driver.find_element_by_partial_link_text('- 博客园')).perform()
time.sleep(10)
driver.quit()
EC回到顶部
再来看EC：
from selenium.webdriver.support import expected_conditions as EC # 各种判断，一般跟等待事件连用，比如说等待某个元素加载出来
print(dir(EC))
'''
['NoAlertPresentException', 'NoSuchElementException', 'NoSuchFrameException', 'StaleElementReferenceException', 'WebDriverException', 'WebElement', '__builtins__', '__cached__', '__doc__', '__file__', '__loader__', '__name__', '__package__', '__spec__', '_element_if_visible', '_find_element', '_find_elements', 'alert_is_present', 'element_located_selection_state_to_be', 'element_located_to_be_selected', 'element_selection_state_to_be', 'element_to_be_clickable', 'element_to_be_selected', 'frame_to_be_available_and_switch_to_it', 'invisibility_of_element', 'invisibility_of_element_located', 'new_window_is_opened', 'number_of_windows_to_be', 'presence_of_all_elements_located', 'presence_of_element_located', 'staleness_of', 'text_to_be_present_in_element', 'text_to_be_present_in_element_value', 'title_contains', 'title_is', 'url_changes', 'url_contains', 'url_matches', 'url_to_be', 'visibility_of', 'visibility_of_all_elements_located', 'visibility_of_any_elements_located', 'visibility_of_element_located']
'''
expected_conditions中封装了各种判断，一般常应用于断言中和等待机制搭配使用。比如判断某个元素是否存在，存在则返回该标签对象，不存在则报报错，并且可以自定制报错：
import time
from selenium import webdriver # 驱动浏览器
from selenium.webdriver import ActionChains # 鼠标的相关操作，比如滑动验证
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.expected_conditions import NoSuchElementException
from selenium.webdriver.support.wait import WebDriverWait # 等待事件
browser = webdriver.Chrome()
# wait = WebDriverWait(browser, 10)
#
browser.get('https://www.baidu.com')
try:
print(EC.presence_of_element_located((By.ID, 'xxx'))(browser)) # 判断某个元素是否被加到了dom树里，并不代表该元素一定可见
except NoSuchElementException as e:
print('找不到这个标签')
如上例，当前页面中没有id名为xxx的，所以报错，需要注意的是，报错也是一个对象，要想得到False结果，需要执行这个对象并且将浏览器驱动对象传进去。
其他的判断：
import time
from selenium import webdriver # 驱动浏览器
from selenium.webdriver import ActionChains # 鼠标的相关操作，比如滑动验证
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()
wait = WebDriverWait(browser, 10)
#
browser.get('https://www.baidu.com')
print(EC.title_contains('XX')(browser)) # 判断当前页面的title是否包含预期字符串
print(EC.visibility_of_element_located((By.ID, 'su'))(browser)) # 判断某个元素是否可见（代表元素非隐藏，元素的宽和高都不等于0）
print(EC.invisibility_of_element_located(((By.ID, 'xxx')))(browser)) # 判断某个元素中是否不存在于dom树或不可见
print(EC.text_to_be_present_in_element((By.ID, 'su'), '百度一下')(browser)) # 判断某个元素中的text是否包含了预期的字符串， 示例中的标签没有text值
print(EC.text_to_be_present_in_element_value((By.ID, 'su'), '百度一下')) # 判断某个元素中的value属性是否包含了预期的字符串
print(EC.alert_is_present()(browser)) # 判断页面上是否存在alert
print(EC.element_to_be_clickable((By.ID, 'su'))(browser)) # 判断某个元素中是否可见并且是enable的，这样的话才叫clickable
其他案例回到顶部
浏览器的前进后退和刷新回到顶部
我们可以使用selenium自动的模拟前进（forward）后退（back）和刷新（refresh）三个动作：
import time
from selenium import webdriver # 驱动浏览器
browser = webdriver.Chrome()
browser.get('https://www.baidu.com')
browser.get('https://www.taobao.com')
browser.get('http://www.sina.com.cn/')
time.sleep(5)
browser.back() # 后退
time.sleep(3)
browser.forward() # 前进
time.sleep(2)
browser.refresh() # 刷新
browser.close() # 关闭当前窗口
time.sleep(3)
browser.quit()
执行js代码回到顶部
在有些时候，需要执行一些js代码来完成任务。
from selenium import webdriver # 驱动浏览器
browser = webdriver.Chrome()
browser.get('https://www.zhihu.com/explore')
browser.execute_script('alert("xxoo")')