feat: add discrete material plan data cleaning tool
Implement a new automation script and utility to handle the cleaning of discrete material plan maintenance data. - Add main_clean.py as the primary entry point, configuring the cleaner instance and reading order IDs. - Add DiscreteMaterialPlanCleaner in utils/ utilizing Playwright to automate browser interactions. - Implement logic for user authentication (login/logout), UI setup for queries, and processing individual orders. - Include verbose logging and debug mode support for troubleshooting.
This commit is contained in:
27
main_clean.py
Normal file
27
main_clean.py
Normal file
@@ -0,0 +1,27 @@
|
||||
"""
|
||||
主程序 - 使用离散备料计划维护数据清理工具
|
||||
"""
|
||||
import os
|
||||
from utils.离散备料计划维护数据清理 import DiscreteMaterialPlanCleaner
|
||||
|
||||
|
||||
def main():
|
||||
# 创建清理器
|
||||
cleaner = DiscreteMaterialPlanCleaner(
|
||||
username="BLDpengqiangqiang",
|
||||
password="Cqbld123456.",
|
||||
headless=False,
|
||||
verbose=True
|
||||
)
|
||||
|
||||
# 设置文件路径
|
||||
order_id_file = os.path.join(os.path.dirname(__file__), "orderID.txt")
|
||||
|
||||
# 执行清理
|
||||
cleaner.clean(order_id_file)
|
||||
|
||||
input("按回车退出...")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
190
utils/离散备料计划维护数据清理.py
Normal file
190
utils/离散备料计划维护数据清理.py
Normal file
@@ -0,0 +1,190 @@
|
||||
"""
|
||||
离散备料计划维护数据清理工具
|
||||
负责登录、逐个清理订单数据
|
||||
"""
|
||||
import os
|
||||
from playwright.sync_api import sync_playwright
|
||||
from utils.auth import login, logout
|
||||
|
||||
|
||||
class DiscreteMaterialPlanCleaner:
|
||||
"""离散备料计划维护数据清理器"""
|
||||
|
||||
def __init__(self, username, password, headless=False, verbose=True):
|
||||
"""
|
||||
初始化清理器
|
||||
|
||||
Args:
|
||||
username: 登录用户名
|
||||
password: 登录密码
|
||||
headless: 是否无头模式运行
|
||||
verbose: 是否打印详细日志
|
||||
"""
|
||||
self.username = username
|
||||
self.password = password
|
||||
self.headless = headless
|
||||
self.verbose = verbose
|
||||
|
||||
def _print(self, *args, **kwargs):
|
||||
"""打印日志(如果 verbose=True)"""
|
||||
if self.verbose:
|
||||
print(*args, **kwargs)
|
||||
|
||||
def read_order_ids(self, file_path):
|
||||
"""读取订单号文件"""
|
||||
with open(file_path, 'r', encoding='utf-8') as f:
|
||||
# 去除空白行和空格
|
||||
order_ids = [line.strip() for line in f if line.strip()]
|
||||
return order_ids
|
||||
|
||||
def process_order(self, inner_frame, order_id, order_index, page1, debug_mode=False, debug_order=None):
|
||||
"""清理单个订单的数据
|
||||
|
||||
Args:
|
||||
inner_frame: 内部 iframe
|
||||
order_id: 订单号
|
||||
order_index: 订单索引
|
||||
page1: 页面对象
|
||||
debug_mode: 是否启用调试模式
|
||||
debug_order: 调试订单号
|
||||
"""
|
||||
from playwright.sync_api import TimeoutError
|
||||
import re
|
||||
import time
|
||||
|
||||
# 清空文本框
|
||||
textbox = inner_frame.get_by_role("textbox", name="生产订单号")
|
||||
textbox.fill("")
|
||||
|
||||
# 填充订单号
|
||||
textbox.fill(order_id)
|
||||
|
||||
# 点击查询
|
||||
inner_frame.locator(".search-component-searchBtn").click()
|
||||
self._print(f"第 {order_index + 1} 个订单查询完成,等待加载结果...")
|
||||
|
||||
# 等待加载完成
|
||||
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
|
||||
self._print(f"第 {order_index + 1} 个订单加载完成,开始清理数据...")
|
||||
|
||||
# 调试模式:只在指定订单暂停
|
||||
if debug_mode and (debug_order is None or order_index == debug_order):
|
||||
self._print(f"=== 调试暂停:第 {order_index + 1} 个订单 ===")
|
||||
page1.pause()
|
||||
page1.pause()
|
||||
time.sleep(1)
|
||||
pass
|
||||
|
||||
def setup_query_interface(self, inner_frame):
|
||||
"""设置查询界面"""
|
||||
import re
|
||||
|
||||
# 点击图标按钮打开查询界面
|
||||
inner_frame.locator(".search-name-wrapper > .iconfont").click()
|
||||
inner_frame.get_by_text("订单号查询").click()
|
||||
inner_frame.get_by_role("tab", name="全部").click()
|
||||
|
||||
# 填充并验证,如果失败则重试
|
||||
max_retries = 3
|
||||
expected_value = "5000"
|
||||
for attempt in range(max_retries):
|
||||
inner_frame.locator("#rc_select_0").fill(expected_value)
|
||||
inner_frame.locator("#rc_select_0").press("Enter")
|
||||
# 检查填充是否成功
|
||||
actual_value = inner_frame.locator("#rc_select_0").input_value()
|
||||
if actual_value == expected_value:
|
||||
self._print(f"文本框填充成功: {expected_value}")
|
||||
break
|
||||
else:
|
||||
self._print(f"第 {attempt + 1} 次填充失败,实际值: {actual_value},重试...")
|
||||
if attempt == max_retries - 1:
|
||||
self._print(f"警告: {max_retries} 次尝试后仍未成功填充,继续执行...")
|
||||
|
||||
def clean(self, order_id_file, debug_mode=False, debug_order=None):
|
||||
"""
|
||||
执行完整的数据清理流程
|
||||
|
||||
Args:
|
||||
order_id_file: 订单号文件路径
|
||||
debug_mode: 是否启用调试模式
|
||||
debug_order: 调试订单索引
|
||||
"""
|
||||
with sync_playwright() as playwright:
|
||||
# 调用登录模块
|
||||
browser, context, page, main_frame = login(
|
||||
playwright=playwright,
|
||||
username=self.username,
|
||||
password=self.password,
|
||||
headless=self.headless,
|
||||
ignore_https_errors=True
|
||||
)
|
||||
|
||||
self._print("=" * 80)
|
||||
self._print("开始执行离散备料计划维护数据清理")
|
||||
self._print("=" * 80)
|
||||
|
||||
# 登录成功后可以进行后续操作
|
||||
# 点击打开"功能菜单"
|
||||
main_frame.locator("i").first.click()
|
||||
|
||||
# 点击打开"离散生产订单维护"
|
||||
with page.expect_popup() as page1_info:
|
||||
main_frame.get_by_title("离散生产订单维护", exact=True).first.click()
|
||||
page1 = page1_info.value
|
||||
|
||||
# 获取 nested iframe
|
||||
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
|
||||
|
||||
# 设置查询界面
|
||||
self.setup_query_interface(inner_frame)
|
||||
|
||||
# 读取订单号文件
|
||||
order_ids = self.read_order_ids(order_id_file)
|
||||
self._print(f"共读取到 {len(order_ids)} 个订单号")
|
||||
|
||||
# 按订单清理
|
||||
for order_index, order_id in enumerate(order_ids):
|
||||
self._print(f"\n=== 开始处理第 {order_index + 1} 个订单,订单号: {order_id} ===")
|
||||
self.process_order(
|
||||
inner_frame, order_id, order_index, page1,
|
||||
debug_mode=debug_mode, debug_order=debug_order
|
||||
)
|
||||
|
||||
# 执行账号注销
|
||||
self._print("\n开始执行账号注销...")
|
||||
logout(main_frame, verbose=self.verbose)
|
||||
|
||||
self._print(f"\n=== 全部完成 ===")
|
||||
|
||||
# 关闭浏览器
|
||||
context.close()
|
||||
browser.close()
|
||||
|
||||
|
||||
def main():
|
||||
"""测试函数"""
|
||||
cleaner = DiscreteMaterialPlanCleaner(
|
||||
username="BLDpengqiangqiang",
|
||||
password="Cqbld123456.",
|
||||
headless=False,
|
||||
verbose=True
|
||||
)
|
||||
|
||||
order_id_file = os.path.join(os.path.dirname(__file__), "orderID.txt")
|
||||
|
||||
cleaner.clean(order_id_file)
|
||||
|
||||
input("按回车退出...")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user