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

五、Selenium之-关闭窗口

close() 关闭单个窗口
quit() 关闭所有窗口

参考链接：
https://www.cnblogs.com/csmashang/p/12706835.html