python3+selenium实现Web自动化5:文件上传，Cookie操作，调用 JavaScript，窗口截图

Posted 2021-06-15 测试baby

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了python3+selenium实现Web自动化5:文件上传，Cookie操作，调用 JavaScript，窗口截图相关的知识，希望对你有一定的参考价值。

在这里插入图片描述

一、Selenium之-文件上传

通过input标签实现的上传功能，可以将其看作是一个输入框，即通过send_keys()指定本地文件路径的方式实现文件上传。

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
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()

二、Selenium之-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 信息

#!/usr/bin/env python
# -*- encoding: utf-8 -*-

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()

"""
cookies = browser.get_cookies()  # 获得所有的cookie
with open("cookies.txt", "w") as fp:
    json.dump(cookies, fp)

browser.get("https://www.baidu.com")
browser.delete_all_cookies()
with open("cookies.txt", "r") as fp:
    cookies = json.load(fp)
    for cookie in cookies:        
        browser.add_cookie(cookie)
"""

# 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()

三、Selenium之-调用 javascript

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

#!/usr/bin/env python
# -*- encoding: utf-8 -*-

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()

四、Selenium之-窗口截图

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

#!/usr/bin/env python
# -*- encoding: utf-8 -*-

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()