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:
@@ -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
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user