'commit'

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()
+playwright.stop()