49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
"""The snapcast component."""
|
|
|
|
import asyncio
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
from homeassistant.helpers import config_validation as cv
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_send
|
|
|
|
DOMAIN = 'snapcast'
|
|
|
|
SERVICE_SNAPSHOT = 'snapshot'
|
|
SERVICE_RESTORE = 'restore'
|
|
SERVICE_JOIN = 'join'
|
|
SERVICE_UNJOIN = 'unjoin'
|
|
|
|
ATTR_MASTER = 'master'
|
|
|
|
SERVICE_SCHEMA = vol.Schema({
|
|
vol.Required(ATTR_ENTITY_ID): cv.entity_ids,
|
|
})
|
|
|
|
JOIN_SERVICE_SCHEMA = SERVICE_SCHEMA.extend({
|
|
vol.Required(ATTR_MASTER): cv.entity_id,
|
|
})
|
|
|
|
|
|
async def async_setup(hass, config):
|
|
"""Handle service configuration."""
|
|
service_event = asyncio.Event()
|
|
|
|
async def service_handle(service):
|
|
"""Dispatch a service call."""
|
|
service_event.clear()
|
|
async_dispatcher_send(hass, DOMAIN, service_event, service.service,
|
|
service.data)
|
|
await service_event.wait()
|
|
|
|
hass.services.async_register(
|
|
DOMAIN, SERVICE_SNAPSHOT, service_handle, schema=SERVICE_SCHEMA)
|
|
hass.services.async_register(
|
|
DOMAIN, SERVICE_RESTORE, service_handle, schema=SERVICE_SCHEMA)
|
|
hass.services.async_register(
|
|
DOMAIN, SERVICE_JOIN, service_handle, schema=JOIN_SERVICE_SCHEMA)
|
|
hass.services.async_register(
|
|
DOMAIN, SERVICE_UNJOIN, service_handle, schema=SERVICE_SCHEMA)
|
|
|
|
return True
|