Add locator helper module and enhance main script for element interaction

main.py

@@ -3,7 +3,9 @@
 """
 from playwright.sync_api import sync_playwright
 from login import login
-
+from time import sleep
+import re
+import time
 def main():
     with sync_playwright() as playwright:
         # 调用登录模块
@@ -16,8 +18,57 @@ def main():
         )
 
         # 登录成功后可以进行后续操作
-        print("登录成功，可以进行后续操作...")
+        # 点击打开“菜单”
+        main_frame.locator("i").first.click()
+        #page.pause()
+        # 点击打开“菜单”
+        with page.expect_popup() as page1_info:
+            main_frame.get_by_title("离散备料计划维护", exact=True).first.click()
+        page1 = page1_info.value
 
+        
+        main_frame=page1.locator("#forwardFrame").content_frame
+        inner_frame_locator = main_frame.locator("#mainiframe")
+        inner_frame_locator.wait_for(state="visible", timeout=15000)
+        inner_frame = inner_frame_locator.content_frame
+
+        # 点击图标按钮
+        inner_frame.locator(".search-name-wrapper > .iconfont").click()
+        inner_frame.get_by_text("订单号查询").click()
+        textbox = inner_frame.get_by_role("textbox", name="来源生产订单号")
+        textbox.fill("SC70202510110003,SC70202510110004")
+        inner_frame.get_by_role("tab", name="全部").click()
+        inner_frame.locator("#rc_select_0").fill("5000")
+        #发送回车键
+        inner_frame.locator("#rc_select_0").press("Enter")
+        inner_frame.locator(".search-component-searchBtn").click()
+        print("查询完成，等待加载结果...")
+        # 等待加载完成
+        loading_locator = inner_frame.locator("div").filter(has_text="加载中").nth(1)
+        try:
+            loading_locator.wait_for(state="visible", timeout=3000)
+            loading_locator.wait_for(state="hidden", timeout=0)  # 无限等待，直到消失
+        except TimeoutError:
+            # 加载很快完成，或者没有出现加载提示
+            pass
+        print("加载完成，开始选择数据...")
+        inner_frame.get_by_role("row", name="序号").get_by_label("").check()
+
+        inner_frame.get_by_role("button", name="更多").hover()
+        inner_frame.get_by_text("输出").click()
+        input_box = inner_frame.locator("div").filter(has_text=re.compile(r"^行数阈值$")).locator("input[type='text']")
+        input_box.fill("300000")
+        # 等待下载事件
+        with page1.expect_download() as download_info:
+            inner_frame.get_by_role("button", name="确定(Y)").click()
+
+        # 获取下载对象
+        download = download_info.value
+
+        # 指定保存路径和文件名
+        download.save_as("D:/python/playwrite/data/导出文件.xlsx")
+
+        page1.pause()
         input("按回车退出...")
 
         # 关闭浏览器