39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
"""Ecovacs util functions."""
|
|
from __future__ import annotations
|
|
|
|
import random
|
|
import string
|
|
from typing import TYPE_CHECKING
|
|
|
|
from .entity import (
|
|
EcovacsCapabilityEntityDescription,
|
|
EcovacsDescriptionEntity,
|
|
EcovacsEntity,
|
|
)
|
|
|
|
if TYPE_CHECKING:
|
|
from .controller import EcovacsController
|
|
|
|
|
|
def get_client_device_id() -> str:
|
|
"""Get client device id."""
|
|
return "".join(
|
|
random.choice(string.ascii_uppercase + string.digits) for _ in range(8)
|
|
)
|
|
|
|
|
|
def get_supported_entitites(
|
|
controller: EcovacsController,
|
|
entity_class: type[EcovacsDescriptionEntity],
|
|
descriptions: tuple[EcovacsCapabilityEntityDescription, ...],
|
|
) -> list[EcovacsEntity]:
|
|
"""Return all supported entities for all devices."""
|
|
entities: list[EcovacsEntity] = []
|
|
|
|
for device in controller.devices:
|
|
for description in descriptions:
|
|
if capability := description.capability_fn(device.capabilities):
|
|
entities.append(entity_class(device, capability, description))
|
|
|
|
return entities
|