from __future__ import annotations

from core.device_manager import DeviceManager
from tools._device import get_driver


def take_screenshot(
    device_id: str | None = None,
    *,
    manager: DeviceManager | None = None,
) -> bytes:
    return get_driver(device_id, manager=manager).screenshot()