20 lines
563 B
Python
20 lines
563 B
Python
from __future__ import annotations
|
|
|
|
from core.models import Scene
|
|
from device.manager import DeviceManager
|
|
from perception.icon_detector import find_icon as find_icon_in_scene
|
|
from tools.describe_screen import describe_screen
|
|
|
|
|
|
def find_icon(scene: Scene, name: str) -> dict[str, object]:
|
|
return find_icon_in_scene(scene, name).to_dict()
|
|
|
|
|
|
def find_icon_on_screen(
|
|
name: str,
|
|
*,
|
|
device_id: str | None = None,
|
|
manager: DeviceManager | None = None,
|
|
) -> dict[str, object]:
|
|
return find_icon(describe_screen(device_id, manager=manager), name)
|