|
|
- """Discover Yeelight bulbs, based on Kodi discoverable."""
- from . import MDNSDiscoverable
- from ..const import ATTR_DEVICE_TYPE
-
- DEVICE_NAME_PREFIX = 'yeelink-light-'
-
-
- class Discoverable(MDNSDiscoverable):
- """Add support for discovering Yeelight."""
-
- def __init__(self, nd):
- """Initialize the Yeelight discovery."""
- super(Discoverable, self).__init__(nd, '_miio._udp.local.')
-
- def info_from_entry(self, entry):
- """Return most important info from mDNS entries."""
- info = super().info_from_entry(entry)
-
- # Example name: yeelink-light-ceiling4_mibt72799069._miio._udp.local.
- info[ATTR_DEVICE_TYPE] = \
- entry.name.replace(DEVICE_NAME_PREFIX, '').split('_', 1)[0]
-
- return info
-
- def get_entries(self):
- """ Return yeelight devices. """
- return self.find_by_device_name(DEVICE_NAME_PREFIX)
|
