98 lines
3.0 KiB
Python
98 lines
3.0 KiB
Python
"""Support for LIFX Cloud scenes."""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from http import HTTPStatus
|
|
import logging
|
|
from typing import Any
|
|
|
|
import aiohttp
|
|
from aiohttp.hdrs import AUTHORIZATION
|
|
import async_timeout
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.scene import Scene
|
|
from homeassistant.const import CONF_PLATFORM, CONF_TIMEOUT, CONF_TOKEN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DEFAULT_TIMEOUT = 10
|
|
|
|
PLATFORM_SCHEMA = vol.Schema(
|
|
{
|
|
vol.Required(CONF_PLATFORM): "lifx_cloud",
|
|
vol.Required(CONF_TOKEN): cv.string,
|
|
vol.Optional(CONF_TIMEOUT, default=DEFAULT_TIMEOUT): cv.positive_int,
|
|
}
|
|
)
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass: HomeAssistant,
|
|
config: ConfigType,
|
|
async_add_entities: AddEntitiesCallback,
|
|
discovery_info: DiscoveryInfoType | None = None,
|
|
) -> None:
|
|
"""Set up the scenes stored in the LIFX Cloud."""
|
|
token = config.get(CONF_TOKEN)
|
|
timeout = config.get(CONF_TIMEOUT)
|
|
|
|
headers = {AUTHORIZATION: f"Bearer {token}"}
|
|
|
|
url = "https://api.lifx.com/v1/scenes"
|
|
|
|
try:
|
|
httpsession = async_get_clientsession(hass)
|
|
async with async_timeout.timeout(timeout):
|
|
scenes_resp = await httpsession.get(url, headers=headers)
|
|
|
|
except (asyncio.TimeoutError, aiohttp.ClientError):
|
|
_LOGGER.exception("Error on %s", url)
|
|
return
|
|
|
|
status = scenes_resp.status
|
|
if status == HTTPStatus.OK:
|
|
data = await scenes_resp.json()
|
|
devices = [LifxCloudScene(hass, headers, timeout, scene) for scene in data]
|
|
async_add_entities(devices)
|
|
return
|
|
if status == HTTPStatus.UNAUTHORIZED:
|
|
_LOGGER.error("Unauthorized (bad token?) on %s", url)
|
|
return
|
|
|
|
_LOGGER.error("HTTP error %d on %s", scenes_resp.status, url)
|
|
|
|
|
|
class LifxCloudScene(Scene):
|
|
"""Representation of a LIFX Cloud scene."""
|
|
|
|
def __init__(self, hass, headers, timeout, scene_data):
|
|
"""Initialize the scene."""
|
|
self.hass = hass
|
|
self._headers = headers
|
|
self._timeout = timeout
|
|
self._name = scene_data["name"]
|
|
self._uuid = scene_data["uuid"]
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the scene."""
|
|
return self._name
|
|
|
|
async def async_activate(self, **kwargs: Any) -> None:
|
|
"""Activate the scene."""
|
|
url = f"https://api.lifx.com/v1/scenes/scene_id:{self._uuid}/activate"
|
|
|
|
try:
|
|
httpsession = async_get_clientsession(self.hass)
|
|
async with async_timeout.timeout(self._timeout):
|
|
await httpsession.put(url, headers=self._headers)
|
|
|
|
except (asyncio.TimeoutError, aiohttp.ClientError):
|
|
_LOGGER.exception("Error on %s", url)
|