diff --git a/AI/Ylt/Upload.py b/AI/Ylt/Upload.py index a50e7463..229c798f 100644 --- a/AI/Ylt/Upload.py +++ b/AI/Ylt/Upload.py @@ -1,11 +1,11 @@ import time -import pyautogui from playwright.sync_api import Playwright, sync_playwright + def run(playwright: Playwright) -> None: # 启动浏览器,禁用无头模式 - browser = playwright.chromium.launch(headless=False) + browser = playwright.chromium.launch(headless=True) context = browser.new_context() page = context.new_page() @@ -48,17 +48,25 @@ def run(playwright: Playwright) -> None: page.locator("img").click() # 触发文件选择对话框 file_chooser = fc_info.value file_chooser.set_files(r"d:\1.mp4") # 设置文件路径 - time.sleep(15) # 等待上传完成 + # 检查上传成功状态 + print("检查上传成功状态...") + while True: + success_message = page.locator("span:has-text('上传素材成功')") # 定位成功消息 + if success_message.is_visible(): # 如果成功消息可见,表示上传成功 + print("上传成功!") + break + print("上传进行中...") + time.sleep(0.5) # 每隔 0.5 秒检查一次 except Exception as e: print(f"操作过程中发生错误: {e}") finally: # 关闭浏览器 print("操作完成,关闭浏览器...") - context.close() browser.close() + # 初始化 Playwright playwright = sync_playwright().start() @@ -66,4 +74,4 @@ playwright = sync_playwright().start() run(playwright) # 结束 Playwright -playwright.stop() \ No newline at end of file +playwright.stop()