selenium之测试卫星资料页面操作(元素遍历)

# 测试气象卫星资料页面功能
# author:gongxr
# date:2017-07-24
import random, time
from selenium import webdriver
from login_page import LoginPage

login_obj = LoginPage()
driver = webdriver.Chrome(executable_path=login_obj.driver_path)
time.sleep(2)  # 智能等待
driver = login_obj.login(driver)  # 登录
# 登录后默认进入综合显示分析
# time.sleep(2)

# 进入气象卫星资料页面
driver.find_element_by_link_text('气象卫星资料').click()
time.sleep(3)

# 定义变量
random_num_to_str = str(random.randint(1000, 9999))  # 随机数，保证不重复
sates = ''  # 存储卫星按钮元素
plots = ''  # 存储全要素填图按钮元素
lights = ""  # 存储闪电按钮元素

def sates_click():
    '''遍历风云、葵花卫星数据展示'''
    sates = driver.find_elements_by_name('satellite')
    del sates[0]  # 默认第一个元素已经选中，跳过
    sate_last = sates[-1]
    for sate in sates:
        sate.click()
        time.sleep(2)
    sate_last.click()  # 再次点击取消显示

def plots_click():
    '''全要素填图遍历'''
    plots = driver.find_elements_by_xpath("//li[@mode='station,isoline']")
    plots[0].click()  # 地面全要素加载较慢，单独处理
    time.sleep(5)
    for plot in plots[1:]:
        plot.click()
        time.sleep(2)

def lights_click():
    '''闪电按钮遍历'''
    lights = driver.find_elements_by_xpath("//div[@class='gro_bolt']/ul/li")
    for light in lights:
        light.click()
        time.sleep(2)

sates_click()
plots_click()
lights_click()

# 关闭浏览器
time.sleep(2)
login_obj.quit()