selenium-八大元素定位(下)(封装自己的元素定位方法)

框架中的find_element

import time
from webdriver_helper import webdriver, get_webdriver
from selenium.webdriver import chrome
from selenium.webdriver.common.by import By

png_count = 0


def find_element(driver, *args, **kwargs):
    """
    1、调用selenium完成元素定位
    2、记录日志
    3、记录截图
    :param driver:
    :param args:
    :param kwargs:
    :return:
    """

    global png_count
    png_count += 1  # 每次截图,变量自增

    print("准备定位元素", args, kwargs)
    driver.get_screenshot_as_file(f"{png_count}_page.png")  # 定位前记录页面的内容截图
    el = driver.find_element(*args, **kwargs)  # 将参数转发给selenium
    # 可能会成功,可能会失败

    print("定位到元素", el.tag_name)
    print("元素的文本", el.text)
    png = el.screenshot_as_png
    with open(f"{png_count}_el.png", "wb") as f:  # 定位后,记录定位到的元素
        f.write(png)
        return el

driver = webdriver.Chrome()  # 实例化 ,刚启动浏览器是空白页面
driver.maximize_window()  # 窗口最大化
driver.get("http://101.34.221.219:8010/")  # 服务器会设置cookie

el = find_element(driver, by=By.XPATH, value='/html/body/div[2]/div/ul[1]/div/div/a[1]')

el.click()

driver.quit()  # 退出浏览器 调用对象的方法


posted @ 2022-11-08 22:11  NiceTwocu  阅读(38)  评论(0编辑  收藏  举报