refactoring

automation/defense_popup.py

@@ -0,0 +1,118 @@
+"""
+automation/defense_popup.py — 수비 팝업 조작
+
+수비 버튼 클릭, 수비 시퀀스 입력, 실책 수비 팝업 처리.
+"""
+from __future__ import annotations
+
+from typing import Any
+
+from playwright.sync_api import Page
+
+from core.config_loader import defense_button_id_map, position_number_map
+from core.field_calculator import (
+    extract_defense_sequence,
+    extract_error_position,
+    infer_error_position_fallback,
+    is_throwing_error,
+)
+from automation.page_helpers import get_last_visible_locator
+
+
+def click_defense_button_robustly(page: Page, position: str, click_count: int = 1) -> bool:
+    """수비 포지션 버튼을 안정적으로 클릭
+
+    ID 기반 → value 기반 → label 기반 순서로 시도
+    """
+    btn_map = defense_button_id_map()
+    pos_num = position_number_map()
+
+    # 1) ID 기반
+    button_selector = btn_map.get(position)
+    if button_selector:
+        loc = page.locator(button_selector)
+        if loc.count() > 0:
+            for _ in range(click_count):
+                loc.click(force=True)
+                page.wait_for_timeout(60)
+            return True
+
+    # 2) value 기반
+    value = pos_num.get(position)
+    if value:
+        loc = page.locator(f"input[name='defenseNumberBtn'][value='{value}']")
+        if loc.count() > 0:
+            for _ in range(click_count):
+                loc.click(force=True)
+                page.wait_for_timeout(60)
+            return True
+
+    # 3) label 기반
+    all_buttons = page.locator("input[name='defenseNumberBtn']").all()
+    for btn in all_buttons:
+        label = (btn.get_attribute("id") or "").lower()
+        if position in label or label in position:
+            for _ in range(click_count):
+                btn.click(force=True)
+                page.wait_for_timeout(60)
+            return True
+
+    return False
+
+
+def clear_defense_selections(page: Page) -> None:
+    """수비 선택 초기화"""
+    page.evaluate(
+        """() => {
+            document.querySelectorAll("input[name='defenseNumberBtn']").forEach(btn => {
+                btn.checked = false;
+            });
+        }"""
+    )
+
+
+def click_defense_sequence_in_popup(
+    page: Page,
+    sequence: list[str],
+    complete_button_selector: str | None = None,
+) -> None:
+    """수비 시퀀스 순서대로 클릭 후 완료 버튼 클릭"""
+    for position in sequence:
+        click_defense_button_robustly(page, position)
+        page.wait_for_timeout(80)
+
+    if complete_button_selector:
+        btn = get_last_visible_locator(page, complete_button_selector)
+        if btn:
+            btn.click()
+            page.wait_for_timeout(120)
+
+
+def fill_runner_out_defense(
+    page: Page, text: str, sequence_override: list[str] | None = None,
+) -> None:
+    """주루 아웃 수비 팝업 처리"""
+    page.wait_for_timeout(300)
+    sequence = sequence_override or extract_defense_sequence(text)
+    if sequence:
+        click_defense_sequence_in_popup(page, sequence)
+
+
+def fill_error_defense_popup(page: Page, text: str) -> None:
+    """실책 수비 팝업 처리"""
+    defense_sequence = extract_defense_sequence(text)
+    if len(defense_sequence) >= 2:
+        click_defense_sequence_in_popup(page, defense_sequence)
+    else:
+        error_position = extract_error_position(text)
+        if not error_position:
+            error_position = infer_error_position_fallback(text)
+        click_count = 2 if is_throwing_error(text) else 1
+        click_defense_button_robustly(page, error_position, click_count)
+
+    complete_button = get_last_visible_locator(page, "#btnNext")
+    if complete_button is None:
+        complete_button = get_last_visible_locator(page, "#btnAdd")
+    if complete_button:
+        complete_button.click()
+        page.wait_for_timeout(120)