Selenium系列（十八） - Web UI 自动化基础实战（5）

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

"""
__title__  =
__Time__   = 2020-03-31 21:30
__Author__ = 小菠萝测试笔记
__Blog__   = https://www.cnblogs.com/poloyy/
"""
import re
from time import sleep
from selenium.webdriver.support.select import Select
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as ec
from selenium.webdriver.support.wait import WebDriverWait

driver = webdriver.Chrome("../resources/chromedriver.exe")


def wait_element(by_, element_):
    element = WebDriverWait(driver, timeout=10).until(
        ec.presence_of_element_located((by_, element_))
    )
    return element


def wait_elements(by_, element_):
    element = WebDriverWait(driver, timeout=10).until(
        ec.presence_of_all_elements_located((by_, element_))
    )
    return element


# 打开网站
driver.get("https://kyfw.12306.cn/otn/leftTicket/init")
driver.maximize_window()

# 出发城市 填写 ‘南京南’，
from_ = wait_element(By.CSS_SELECTOR, "input#fromStationText")
from_.click()
from_.send_keys("南京南")
# 选择南京南
driver.find_element_by_css_selector("div#citem_0").click()

# 到达城市 填写 ‘杭州东’
to = wait_element(By.CSS_SELECTOR, "input#toStationText")
to.click()
to.send_keys("杭州东")
# 选择杭州东
driver.find_element_by_css_selector("div#citem_0").click()

# 发车时间 选 06:00--12:00
select = Select(driver.find_element_by_css_selector("select#cc_start_time"))
# 选择
select.select_by_visible_text("06:00--12:00")

# 发车日期选当前时间的下一天，也就是日期标签栏的，第二个标签
date_range = driver.find_elements_by_css_selector("div#date_range>ul>li")
date_range[1].click()

# 数据列表
lists = wait_elements(By.CSS_SELECTOR, "tbody#queryLeftTable>tr")[0::2]
print(len(lists))
#  循环数据
for data in lists:
    number = data.find_element_by_css_selector("td>div>div.train a.number").text
    two = data.find_elements_by_css_selector("td")[3].text
    if re.findall("有|\d+", two):
        print(number)

sleep(10)
driver.quit()