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

style: apply code formatting to discrete_material_plan_cleaner.py

Browse Source 
Apply consistent code formatting including:
- Single quotes to double quotes for string literals
- Multi-line formatting for long expressions
- Trailing whitespace cleanup
- Blank line adjustments for PEP 8 compliance

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
Misaka
2026-02-24 22:23:34 +08:00
parent a520c9a05c
commit de25841364
1 changed files with 59 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

93
utils/discrete_material_plan_cleaner.py
View File

@@ -24,11 +24,12 @@ from db.materials_to_delete import get_materials_to_delete
# --- 日志配置 --- # --- 日志配置 ---
logging.basicConfig( logging.basicConfig(
level=logging.INFO, level=logging.INFO,
format='%(asctime)s [%(levelname)s] %(message)s', format="%(asctime)s [%(levelname)s] %(message)s",
datefmt='%Y-%m-%d %H:%M:%S' datefmt="%Y-%m-%d %H:%M:%S",
) )
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class DiscreteMaterialPlanCleaner: class DiscreteMaterialPlanCleaner:
"""离散备料计划维护数据清理器""" """离散备料计划维护数据清理器"""
@@ -44,9 +45,12 @@ class DiscreteMaterialPlanCleaner:
def _log(self, message, level="info"): def _log(self, message, level="info"):
"""统一日志输出控制""" """统一日志输出控制"""
if self.verbose: if self.verbose:
if level == "info": logger.info(message) if level == "info":
elif level == "warn": logger.warning(message) logger.info(message)
elif level == "error": logger.error(message) elif level == "warn":
logger.warning(message)
elif level == "error":
logger.error(message)
def _is_button_enabled(self, button_locator): def _is_button_enabled(self, button_locator):
"""判定按钮是否可用""" """判定按钮是否可用"""
@@ -58,9 +62,12 @@ class DiscreteMaterialPlanCleaner:
def _get_input_value(self, container, label_regex): def _get_input_value(self, container, label_regex):
"""通用辅助函数：根据 Label 正则获取 Input 的值""" """通用辅助函数：根据 Label 正则获取 Input 的值"""
return container.locator("div").filter( return (
has_text=re.compile(label_regex, re.MULTILINE) container.locator("div")
).locator("input").first.input_value() .filter(has_text=re.compile(label_regex, re.MULTILINE))
.locator("input")
.first.input_value()
)
def preload_data(self): def preload_data(self):
"""批量预取数据库数据""" """批量预取数据库数据"""
@@ -74,7 +81,9 @@ class DiscreteMaterialPlanCleaner:
"""读取文件并查询生产订单号""" """读取文件并查询生产订单号"""
production_ids = read_production_ids(production_id_file) production_ids = read_production_ids(production_id_file)
order_ids = query_production_order_numbers(production_ids) order_ids = query_production_order_numbers(production_ids)
self._log(f"读取到 {len(production_ids)} 个总排号 -> 匹配到 {len(order_ids)} 个生产订单号") self._log(
f"读取到 {len(production_ids)} 个总排号 -> 匹配到 {len(order_ids)} 个生产订单号"
)
return order_ids return order_ids
def process_order(self, inner_frame, order_id, order_index, page1): def process_order(self, inner_frame, order_id, order_index, page1):
@@ -88,7 +97,7 @@ class DiscreteMaterialPlanCleaner:
loading_locator = inner_frame.locator("div").filter(has_text="加载中").nth(1) loading_locator = inner_frame.locator("div").filter(has_text="加载中").nth(1)
try: try:
loading_locator.wait_for(state="visible", timeout=3000) loading_locator.wait_for(state="visible", timeout=3000)
loading_locator.wait_for(state="hidden", timeout=60000) loading_locator.wait_for(state="hidden", timeout=60000)
except TimeoutError: except TimeoutError:
pass pass
@@ -103,32 +112,42 @@ class DiscreteMaterialPlanCleaner:
detail_inner_frame = detail_main_frame.locator("#mainiframe").content_frame detail_inner_frame = detail_main_frame.locator("#mainiframe").content_frame
# 5. 提取订单状态信息 # 5. 提取订单状态信息
plan_code_locator = detail_inner_frame.get_by_text(re.compile(r"^离散备料计划维护：")) plan_code_locator = detail_inner_frame.get_by_text(
re.compile(r"^离散备料计划维护：")
)
plan_code_locator.wait_for(state="visible", timeout=15000) plan_code_locator.wait_for(state="visible", timeout=15000)
detail_count_text = detail_inner_frame.get_by_text(re.compile(r"^详细信息 \(\d+\)$")).inner_text() detail_count_text = detail_inner_frame.get_by_text(
re.compile(r"^详细信息 \(\d+\)$")
).inner_text()
detail_count = int(re.search(r"\((\d+)\)", detail_count_text).group(1)) detail_count = int(re.search(r"\((\d+)\)", detail_count_text).group(1))
status_text = detail_inner_frame.get_by_text(re.compile(r"^备料状态:.+$")).inner_text() status_text = detail_inner_frame.get_by_text(
detail_status = re.search(r"备料状态:(.+)$", status_text.replace("\n", "")).group(1) re.compile(r"^备料状态:.+$")
).inner_text()
detail_status = re.search(
r"备料状态:(.+)$", status_text.replace("\n", "")
).group(1)
# 6. 执行清理逻辑 # 6. 执行清理逻辑
if detail_status == "审批通过": if detail_status == "审批通过":
if detail_count >0: if detail_count > 0:
# --- 点击修改并等待状态切换 (保留原逻辑) --- # --- 点击修改并等待状态切换 (保留原逻辑) ---
detail_inner_frame.get_by_role("button", name="修改").click() detail_inner_frame.get_by_role("button", name="修改").click()
# 关键判断：等待保存按钮出现，确认进入编辑模式 # 关键判断：等待保存按钮出现，确认进入编辑模式
save_button_locator = detail_inner_frame.get_by_role("button", name="保存") save_button_locator = detail_inner_frame.get_by_role(
"button", name="保存"
)
save_button_locator.wait_for(state="visible", timeout=10000) save_button_locator.wait_for(state="visible", timeout=10000)
self._log("已进入编辑模式（保存按钮已就绪）") self._log("已进入编辑模式（保存按钮已就绪）")
# --------------------------------------- # ---------------------------------------
detail_inner_frame.get_by_text("展开").first.click() detail_inner_frame.get_by_text("展开").first.click()
child_form = detail_inner_frame.locator(".card-table-side-box") child_form = detail_inner_frame.locator(".card-table-side-box")
button_wrapper = child_form.locator(".button-wrapper") button_wrapper = child_form.locator(".button-wrapper")
delete_row_btn = button_wrapper.get_by_role("button", name="删行") delete_row_btn = button_wrapper.get_by_role("button", name="删行")
next_btn = button_wrapper.locator(".icon-jiantouyou") next_btn = button_wrapper.locator(".icon-jiantouyou")
collapse_btn = button_wrapper.locator(".icon-celashouqi") collapse_btn = button_wrapper.locator(".icon-celashouqi")
@@ -138,7 +157,7 @@ class DiscreteMaterialPlanCleaner:
# 稳定性检查：等待行号更新 # 稳定性检查：等待行号更新
current_row = self._get_input_value(child_form, r"^行号$") current_row = self._get_input_value(child_form, r"^行号$")
if current_row == last_row_number: if current_row == last_row_number:
time.sleep(0.5) time.sleep(0.5)
material_code = self._get_input_value(child_form, r"^材料编码") material_code = self._get_input_value(child_form, r"^材料编码")
material_name = self._get_input_value(child_form, r"^材料名称") material_name = self._get_input_value(child_form, r"^材料名称")
@@ -153,14 +172,16 @@ class DiscreteMaterialPlanCleaner:
else: else:
self._log(f"⚠️ 待发数量为 {pending_qty}，跳过删除", "warn") self._log(f"⚠️ 待发数量为 {pending_qty}，跳过删除", "warn")
else: else:
self._log(f"物料 {material_name} ({material_code}) 不在删除列表中，不做处理") self._log(
f"物料 {material_name} ({material_code}) 不在删除列表中，不做处理"
)
if self._is_button_enabled(next_btn): if self._is_button_enabled(next_btn):
last_row_number = current_row last_row_number = current_row
next_btn.click() next_btn.click()
else: else:
break break
collapse_btn.click() collapse_btn.click()
# 执行最终保存逻辑（如业务需要） # 执行最终保存逻辑（如业务需要）
# save_button_locator.click() # save_button_locator.click()
else: else:
@@ -176,7 +197,7 @@ class DiscreteMaterialPlanCleaner:
inner_frame.locator(".search-name-wrapper > .iconfont").click() inner_frame.locator(".search-name-wrapper > .iconfont").click()
inner_frame.get_by_text("订单号查询").click() inner_frame.get_by_text("订单号查询").click()
inner_frame.get_by_role("tab", name="全部").click() inner_frame.get_by_role("tab", name="全部").click()
# 填充每页显示条数（5000条测试值） # 填充每页显示条数（5000条测试值）
input_el = inner_frame.locator("#rc_select_0") input_el = inner_frame.locator("#rc_select_0")
input_el.fill("5000") input_el.fill("5000")
@@ -196,8 +217,8 @@ class DiscreteMaterialPlanCleaner:
ignore_https_errors=True, ignore_https_errors=True,
) )
self._log("="*30 + " 开始清理任务 " + "="*30) self._log("=" * 30 + " 开始清理任务 " + "=" * 30)
# 进入功能页面 # 进入功能页面
main_frame.locator("i").first.click() main_frame.locator("i").first.click()
with page.expect_popup() as page1_info: with page.expect_popup() as page1_info:
@@ -207,7 +228,9 @@ class DiscreteMaterialPlanCleaner:
# 定位主 Iframe # 定位主 Iframe
work_main_frame = page1.locator("#forwardFrame").content_frame work_main_frame = page1.locator("#forwardFrame").content_frame
inner_frame = work_main_frame.locator("#mainiframe").content_frame inner_frame = work_main_frame.locator("#mainiframe").content_frame
inner_frame.locator("#hot-key-head_list").wait_for(state="visible", timeout=15000) inner_frame.locator("#hot-key-head_list").wait_for(
state="visible", timeout=15000
)
self.setup_query_interface(inner_frame) self.setup_query_interface(inner_frame)
order_ids = self.get_production_order_numbers(production_id_file) order_ids = self.get_production_order_numbers(production_id_file)
@@ -219,13 +242,14 @@ class DiscreteMaterialPlanCleaner:
self.process_order(inner_frame, order_id, index, page1) self.process_order(inner_frame, order_id, index, page1)
except Exception as e: except Exception as e:
self._log(f"处理单号 {order_id} 时发生异常: {e}", "error") self._log(f"处理单号 {order_id} 时发生异常: {e}", "error")
continue # 单个失败不影响整体执行 continue # 单个失败不影响整体执行
# 登出清理 # 登出清理
logout(work_main_frame, verbose=self.verbose) logout(work_main_frame, verbose=self.verbose)
context.close() context.close()
browser.close() browser.close()
self._log("="*30 + " 任务全部完成 " + "="*30) self._log("=" * 30 + " 任务全部完成 " + "=" * 30)
def main(): def main():
# 路径配置 # 路径配置
@@ -236,11 +260,12 @@ def main():
username="BLDpengqiangqiang", username="BLDpengqiangqiang",
password="your_password_here", password="your_password_here",
manager_name="彭羽", manager_name="彭羽",
headless=False headless=False,
) )
cleaner.clean(id_file) cleaner.clean(id_file)
input("执行完毕，按回车键退出程序...") input("执行完毕，按回车键退出程序...")
if __name__ == "__main__": if __name__ == "__main__":
main() main()