From 6a0f12ab2de03cdbcc283f84f43b8ee86cc171d6 Mon Sep 17 00:00:00 2001 From: Bas Nijholt Date: Mon, 7 Sep 2020 20:21:31 +0200 Subject: [PATCH] setup a sensor per profile --- .../circadian_lighting/sensor.py | 24 ++++++++----------- 1 file changed, 10 insertions(+), 14 deletions(-) diff --git a/custom_components/circadian_lighting/sensor.py b/custom_components/circadian_lighting/sensor.py index b98c026d..4461fa6a 100755 --- a/custom_components/circadian_lighting/sensor.py +++ b/custom_components/circadian_lighting/sensor.py @@ -13,20 +13,12 @@ ICON = "mdi:theme-light-dark" def setup_platform(hass, config, add_devices, discovery_info=None): """Set up the Circadian Lighting sensor.""" - circadian_lighting = hass.data.get(DOMAIN, {}).get(DEFAULT_PROFILE) - if circadian_lighting is not None: - sensor = CircadianSensor(hass, circadian_lighting) - add_devices([sensor], True) - - def update(call=None): - """Update component.""" - circadian_lighting.update() - - service_name = "values_update" - hass.services.register(DOMAIN, service_name, update) - return True - else: - return False + sensors = [ + CircadianSensor(hass, circadian_lighting) + for circadian_lighting in hass.data[DOMAIN].values() + ] + add_devices(sensors, True) + return True class CircadianSensor(Entity): @@ -37,6 +29,10 @@ class CircadianSensor(Entity): self._circadian_lighting = circadian_lighting self._name = "Circadian Values" self._entity_id = "sensor.circadian_values" + profile = circadian_lighting._profile + if profile != DEFAULT_PROFILE: + self._name += f" {profile}" + self._entity_id += f"_{profile.lower()}" self._unit_of_measurement = "%" self._icon = ICON