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

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

"""
__title__  =
__Time__   = 2020/3/25 17:52
__Author__ = 小菠萝测试笔记
__Blog__   = https://www.cnblogs.com/poloyy/
"""
from time import sleep
from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.by import By


# 设置元素等待实例，最多等10秒，每0.5秒查找一次
def wait_element(driver, by_, element_, timeout=10):
    element = WebDriverWait(driver=driver, timeout=timeout).until(
        lambda x: x.find_element(by=by_, value=element_)
    )
    return element


# 设置元素等待实例，最多等10秒，每0.5秒查找一次
def wait_elements(driver, by_, element_, timeout=10):
    element = WebDriverWait(driver=driver, timeout=timeout).until(
        lambda x: x.find_elements(by=by_, value=element_)
    )
    return element


# 加载驱动
driver = webdriver.Chrome("../resources/chromedriver.exe")

# 打开网站
driver.get("http://www.51job.com")
driver.maximize_window()
# 搜索框
wait_element(driver, By.CSS_SELECTOR, "#kwdselectid").send_keys("python")

# 地区按钮
wait_element(driver, By.CSS_SELECTOR, "#work_position_input").click()

# 热门城市列表
city_lists = wait_elements(driver, By.CSS_SELECTOR, "div#work_position_click_center_right_list_000000 td em.on")

# 选中北京，取消选中其他城市
for city in city_lists:
    sleep(1)
    city.click()

wait_element(driver, By.CSS_SELECTOR, "em#work_position_click_center_right_list_category_000000_010000").click()

# 确定按钮
driver.find_element_by_css_selector("#work_position_click_bottom_save").click()

# 搜索按钮点击
wait_element(driver, By.CSS_SELECTOR, "div.top_wrap button").click()

# 找到职位列表
lists = wait_elements(driver, By.CSS_SELECTOR, "div#resultList>div.el")[1:]

for data in lists:
    spans = [i.text for i in data.find_elements_by_css_selector("span")]
    print(" | ".join(spans))

sleep(10)
# 退出浏览器
driver.quit()