89 lines
2.5 KiB
Python
89 lines
2.5 KiB
Python
"""
|
|
Starts a service to scan in intervals for new devices.
|
|
|
|
Will emit EVENT_SERVICE_DISCOVERED whenever a new service has been discovered.
|
|
|
|
Knows which components handle certain types, will make sure they are
|
|
loaded before the EVENT_SERVICE_DISCOVERED is fired.
|
|
|
|
"""
|
|
import logging
|
|
import threading
|
|
|
|
# pylint: disable=no-name-in-module, import-error
|
|
from homeassistant.external.netdisco.netdisco import DiscoveryService
|
|
import homeassistant.external.netdisco.netdisco.const as services
|
|
|
|
from homeassistant import bootstrap
|
|
from homeassistant.const import EVENT_HOMEASSISTANT_START, ATTR_SERVICE
|
|
|
|
DOMAIN = "discovery"
|
|
DEPENDENCIES = []
|
|
|
|
EVENT_SERVICE_DISCOVERED = "service_discovered"
|
|
|
|
ATTR_DISCOVERED = "discovered"
|
|
|
|
SCAN_INTERVAL = 300 # seconds
|
|
|
|
SERVICE_HANDLERS = {
|
|
services.BELKIN_WEMO: "switch",
|
|
services.GOOGLE_CAST: "chromecast",
|
|
services.PHILIPS_HUE: "light",
|
|
}
|
|
|
|
|
|
def listen(hass, service, callback):
|
|
"""
|
|
Setup listener for discovery of specific service.
|
|
Service can be a string or a list/tuple.
|
|
"""
|
|
|
|
if not isinstance(service, str):
|
|
service = (service,)
|
|
|
|
def discovery_event_listener(event):
|
|
""" Listens for discovery events. """
|
|
if event.data[ATTR_SERVICE] in service:
|
|
callback(event.data[ATTR_SERVICE], event.data[ATTR_DISCOVERED])
|
|
|
|
hass.bus.listen(EVENT_SERVICE_DISCOVERED, discovery_event_listener)
|
|
|
|
|
|
def setup(hass, config):
|
|
""" Starts a discovery service. """
|
|
|
|
# Disable zeroconf logging, it spams
|
|
logging.getLogger('zeroconf').setLevel(logging.CRITICAL)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
lock = threading.Lock()
|
|
|
|
def new_service_listener(service, info):
|
|
""" Called when a new service is found. """
|
|
with lock:
|
|
component = SERVICE_HANDLERS.get(service)
|
|
|
|
logger.info("Found new service: %s %s", service, info)
|
|
|
|
if component and component not in hass.components:
|
|
if bootstrap.setup_component(hass, component, config):
|
|
hass.pool.add_worker()
|
|
|
|
hass.bus.fire(EVENT_SERVICE_DISCOVERED, {
|
|
ATTR_SERVICE: service,
|
|
ATTR_DISCOVERED: info
|
|
})
|
|
|
|
# pylint: disable=unused-argument
|
|
def start_discovery(event):
|
|
""" Start discovering. """
|
|
netdisco = DiscoveryService(SCAN_INTERVAL)
|
|
netdisco.add_listener(new_service_listener)
|
|
netdisco.start()
|
|
|
|
hass.bus.listen_once(EVENT_HOMEASSISTANT_START, start_discovery)
|
|
|
|
return True
|