chendelian
71091e38a9
All checks were successful
Publish Python Package to Gitea / publish (push) Successful in 29s
Expose ensure_ui_ready via DuiLib GetAccessible inject path; package x86 native helpers for host/skill consumers. Co-authored-by: Cursor <cursoragent@cursor.com>
82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
"""冷/暖探测。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Optional, Union
|
||
|
||
from .errors import DesktopWakeError, ERR_UIA_UNAVAILABLE
|
||
from .profiles import load_profile
|
||
from .types import ProbeResult, Profile
|
||
from . import uia_backend
|
||
from .window import (
|
||
find_hwnd_for_profile,
|
||
get_window_class,
|
||
get_window_title,
|
||
require_windows,
|
||
)
|
||
|
||
|
||
def probe(
|
||
profile: Union[str, Profile] = "wecom",
|
||
*,
|
||
hwnd: Optional[int] = None,
|
||
require_uia: bool = False,
|
||
) -> ProbeResult:
|
||
"""探测目标窗 UIA 是否已暖。
|
||
|
||
warm 判定：findall >= warm_min_findall，且 warm_markers 全部找到（若配置了 markers）。
|
||
"""
|
||
require_windows()
|
||
prof = load_profile(profile) if isinstance(profile, str) else profile
|
||
h = find_hwnd_for_profile(prof, hwnd=hwnd)
|
||
title = get_window_title(h)
|
||
cls = get_window_class(h)
|
||
notes: list[str] = []
|
||
|
||
available = uia_backend.uia_available()
|
||
if require_uia and not available:
|
||
raise DesktopWakeError(
|
||
ERR_UIA_UNAVAILABLE,
|
||
"未安装 uiautomation，请在共享 Python 环境中: pip install uiautomation",
|
||
)
|
||
|
||
findall_count = 0
|
||
names: list[str] = []
|
||
if available:
|
||
findall_count, names = uia_backend.count_findall(h)
|
||
if findall_count == 0 and not names:
|
||
notes.append("uia_timeout_or_empty")
|
||
markers_found, markers_missing = uia_backend.find_markers_on_hwnd(
|
||
h, list(prof.warm_markers)
|
||
)
|
||
# 若 Exists 未命中，用 Name 列表兜底
|
||
if markers_missing and names:
|
||
fb_found, fb_missing = uia_backend.markers_present(names, markers_missing)
|
||
markers_found = list(dict.fromkeys(markers_found + fb_found))
|
||
markers_missing = fb_missing
|
||
else:
|
||
notes.append("uiautomation_not_installed")
|
||
markers_found, markers_missing = [], list(prof.warm_markers)
|
||
|
||
density_ok = findall_count >= int(prof.warm_min_findall)
|
||
markers_ok = not prof.warm_markers or not markers_missing
|
||
warm = bool(available and density_ok and markers_ok)
|
||
|
||
if available and not density_ok:
|
||
notes.append(f"sparse_findall={findall_count}<{prof.warm_min_findall}")
|
||
if available and prof.warm_markers and markers_missing:
|
||
notes.append(f"missing_markers={','.join(markers_missing)}")
|
||
|
||
return ProbeResult(
|
||
hwnd=h,
|
||
profile_id=prof.id,
|
||
findall_count=findall_count,
|
||
markers_found=markers_found,
|
||
markers_missing=markers_missing,
|
||
warm=warm,
|
||
window_title=title,
|
||
window_class=cls,
|
||
uia_available=available,
|
||
notes=notes,
|
||
)
|