74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
"""Services for the HEOS integration."""
|
|
import functools
|
|
import logging
|
|
|
|
from pyheos import CommandFailedError, Heos, HeosError, const
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.helpers import config_validation as cv
|
|
from homeassistant.helpers.typing import HomeAssistantType
|
|
|
|
from .const import (
|
|
ATTR_PASSWORD,
|
|
ATTR_USERNAME,
|
|
DOMAIN,
|
|
SERVICE_SIGN_IN,
|
|
SERVICE_SIGN_OUT,
|
|
)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
HEOS_SIGN_IN_SCHEMA = vol.Schema(
|
|
{vol.Required(ATTR_USERNAME): cv.string, vol.Required(ATTR_PASSWORD): cv.string}
|
|
)
|
|
|
|
HEOS_SIGN_OUT_SCHEMA = vol.Schema({})
|
|
|
|
|
|
def register(hass: HomeAssistantType, controller: Heos):
|
|
"""Register HEOS services."""
|
|
hass.services.async_register(
|
|
DOMAIN,
|
|
SERVICE_SIGN_IN,
|
|
functools.partial(_sign_in_handler, controller),
|
|
schema=HEOS_SIGN_IN_SCHEMA,
|
|
)
|
|
hass.services.async_register(
|
|
DOMAIN,
|
|
SERVICE_SIGN_OUT,
|
|
functools.partial(_sign_out_handler, controller),
|
|
schema=HEOS_SIGN_OUT_SCHEMA,
|
|
)
|
|
|
|
|
|
def remove(hass: HomeAssistantType):
|
|
"""Unregister HEOS services."""
|
|
hass.services.async_remove(DOMAIN, SERVICE_SIGN_IN)
|
|
hass.services.async_remove(DOMAIN, SERVICE_SIGN_OUT)
|
|
|
|
|
|
async def _sign_in_handler(controller, service):
|
|
"""Sign in to the HEOS account."""
|
|
if controller.connection_state != const.STATE_CONNECTED:
|
|
_LOGGER.error("Unable to sign in because HEOS is not connected")
|
|
return
|
|
username = service.data[ATTR_USERNAME]
|
|
password = service.data[ATTR_PASSWORD]
|
|
try:
|
|
await controller.sign_in(username, password)
|
|
except CommandFailedError as err:
|
|
_LOGGER.error("Sign in failed: %s", err)
|
|
except HeosError as err:
|
|
_LOGGER.error("Unable to sign in: %s", err)
|
|
|
|
|
|
async def _sign_out_handler(controller, service):
|
|
"""Sign out of the HEOS account."""
|
|
if controller.connection_state != const.STATE_CONNECTED:
|
|
_LOGGER.error("Unable to sign out because HEOS is not connected")
|
|
return
|
|
try:
|
|
await controller.sign_out()
|
|
except HeosError as err:
|
|
_LOGGER.error("Unable to sign out: %s", err)
|