evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use Platform enum [emulated_hue] (#63498)

Browse source
This commit is contained in:
Marc Mueller 2022-01-05 20:27:01 +01:00 • committed by GitHub
parent c3a432620c
commit 4fd555f508
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

50
homeassistant/components/emulated_hue/hue_api.py
View file

@ -7,16 +7,7 @@ import logging
import time import time
from homeassistant import core from homeassistant import core
from homeassistant.components import ( from homeassistant.components import light, script
climate,
cover,
fan,
humidifier,
light,
media_player,
scene,
script,
)
from homeassistant.components.climate.const import ( from homeassistant.components.climate.const import (
SERVICE_SET_TEMPERATURE, SERVICE_SET_TEMPERATURE,
SUPPORT_TARGET_TEMPERATURE, SUPPORT_TARGET_TEMPERATURE,
@ -64,6 +55,7 @@ from homeassistant.const import (
STATE_OFF, STATE_OFF,
STATE_ON, STATE_ON,
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
Platform,
) )
from homeassistant.helpers.event import async_track_state_change_event from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.util.network import is_local from homeassistant.util.network import is_local
@ -347,7 +339,7 @@ class HueOneLightChangeView(HomeAssistantView):
# Get the entity's supported features # Get the entity's supported features
entity_features = entity.attributes.get(ATTR_SUPPORTED_FEATURES, 0) entity_features = entity.attributes.get(ATTR_SUPPORTED_FEATURES, 0)
if entity.domain == light.DOMAIN: if entity.domain == Platform.LIGHT:
color_modes = entity.attributes.get(light.ATTR_SUPPORTED_COLOR_MODES, []) color_modes = entity.attributes.get(light.ATTR_SUPPORTED_COLOR_MODES, [])
# Parse the request # Parse the request
@ -393,23 +385,23 @@ class HueOneLightChangeView(HomeAssistantView):
return self.json_message("Bad request", HTTPStatus.BAD_REQUEST) return self.json_message("Bad request", HTTPStatus.BAD_REQUEST)
if HUE_API_STATE_BRI in request_json: if HUE_API_STATE_BRI in request_json:
if entity.domain == light.DOMAIN: if entity.domain == Platform.LIGHT:
if light.brightness_supported(color_modes): if light.brightness_supported(color_modes):
parsed[STATE_ON] = parsed[STATE_BRIGHTNESS] > 0 parsed[STATE_ON] = parsed[STATE_BRIGHTNESS] > 0
else: else:
parsed[STATE_BRIGHTNESS] = None parsed[STATE_BRIGHTNESS] = None
elif entity.domain == scene.DOMAIN: elif entity.domain == Platform.SCENE:
parsed[STATE_BRIGHTNESS] = None parsed[STATE_BRIGHTNESS] = None
parsed[STATE_ON] = True parsed[STATE_ON] = True
elif entity.domain in [ elif entity.domain in [
script.DOMAIN, script.DOMAIN,
media_player.DOMAIN, Platform.MEDIA_PLAYER,
fan.DOMAIN, Platform.FAN,
cover.DOMAIN, Platform.COVER,
climate.DOMAIN, Platform.CLIMATE,
humidifier.DOMAIN, Platform.HUMIDIFIER,
]: ]:
# Convert 0-254 to 0-100 # Convert 0-254 to 0-100
level = (parsed[STATE_BRIGHTNESS] / HUE_API_STATE_BRI_MAX) * 100 level = (parsed[STATE_BRIGHTNESS] / HUE_API_STATE_BRI_MAX) * 100
@ -430,7 +422,7 @@ class HueOneLightChangeView(HomeAssistantView):
# If the requested entity is a light, set the brightness, hue, # If the requested entity is a light, set the brightness, hue,
# saturation and color temp # saturation and color temp
if entity.domain == light.DOMAIN: if entity.domain == Platform.LIGHT:
if parsed[STATE_ON]: if parsed[STATE_ON]:
if ( if (
light.brightness_supported(color_modes) light.brightness_supported(color_modes)
@ -483,7 +475,7 @@ class HueOneLightChangeView(HomeAssistantView):
data["variables"]["requested_level"] = parsed[STATE_BRIGHTNESS] data["variables"]["requested_level"] = parsed[STATE_BRIGHTNESS]
# If the requested entity is a climate, set the temperature # If the requested entity is a climate, set the temperature
elif entity.domain == climate.DOMAIN: elif entity.domain == Platform.CLIMATE:
# We don't support turning climate devices on or off, # We don't support turning climate devices on or off,
# only setting the temperature # only setting the temperature
service = None service = None
@ -497,7 +489,7 @@ class HueOneLightChangeView(HomeAssistantView):
data[ATTR_TEMPERATURE] = parsed[STATE_BRIGHTNESS] data[ATTR_TEMPERATURE] = parsed[STATE_BRIGHTNESS]
# If the requested entity is a humidifier, set the humidity # If the requested entity is a humidifier, set the humidity
elif entity.domain == humidifier.DOMAIN: elif entity.domain == Platform.HUMIDIFIER:
if parsed[STATE_BRIGHTNESS] is not None: if parsed[STATE_BRIGHTNESS] is not None:
turn_on_needed = True turn_on_needed = True
domain = entity.domain domain = entity.domain
@ -505,7 +497,7 @@ class HueOneLightChangeView(HomeAssistantView):
data[ATTR_HUMIDITY] = parsed[STATE_BRIGHTNESS] data[ATTR_HUMIDITY] = parsed[STATE_BRIGHTNESS]
# If the requested entity is a media player, convert to volume # If the requested entity is a media player, convert to volume
elif entity.domain == media_player.DOMAIN: elif entity.domain == Platform.MEDIA_PLAYER:
if ( if (
entity_features & SUPPORT_VOLUME_SET entity_features & SUPPORT_VOLUME_SET
and parsed[STATE_BRIGHTNESS] is not None and parsed[STATE_BRIGHTNESS] is not None
@ -517,7 +509,7 @@ class HueOneLightChangeView(HomeAssistantView):
data[ATTR_MEDIA_VOLUME_LEVEL] = parsed[STATE_BRIGHTNESS] / 100.0 data[ATTR_MEDIA_VOLUME_LEVEL] = parsed[STATE_BRIGHTNESS] / 100.0
# If the requested entity is a cover, convert to open_cover/close_cover # If the requested entity is a cover, convert to open_cover/close_cover
elif entity.domain == cover.DOMAIN: elif entity.domain == Platform.COVER:
domain = entity.domain domain = entity.domain
if service == SERVICE_TURN_ON: if service == SERVICE_TURN_ON:
service = SERVICE_OPEN_COVER service = SERVICE_OPEN_COVER
@ -534,7 +526,7 @@ class HueOneLightChangeView(HomeAssistantView):
# If the requested entity is a fan, convert to speed # If the requested entity is a fan, convert to speed
elif ( elif (
entity.domain == fan.DOMAIN entity.domain == Platform.FAN
and entity_features & SUPPORT_SET_SPEED and entity_features & SUPPORT_SET_SPEED
and parsed[STATE_BRIGHTNESS] is not None and parsed[STATE_BRIGHTNESS] is not None
): ):
@ -664,21 +656,21 @@ def get_entity_state(config, entity):
data[STATE_SATURATION] = 0 data[STATE_SATURATION] = 0
data[STATE_COLOR_TEMP] = 0 data[STATE_COLOR_TEMP] = 0
if entity.domain == climate.DOMAIN: if entity.domain == Platform.CLIMATE:
temperature = entity.attributes.get(ATTR_TEMPERATURE, 0) temperature = entity.attributes.get(ATTR_TEMPERATURE, 0)
# Convert 0-100 to 0-254 # Convert 0-100 to 0-254
data[STATE_BRIGHTNESS] = round(temperature * HUE_API_STATE_BRI_MAX / 100) data[STATE_BRIGHTNESS] = round(temperature * HUE_API_STATE_BRI_MAX / 100)
elif entity.domain == humidifier.DOMAIN: elif entity.domain == Platform.HUMIDIFIER:
humidity = entity.attributes.get(ATTR_HUMIDITY, 0) humidity = entity.attributes.get(ATTR_HUMIDITY, 0)
# Convert 0-100 to 0-254 # Convert 0-100 to 0-254
data[STATE_BRIGHTNESS] = round(humidity * HUE_API_STATE_BRI_MAX / 100) data[STATE_BRIGHTNESS] = round(humidity * HUE_API_STATE_BRI_MAX / 100)
elif entity.domain == media_player.DOMAIN: elif entity.domain == Platform.MEDIA_PLAYER:
level = entity.attributes.get( level = entity.attributes.get(
ATTR_MEDIA_VOLUME_LEVEL, 1.0 if data[STATE_ON] else 0.0 ATTR_MEDIA_VOLUME_LEVEL, 1.0 if data[STATE_ON] else 0.0
) )
# Convert 0.0-1.0 to 0-254 # Convert 0.0-1.0 to 0-254
data[STATE_BRIGHTNESS] = round(min(1.0, level) * HUE_API_STATE_BRI_MAX) data[STATE_BRIGHTNESS] = round(min(1.0, level) * HUE_API_STATE_BRI_MAX)
elif entity.domain == fan.DOMAIN: elif entity.domain == Platform.FAN:
speed = entity.attributes.get(ATTR_SPEED, 0) speed = entity.attributes.get(ATTR_SPEED, 0)
# Convert 0.0-1.0 to 0-254 # Convert 0.0-1.0 to 0-254
data[STATE_BRIGHTNESS] = 0 data[STATE_BRIGHTNESS] = 0
@ -688,7 +680,7 @@ def get_entity_state(config, entity):
data[STATE_BRIGHTNESS] = 170 data[STATE_BRIGHTNESS] = 170
elif speed == SPEED_HIGH: elif speed == SPEED_HIGH:
data[STATE_BRIGHTNESS] = HUE_API_STATE_BRI_MAX data[STATE_BRIGHTNESS] = HUE_API_STATE_BRI_MAX
elif entity.domain == cover.DOMAIN: elif entity.domain == Platform.COVER:
level = entity.attributes.get(ATTR_CURRENT_POSITION, 0) level = entity.attributes.get(ATTR_CURRENT_POSITION, 0)
data[STATE_BRIGHTNESS] = round(level / 100 * HUE_API_STATE_BRI_MAX) data[STATE_BRIGHTNESS] = round(level / 100 * HUE_API_STATE_BRI_MAX)
else: else: