【自动化】Python SeleniumUtil 工具
Python SeleniumUtil
python
import time
from selenium.webdriver.support.wait import WebDriverWait
from selenium import webdriver
from selenium.webdriver.chrome.service import Service as ChromeService
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
class SeleniumUtil:
class 标签页:
@staticmethod
def 切换到新窗口句柄(driver, before_handles, after_handles):
# 检查是否有新的窗口句柄出现
new_handles = after_handles - before_handles
if new_handles:
# 切换到新打开的标签页
new_window = new_handles.pop()
driver.switch_to.window(new_window)
return True
else:
return False
@staticmethod
def 获取所有已知窗口句柄(driver):
return set(driver.window_handles)
@staticmethod
def 切换到第一个窗口(driver):
# 切换到第一个窗口
driver.switch_to.window(driver.window_handles[0])
@staticmethod
def 切换到当前窗口(driver):
# 存储窗口的 ID
original_window = driver.current_window_handle
print(original_window)
# 检查没有打开其他的窗口
assert len(driver.window_handles) == 1
class Driver:
#初始化浏览器
@staticmethod
def 初始化驱动通过Debug模式(驱动路径, debuggerAddress):
# 设置Edge选项
options = webdriver.ChromeOptions()
# options.add_experimental_option('prefs',{"extensions.ui.developer_mode": True,})
options.add_experimental_option("debuggerAddress", debuggerAddress)
# C:\\chromedriver\\129.0.6656.0\\chromedriver.exe
service = ChromeService(executable_path = 驱动路径)
# 创建一个新的Edge浏览器实例，并启用无痕模式
driver = webdriver.Chrome(service = service, options = options)
return driver
# 如果你只需要确认元素存在，但不立即与其交互，那么可以使用 presence_of_element_located。
# 如果你需要确保元素不仅存在而且可以安全地被用户交互（如点击），则应使用 element_to_be_clickable
class 页面元素:
@staticmethod
def 判断元素是否存在(driver, 标签, 内容, 超时时间):
element = SeleniumUtil.页面元素.获取元素(driver, 标签, 内容, 超时时间)
if element is not None:
return True
else:
return False
@staticmethod
def 点击元素(元素):
if 元素 is not None:
元素.click()
return True
else:
return False
@staticmethod
def get元素(driver, 需要交互, 类型, 值, 超时时间):
if 超时时间 is None:
超时时间 = 7
if 需要交互 is None:
需要交互 = True
try:
wait = WebDriverWait(driver, 超时时间)
if 需要交互:
元素 = wait.until(EC.element_to_be_clickable((类型, 值)))
if 元素 is not None:
return 元素
else:
return driver.find_element(类型, 值)
else:
元素 = wait.until(EC.presence_of_element_located((类型, 值)))
if 元素 is not None:
return 元素
else:
return driver.find_element(类型, 值)
except Exception as e:
try:
return driver.find_element(类型, 值)
except Exception as e2:
time.sleep(0)
return None
@staticmethod
def get元素ById(driver, id, 需要交互, 超时时间):
return SeleniumUtil.页面元素.get元素(driver, 需要交互, By.ID, id, 超时时间)
@staticmethod
def get元素By标签及内容(driver, 标签, 内容, 需要交互, 超时时间):
return SeleniumUtil.页面元素.get元素(driver, 需要交互, By.XPATH, f"//{标签}[text()='{内容}']", 超时时间)
@staticmethod
def get元素ByName属性(driver, name, 需要交互, 超时时间):
return SeleniumUtil.页面元素.get元素(driver, 需要交互, By.NAME, name, 超时时间)
@staticmethod
def get元素By标签及属性(driver, 标签, 属性, 属性值, 需要交互, 超时时间):
return SeleniumUtil.页面元素.get元素(driver, 需要交互, By.CSS_SELECTOR, f'{标签}[{属性}="{属性值}"]', 超时时间)
@staticmethod
def get元素By标签及属性立即(driver, 标签, 属性, 属性值, 需要交互, 超时时间):
值 = f'{标签}[{属性}="{属性值}"]'
print(值)
return driver.find_element(By.CSS_SELECTOR, 值)
#
# @staticmethod
# def 获取元素通过标签及属性(driver, 标签, 属性, 属性值):
# try:
# element = driver.find_element(By.CSS_SELECTOR, f'{标签}[{属性}="{属性值}"]')
# return element
# except Exception as e:
# time.sleep(0)
# return None
class 扩展程序:
@staticmethod
def 打开扩展程序页(driver):
driver.get('chrome://extensions')
@staticmethod
def 查询开发者模式状态(driver):
# 打开目标网页
SeleniumUtil.扩展程序.打开扩展程序页(driver)
# 使用WebDriverWait等待页面加载完成，代替固定等待时间
# time.sleep(5)
try:
wait = WebDriverWait(driver, 5)
time.sleep(1)
js = """
// window.location.href = 'chrome://extensions'
//获取开发者模式状态
let flag = document.querySelector("body > extensions-manager").shadowRoot.querySelector("#toolbar").shadowRoot.querySelector("#devMode").getAttribute("aria-pressed");
if(flag === 'true'){
return true;
}else{
return false;
}
"""
status = driver.execute_script(js)
print(f"当前开发者模式状态-{status}")
return status
except Exception as e:
time.sleep(0)
@staticmethod
def 切换开发者模式状态(driver):
try:
wait = WebDriverWait(driver, 7)
time.sleep(1)
js = """
document.querySelector("body > extensions-manager").shadowRoot.querySelector("#toolbar").shadowRoot.querySelector("#devMode").click();
return true;
"""
driver.execute_script(js)
except Exception as e:
time.sleep(0)
@staticmethod
def 打开开发者模式(driver):
SeleniumUtil.扩展程序.修改开发者模式状态(driver, True)
@staticmethod
def 修改开发者模式状态(driver, flag):
# 打开目标网页
SeleniumUtil.扩展程序.打开扩展程序页(driver)
# 使用WebDriverWait等待页面加载完成，代替固定等待时间
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
开发者模式状态 = SeleniumUtil.扩展程序.查询开发者模式状态(driver)
if 开发者模式状态 == flag:
time.sleep(0)
else:
SeleniumUtil.扩展程序.切换开发者模式状态(driver)
print(f"修改开发者模式状态-{flag}")