Selenium Web自动化之键盘操作，文件上传，Cookie，调用 js，错误截图

一、文件上传

 

from selenium import webdriver
import os
 
 
driver = webdriver.Chrome('../tools/chromedriver.exe')
file_path = "file:///" + os.path.abspath('upfile.html')
driver.get(file_path)
 
# 定位上传按钮的位置
driver.find_element_by_name('file').send_keys(os.path.abspath('upfile.txt'))
driver.quit()

 

二、Cookie操作

 

网站为了辨别用户身份、进行 session 跟踪而存储在用户本地终端上的数据，也可以叫做浏览器缓存。webdriver 对 cookie 的常用操作有添加、删除、读取。

 

(1) get_cookies()-----获得所有的 cookie 信息
(2) get_cookie(name)-----活得 key 值为 name 的 cookie 的信息
(3) add_cookie(cookie_dict)----添加 cookie。"cookie_dict" 指字典对象，必须有 name 和 value 值
(4) delete_cookie(name,optionsString)：删除cookie信息。“name”是要删除的cookie的名称，“optionsString”是该cookie的选项，目前支持的选项包括“路径”，“域”
(5) delete_all_cookies()----删除所有 cookie 信息

 

 

from selenium import webdriver
from time import sleep
 
 
base_url = 'https://www.baidu.com/'
browser = webdriver.Chrome('../tools/chromedriver.exe')
browser.get(base_url)
 
# 1. 获取 cookie 信息
cookies = browser.get_cookies()
print(cookies)
sleep(2)
browser.quit()
 
# 2. cookie 写入
browser.add_cookie(
    {
        'name': 'add-cookie',
        'value': 'add-cookie-value'
    }
)
# 遍历cookies打印cookie信息
for cookie in browser.get_cookies():
    print("%s ---> %s" % (cookie['name'], cookie['value']))
sleep(2)
browser.quit()

 

三、调用 JavaScript

 

对于 webdriver 中无法操作的动作（例如：滚动浏览器的侧边栏），可以调用 webdriver 进行浏览器的控制。webdriver 提供了execute_script()方法来执行 JavaScript 代码。

 

from selenium import webdriver
from time import sleep
 
 
base_url = 'https://www.baidu.com'
browser = webdriver.Chrome('../tools/chromedriver.exe')
browser.get(base_url)
 
# window.scrollTo()方法用于设置浏览器窗口滚动条的水平和垂直位置。方法的第一个参数表示水平的左间距，第二个参数表示垂直的上边距。
browser.set_window_size(500, 500)
browser.find_element_by_id('kw').send_keys('百度')
browser.find_element_by_id('su').click()
sleep(2)
 
# 通过javascript设置浏览器窗口的滚动条位置
js = "window.scrollTo(100, 450);"
browser.execute_script(js)
sleep(2)
 
browser.quit()

 

四、窗口截图

自动化用例是由程序去执行的，因此有时候打印的错误信息并不十分明确。如果在脚本执行出错的时候能对当前窗口截图保存，那么通过图片就可以非常直观地看出出错的原因。webdriver 提供了截图函数 get_screenshot_as_file() 来截取当前窗口。

 

from selenium import webdriver
from time import sleep
 
base_url = 'http://www.baidu.com/'
browser = webdriver.Chrome('../tools/chromedriver.exe')
 
browser.get(base_url)
 
browser.find_element_by_id('kw').send_keys('python selenium')
browser.find_element_by_id('su').click()
sleep(2)
 
# 截取当前窗口并指定报错截图的位置
# browser.get_screenshot_as_file('ScreenShot/14_screenShot.jpg')
browser.get_screenshot_as_file('ScreenShot/14_screenShot.png')
 
browser.quit()

 五、键盘操作

from selenium import webdriver
import time
from selenium.webdriver.common.keys import Keys
driver = webdriver.Chrome()
driver.get('www.baidu.com')                        #打开指定url

element = driver.find_element_by_id()
element.send_keys('admin123')
time.sleep(1)
element.send_keys(Keys.BACK_SPACE)                 #删除键
element.send_keys(Keys.SPACE)                      #空格键
element.send_keys(Keys.TAB)                        #制表键
element.send_keys(Keys.ESCAPE)                     #回退键
element.send_keys(Keys.ENTER)                      #回车键
element.send_keys(Keys.CONTROL,'a')                #全选
element.send_keys(Keys.CONTROL,'c')                #复制
time.sleep(1)
driver.quit()