You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
350 lines
13 KiB
350 lines
13 KiB
from playwright.sync_api import Playwright, sync_playwright
|
|
from CommonUtil import *
|
|
|
|
|
|
# 删除节目
|
|
def delProgram(page, keyword):
|
|
# 点击“节目”
|
|
changeJieMu(page)
|
|
# 定位所有符合条件的 <div> 元素
|
|
divs = page.locator("div.content-title")
|
|
# 获取所有 <div> 元素中的文本内容
|
|
text_list = divs.evaluate_all("elements => elements.map(div => div.textContent.trim())")
|
|
# 遍历文本内容,找到包含 keyword 的元素
|
|
cnt = 0
|
|
# 从后往前遍历
|
|
for index in range(len(text_list) - 1, -1, -1):
|
|
text = text_list[index]
|
|
if keyword in text:
|
|
print(f"Found element with text: {text}")
|
|
# 点击
|
|
divs.nth(index).click()
|
|
# 等待
|
|
wait()
|
|
# 点击删除按钮
|
|
delete_icon = page.locator("mat-icon:has-text('delete')").nth(0)
|
|
if delete_icon.is_visible():
|
|
delete_icon.click() # 点击该元素
|
|
# 定位并点击 <button> 元素
|
|
confirm_button = page.locator("button:has-text('确定')")
|
|
if confirm_button.is_visible():
|
|
confirm_button.click() # 点击该元素
|
|
cnt = cnt + 1
|
|
# 等待
|
|
wait()
|
|
if cnt > 0:
|
|
print("删除" + str(cnt) + "个节目")
|
|
else:
|
|
print("没有找到需要删除的节目,前缀:" + keyword)
|
|
|
|
|
|
# 删除以前上传的素材
|
|
def delMaterial(page):
|
|
# 点击“媒体库”
|
|
media_library = page.get_by_text("媒体库") # 通过 text 定位媒体库
|
|
media_library.click()
|
|
|
|
# 先删除历史文件,防止垃圾太多
|
|
# 点击 MATERIAL_FOLDER_NAME
|
|
playback_charge_status = page.get_by_text(MATERIAL_FOLDER_NAME)
|
|
playback_charge_status.dblclick()
|
|
# 等待
|
|
wait()
|
|
# 检查 <span> 元素是否存在
|
|
if page.locator("span.media-group-name").count() > 0:
|
|
# 鼠标移动到第一个 <span> 元素
|
|
first_span = page.locator("span.media-group-name").nth(0) # 定位第一个 <span> 元素
|
|
first_span.hover() # 鼠标移动到该元素
|
|
# 等待
|
|
wait()
|
|
# 定位并点击外层的 <span> 元素
|
|
checkbox_span = page.locator("span.mat-checkbox-inner-container").nth(0) # 定位 <span> 元素
|
|
if checkbox_span.is_visible():
|
|
checkbox_span.click(force=True) # 强制点击该元素
|
|
# 定位并点击 <mat-icon> 元素
|
|
delete_icon = page.locator("mat-icon:has-text('delete')").nth(0)
|
|
if delete_icon.is_visible():
|
|
delete_icon.click() # 点击该元素
|
|
# 定位并点击 <button> 元素
|
|
confirm_button = page.locator("button:has-text('确定')")
|
|
if confirm_button.is_visible():
|
|
confirm_button.click() # 点击该元素
|
|
# 等待
|
|
wait()
|
|
print("删除素材完成!")
|
|
|
|
|
|
# 上传文件
|
|
def uploadVideo(page, video_path):
|
|
# 点击“编辑节目” 点一下编辑节目是为了切换一下页面
|
|
jiemu_library = page.get_by_text("编辑节目") # 通过 text 定位编辑节目
|
|
jiemu_library.click()
|
|
|
|
# 点击“媒体库”
|
|
media_library = page.get_by_text("媒体库") # 通过 text 定位媒体库
|
|
media_library.click()
|
|
|
|
# 点击 MATERIAL_FOLDER_NAME
|
|
playback_charge_status = page.get_by_text(MATERIAL_FOLDER_NAME).nth(0)
|
|
playback_charge_status.dblclick()
|
|
|
|
# 点击“上传”按钮
|
|
upload_button = page.get_by_text("上传") # 通过 text 定位上传按钮
|
|
upload_button.click()
|
|
|
|
with page.expect_file_chooser() as fc_info:
|
|
page.locator("img").click() # 触发文件选择对话框
|
|
file_chooser = fc_info.value
|
|
file_chooser.set_files(video_path) # 设置文件路径
|
|
# 检查上传成功状态
|
|
print("检查上传成功状态...")
|
|
while True:
|
|
success_message = page.locator("span:has-text('上传素材成功')") # 定位成功消息
|
|
if success_message.is_visible(): # 如果成功消息可见,表示上传成功
|
|
print("上传成功!")
|
|
break
|
|
print("上传进行中...")
|
|
# 等待
|
|
wait()
|
|
|
|
|
|
# 发布节目
|
|
def publish(page, title, width, height):
|
|
print("点击编辑节目...")
|
|
jiemu_library = page.get_by_text("编辑节目") # 通过 text 定位编辑节目
|
|
jiemu_library.click()
|
|
# 等待
|
|
wait()
|
|
|
|
# 点击 MATERIAL_FOLDER_NAME
|
|
print("点击" + MATERIAL_FOLDER_NAME + "...")
|
|
playback_charge_status = page.locator("span:has-text('" + MATERIAL_FOLDER_NAME + "')").nth(0) # 定位 <span> 元素
|
|
if playback_charge_status.is_visible():
|
|
playback_charge_status.click() # 点击该元素
|
|
# 点击 <img> 元素
|
|
print("点击 <img> 元素...")
|
|
img_element = page.locator("img.grid-img").nth(0) # 定位 <img> 元素
|
|
if img_element.is_visible():
|
|
# 定位目标 <canvas> 元素
|
|
print("定位目标 <canvas> 元素...")
|
|
canvas_element = page.locator("canvas#proCanvas") # 定位 <canvas> 元素
|
|
|
|
wait()
|
|
# 执行拖动操作
|
|
print("执行拖动操作...")
|
|
img_element.drag_to(canvas_element) # 将图片拖动到 <canvas> 中
|
|
wait()
|
|
|
|
# 定位并输入文本
|
|
print("定位并输入文本...")
|
|
input_element = page.locator("input[formcontrolname='contentDisplayName']") # 定位 <input> 元素
|
|
if input_element.is_visible():
|
|
input_element.fill(title) # 输入文本
|
|
else:
|
|
print("<input> 元素未找到")
|
|
wait()
|
|
# 定位并输入数值
|
|
print("定位并输入数值...")
|
|
input_element = page.locator("input[formcontrolname='width']").nth(0) # 定位 <input> 元素
|
|
if input_element.is_visible():
|
|
input_element.fill(width) # 输入数值
|
|
else:
|
|
print("<input> 元素未找到")
|
|
wait()
|
|
# 定位并输入数值
|
|
print("定位并输入数值...")
|
|
input_element = page.locator("input[formcontrolname='height']").nth(0) # 定位 <input> 元素
|
|
if input_element.is_visible():
|
|
input_element.fill(height) # 输入数值
|
|
else:
|
|
print("<input> 元素未找到")
|
|
wait()
|
|
# 定位并点击“发布”按钮
|
|
print("定位并点击“发布”按钮...")
|
|
publish_button = page.locator("div.btn.publish") # 定位 <div> 元素
|
|
if publish_button.is_visible():
|
|
publish_button.click() # 点击该元素
|
|
# 等待
|
|
wait(2)
|
|
tree_node = page.locator("mat-tree-node.node-custom-style", has_text="充电站")
|
|
# 点击 <mat-tree-node> 元素
|
|
tree_node.click()
|
|
# 点击确定按钮
|
|
# 定位目标 <span> 元素
|
|
confirm_button = page.locator("span.mat-button-wrapper", has_text="确认")
|
|
# 点击 <span> 元素
|
|
confirm_button.click()
|
|
|
|
|
|
# 切换到节目
|
|
def changeJieMu(page):
|
|
# 点击“节目”
|
|
print("点击节目...")
|
|
program_library = page.get_by_text("节目").nth(0) # 通过 text 定位媒体库
|
|
program_library.click()
|
|
# 等待
|
|
wait()
|
|
|
|
|
|
# 排程
|
|
def paiCheng(page, keyword, stHour, stMin, stSec, endHour, endMin, endSec):
|
|
pc_library = page.get_by_text("排程").nth(0)
|
|
pc_library.click()
|
|
|
|
# 等待
|
|
wait()
|
|
# 定位目标 <span> 元素
|
|
today_span = page.locator("span.mat-button-toggle-label-content", has_text="日")
|
|
today_span.click()
|
|
# 等待
|
|
wait()
|
|
|
|
# 定位目标 <mat-icon> 元素
|
|
add_icon = page.locator("mat-icon", has_text="add").nth(1)
|
|
|
|
# 点击 <mat-icon> 元素
|
|
if add_icon.count() > 0: # 确保元素存在
|
|
add_icon.click()
|
|
# 定位目标 <span> 元素
|
|
broadcast_span = page.locator("span.mat-radio-label-content", has_text="插播")
|
|
|
|
# 点击 <span> 元素
|
|
if broadcast_span.count() > 0: # 确保元素存在
|
|
broadcast_span.click()
|
|
# 等待
|
|
wait()
|
|
|
|
# 点击 + 号
|
|
add_icon = page.locator("mat-icon", has_text="add").nth(2)
|
|
add_icon.click()
|
|
# 等待
|
|
wait()
|
|
|
|
# 等待目标对话框出现
|
|
dialog = page.get_by_role('dialog', name='选择节目') # 通过对话框标题定位
|
|
dialog.wait_for(state="visible")
|
|
|
|
# 在对话框内操作
|
|
# 定位所有 <span> 元素
|
|
spans = dialog.locator("span")
|
|
|
|
# 获取所有 <span> 元素的文本内容
|
|
text_list = spans.evaluate_all("elements => elements.map(span => span.textContent.trim())")
|
|
|
|
# 找到第一个以 keyword 开头的 <span> 元素
|
|
for index, text in enumerate(text_list):
|
|
if text.startswith(keyword):
|
|
print(f"Found first span with text: {text}")
|
|
# 获取元素的句柄并强制点击
|
|
span_handle = spans.nth(index).element_handle()
|
|
span_handle.evaluate("element => element.click()")
|
|
break # 找到第一个后退出循环
|
|
|
|
for index, text in enumerate(text_list):
|
|
if text.startswith("确认"):
|
|
print(f"Found first span with text: {text}")
|
|
span_handle = spans.nth(index).element_handle()
|
|
span_handle.evaluate("element => element.click()")
|
|
break
|
|
wait()
|
|
|
|
# 定位目标滑动开关的“滑块”部分
|
|
toggle_thumb = page.locator("div.mat-slide-toggle-thumb").nth(0)
|
|
# 点击“滑块”部分
|
|
toggle_thumb.click()
|
|
wait()
|
|
|
|
# 输入开始时间
|
|
input_element = page.locator("input#mat-input-3") # 通过 id 定位
|
|
input_element.click()
|
|
wait()
|
|
|
|
# 定位目标 <input> 元素
|
|
hour_input = page.locator("input.owl-dt-timer-input").nth(0) # 第一个输入框(小时)
|
|
minute_input = page.locator("input.owl-dt-timer-input").nth(1) # 第二个输入框(分钟)
|
|
second_input = page.locator("input.owl-dt-timer-input").nth(2) # 第三个输入框(秒)
|
|
# 输入时间
|
|
hour_input.fill(stHour) # 输入小时
|
|
wait()
|
|
minute_input.fill(stMin) # 输入分钟
|
|
wait()
|
|
second_input.fill(stSec) # 输入秒
|
|
wait()
|
|
# 设定
|
|
set_button = page.locator("button.owl-dt-control-button", has_text="Set")
|
|
set_button.click()
|
|
wait()
|
|
|
|
# ========================================== 输入结束时间==========================================
|
|
time_input = page.locator("input#mat-input-4") # 通过 id 定位
|
|
time_input.click()
|
|
wait()
|
|
|
|
hour_input = page.locator("input.owl-dt-timer-input").nth(0) # 第一个输入框(小时)
|
|
minute_input = page.locator("input.owl-dt-timer-input").nth(1) # 第二个输入框(分钟)
|
|
second_input = page.locator("input.owl-dt-timer-input").nth(2) # 第三个输入框(秒)
|
|
|
|
# 输入时间
|
|
hour_input.fill(endHour) # 输入小时
|
|
wait()
|
|
minute_input.fill(endMin) # 输入分钟
|
|
wait()
|
|
second_input.fill(endSec) # 输入秒
|
|
wait()
|
|
|
|
# 定位 <button> 元素
|
|
set_button = page.locator("button.owl-dt-control-button", has_text="Set")
|
|
set_button.click()
|
|
wait()
|
|
|
|
# 定位目标 <span> 元素
|
|
confirm_span = page.locator("span.mat-button-wrapper", has_text="确认")
|
|
# 点击 <span> 元素
|
|
confirm_span.click()
|
|
wait()
|
|
|
|
# 定位目标 <span> 元素
|
|
apply_span = page.locator("span", has_text="应用").nth(0)
|
|
apply_span.click()
|
|
wait()
|
|
|
|
|
|
def run(playwright: Playwright) -> None:
|
|
# 启动浏览器,禁用无头模式
|
|
browser = playwright.chromium.launch(headless=HEADLESS)
|
|
context = browser.new_context()
|
|
page = context.new_page()
|
|
|
|
try:
|
|
# 登录
|
|
login(page)
|
|
|
|
# 只删除固定前缀的节目
|
|
delProgram(page, PREFIX_NAME)
|
|
|
|
# 删除素材
|
|
delMaterial(page)
|
|
|
|
# 上传文件
|
|
uploadVideo(page, VIDEO_PATH)
|
|
|
|
# 发布节目
|
|
publish(page, 'Test1', str(1280), str(720))
|
|
|
|
|
|
except Exception as e:
|
|
print(f"操作过程中发生错误: {e}")
|
|
finally:
|
|
# 关闭浏览器
|
|
print("操作完成,关闭浏览器...")
|
|
context.close()
|
|
browser.close()
|
|
|
|
|
|
def doUpload():
|
|
# 初始化 Playwright
|
|
playwright = sync_playwright().start()
|
|
# 运行测试
|
|
run(playwright)
|
|
# 结束 Playwright
|
|
playwright.stop() |