feat: add material matching logic for discrete material plan cleaner

- Add db/materials_to_delete.py for querying materials to delete by manager
- Add manager_name parameter to DiscreteMaterialPlanCleaner
- Implement material keyword matching logic in process_order
- Update file paths from orderID.txt to ProductionID.txt

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
Misaka_Company
2026-01-23 16:50:14 +08:00
parent 902ff9b831
commit c176063bfa
4 changed files with 64 additions and 8 deletions

View File

@@ -6,23 +6,26 @@ import os
from playwright.sync_api import sync_playwright
from utils.auth import login, logout
from db.production_order_query import read_production_ids, query_production_order_numbers
from db.materials_to_delete import get_materials_to_delete
class DiscreteMaterialPlanCleaner:
"""离散备料计划维护数据清理器"""
def __init__(self, username, password, headless=False, verbose=True):
def __init__(self, username, password, manager_name, headless=False, verbose=True):
"""
初始化清理器
Args:
username: 登录用户名
password: 登录密码
manager_name: 负责人姓名
headless: 是否无头模式运行
verbose: 是否打印详细日志
"""
self.username = username
self.password = password
self.manager_name = manager_name
self.headless = headless
self.verbose = verbose
@@ -51,7 +54,7 @@ class DiscreteMaterialPlanCleaner:
return order_ids
def process_order(self, inner_frame, order_id, order_index, page1, debug_mode=False, debug_order=None):
def process_order(self, inner_frame, order_id, order_index, page1, materials_to_delete=None, debug_mode=False, debug_order=None):
"""清理单个订单的数据
Args:
@@ -59,9 +62,12 @@ class DiscreteMaterialPlanCleaner:
order_id: 订单号
order_index: 订单索引
page1: 页面对象
materials_to_delete: 待删除物料关键字列表
debug_mode: 是否启用调试模式
debug_order: 调试订单号
"""
if materials_to_delete is None:
materials_to_delete = []
from playwright.sync_api import TimeoutError
import re
import time
@@ -178,7 +184,8 @@ class DiscreteMaterialPlanCleaner:
# 获取材料名称
input_box = child_form.locator("div").filter(has_text=re.compile(r"^材料名称$")).locator("input[type='text']")
self._print(f"材料名称:{input_box.input_value()}")
material_name = input_box.input_value()
self._print(f"材料名称:{material_name}")
# 获取累计待发数量
input_box = child_form.locator("div").filter(has_text=re.compile(r"^累计待发数量$")).locator("input[type='text']")
@@ -188,6 +195,21 @@ class DiscreteMaterialPlanCleaner:
input_box = child_form.locator("div").filter(has_text=re.compile(r"^累计出库数量$")).locator("input[type='text']")
self._print(f"累计出库数量:{input_box.input_value()}")
# 检查是否需要清理该物料
should_delete = False
matched_keyword = None
for keyword in materials_to_delete:
if keyword in material_name:
should_delete = True
matched_keyword = keyword
break
if should_delete:
self._print(f">>> 需要清理:材料名称【{material_name}】匹配关键字【{matched_keyword}")
# TODO: 执行删除操作
else:
self._print(f"保留:材料名称【{material_name}】无需清理")
if id != detail_count - 1:
child_form.get_by_role("button").filter(has_text=re.compile(r"^$")).nth(2).click()
else:
@@ -246,6 +268,11 @@ class DiscreteMaterialPlanCleaner:
debug_mode: 是否启用调试模式
debug_order: 调试订单索引
"""
# 获取待删除物料列表
self._print(f"正在查询负责人 [{self.manager_name}] 的待删除物料列表...")
materials_to_delete = get_materials_to_delete(self.manager_name)
self._print(f"查询到 {len(materials_to_delete)} 个待删除物料关键字")
with sync_playwright() as playwright:
# 调用登录模块
browser, context, page, main_frame = login(
@@ -285,7 +312,7 @@ class DiscreteMaterialPlanCleaner:
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,
inner_frame, order_id, order_index, page1, materials_to_delete,
debug_mode=debug_mode, debug_order=debug_order
)
@@ -305,6 +332,7 @@ def main():
cleaner = DiscreteMaterialPlanCleaner(
username="BLDpengqiangqiang",
password="Cqbld123456.",
manager_name="彭羽",
headless=False,
verbose=True
)