admin/playwrite
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: normalize code formatting across the codebase

Browse Source 
- Standardize quote style (single to double quotes)
- Improve code formatting consistency
- Apply formatting to utilities, GUI components, and tools
- Update imports and docstrings for consistency

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
Misaka_Company
2026-02-05 14:56:35 +08:00
parent c34a300f0b
commit 2ca53f6a55
27 changed files with 693 additions and 385 deletions
Download Patch File Download Diff File Expand all files Collapse all files

91
record.py
View File

@@ -1,42 +1,52 @@
import re
import time
from playwright.sync_api import Playwright, sync_playwright, expect, TimeoutError as PWTimeoutError
from playwright.sync_api import (
Playwright,
sync_playwright,
expect,
TimeoutError as PWTimeoutError,
)
from utils.auth import login
def click_button_until_disappear(frame, button_name="保存提交", interval=5, max_attempts=None, max_duration=None):
def click_button_until_disappear(
frame, button_name="保存提交", interval=5, max_attempts=None, max_duration=None
):
"""
持续点击按钮直到按钮消失
参数:
frame: iframe对象
button_name: 按钮名称
interval: 点击间隔时间（秒）
max_attempts: 最大点击次数，None表示不限制
max_duration: 最大持续时间（秒），None表示不限制
返回:
dict: 包含点击次数、耗时等信息
"""
click_count = 0
start_time = time.time()
print(f"开始监控'{button_name}'按钮...")
while True:
try:
# 检查最大点击次数
if max_attempts and click_count >= max_attempts:
print(f"已达到最大点击次数{max_attempts}次，停止操作")
break
# 检查最大持续时间
if max_duration and (time.time() - start_time) >= max_duration:
elapsed = time.time() - start_time
print(f"已达到最大持续时间{max_duration}秒（实际{elapsed:.1f}秒），停止操作")
print(
f"已达到最大持续时间{max_duration}秒（实际{elapsed:.1f}秒），停止操作"
)
break
button = frame.get_by_role("button", name=button_name)
# 检查按钮是否存在
if button.count() == 0:
elapsed = time.time() - start_time
@@ -45,17 +55,17 @@ def click_button_until_disappear(frame, button_name="保存提交", interval=5,
return {
"success": True,
"click_count": click_count,
"elapsed_time": elapsed
"elapsed_time": elapsed,
}
# 点击按钮
button.click()
click_count += 1
print(f"[{time.strftime('%H:%M:%S')}] 第{click_count}次点击'{button_name}'")
# 等待指定时间
time.sleep(interval)
except Exception as e:
elapsed = time.time() - start_time
print(f"点击过程中出现异常: {e}")
@@ -63,20 +73,19 @@ def click_button_until_disappear(frame, button_name="保存提交", interval=5,
"success": False,
"click_count": click_count,
"elapsed_time": elapsed,
"error": str(e)
"error": str(e),
}
def get_input_by_label(frame, label_text: str, label_locator=None):
"""
通过标签文本获取对应的输入框对象
参数:
frame: iframe对象
label_text: 标签文本（用于查找和日志输出）
label_locator: 可选，已经定位好的标签locator。如果为None，则函数内部查找
返回:
成功：返回输入框的locator对象
失败：返回None
@@ -84,21 +93,25 @@ def get_input_by_label(frame, label_text: str, label_locator=None):
try:
# 如果没有传入label_locator，则根据label_text查找
if label_locator is None:
label_locator = frame.locator("div").filter(has_text=re.compile(f"^{label_text}$")).first
label_locator = (
frame.locator("div")
.filter(has_text=re.compile(f"^{label_text}$"))
.first
)
print(f"找到{label_text}标签")
# 向上找到包含标签和输入框的共同父容器
parent_container = label_locator.locator("..") # 父元素
# 在父容器中查找输入框
input_box = parent_container.locator("input").first
# 如果父元素中没有，再向上一层
if input_box.count() == 0:
print(f"{label_text}: 在父元素中未找到，向上一层查找...")
grandparent = parent_container.locator("..") # 祖父元素
input_box = grandparent.locator("input").first
# 验证是否找到输入框
if input_box.count() > 0:
input_box.wait_for(state="visible", timeout=5000)
@@ -109,11 +122,12 @@ def get_input_by_label(frame, label_text: str, label_locator=None):
else:
print(f"✗ {label_text}: 在父容器中找不到输入框")
return None
except Exception as e:
print(f"✗ {label_text}失败: {e}")
return None
def run(playwright: Playwright) -> None:
# 1. 登录
browser, context, page, main_frame = login(
@@ -121,7 +135,7 @@ def run(playwright: Playwright) -> None:
username="BLDpengqiangqiang",
password="Cqbld123456.",
headless=False,
ignore_https_errors=True
ignore_https_errors=True,
)
# 3. 点击打开“补货安排”
@@ -138,7 +152,7 @@ def run(playwright: Playwright) -> None:
# 提取外层 forwardFrame
outer_frame = page1.locator("#forwardFrame").content_frame
# 关键：等待内层 #mainiframe 出现并加载
inner_frame_locator = outer_frame.locator("#mainiframe")
inner_frame_locator.wait_for(state="visible", timeout=15000) # 最多等15秒
@@ -153,7 +167,7 @@ def run(playwright: Playwright) -> None:
# 输入排产号
textbox = inner_frame.get_by_role("textbox", name="排产号")
textbox.fill("R") # 直接 fill，不需要 press CapsLock
# 输入日期
inner_frame.get_by_role("textbox", name="单据日期结束日期").click()
inner_frame.get_by_text("今日").click()
@@ -170,7 +184,7 @@ def run(playwright: Playwright) -> None:
summary_locator = inner_frame.get_by_text(re.compile(r"合计:\s*\d+\s*行"))
# 获取元素的完整文本
summary_text = summary_locator.inner_text() # 例如 "合计: 135 行"
summary_text = summary_locator.inner_text() # 例如 "合计: 135 行"
# 用正则提取数字
match = re.search(r"\d+", summary_text)
@@ -181,14 +195,12 @@ def run(playwright: Playwright) -> None:
print("未匹配到行数")
row_count = 0
inner_frame.get_by_role("button").filter(has_text="补货安排").hover()
inner_frame.get_by_text("生产订单").click()
inner_frame.get_by_role("textbox", name="工厂").fill("10010705")
with page1.expect_popup(timeout=60000) as page2_info:
inner_frame.get_by_role("button", name="确定(Y)").click()
page2=page2_info.value
page2 = page2_info.value
# 新页面：等待页面加载完成 + 提取嵌套 iframe
print("新页面已打开，正在等待内层 iframe 加载...")
@@ -204,7 +216,6 @@ def run(playwright: Playwright) -> None:
# label_div = inner_frame.locator("div").filter(has_text=re.compile(r"^生产部门$")).first
# input_box = label_div.locator("..").locator(".wui-input-close > .wui-input")
time.sleep(25) # 等待页面完全加载
# pro_dep_input = get_input_by_label(inner_frame, "生产部门")
# if pro_dep_input:
@@ -216,18 +227,14 @@ def run(playwright: Playwright) -> None:
# if pro_SN_input:
# print(pro_SN_input.input_value())
#result = click_button_until_disappear(inner_frame, "保存提交", interval=5)
#print(result)
# result = click_button_until_disappear(inner_frame, "保存提交", interval=5)
# print(result)
input("操作完成，按回车关闭...")
context.close()
browser.close()
with sync_playwright() as playwright:
run(playwright)
run(playwright)