mash-server/src/models/helper.py

from typing import TypeVar
from models.devices import GenericDevice, LightDevice, SwitchDevice
from models.exceptions import NoDeviceFoundError
from models.groups import DeviceGroup, Room


DEVICE_TYPE = TypeVar("DEVICE_TYPE", type(GenericDevice), type(SwitchDevice), type(LightDevice), type(Room), type(DeviceGroup))


def filter_devices(devices: list[GenericDevice], type: DEVICE_TYPE) -> list[DEVICE_TYPE]:
    """Filters out devices that are not of a specific type."""
    filtered_devices: list[type] = [device for device in devices if isinstance(device, type)]
    
    if len(filtered_devices) == 0:
        raise NoDeviceFoundError(f"No devices of type {type} found.")
    
    return filtered_devices