From 73226ba138538b5688693ab8c81d1b0e9a3cd74f Mon Sep 17 00:00:00 2001 From: Parker Brown <17183625+parkerbxyz@users.noreply.github.com> Date: Thu, 6 Feb 2025 02:48:41 +0000 Subject: [PATCH] Use Color enum for status options --- homeassistant/components/aranet/sensor.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/aranet/sensor.py b/homeassistant/components/aranet/sensor.py index 17a720f13b2..1b9552f3641 100644 --- a/homeassistant/components/aranet/sensor.py +++ b/homeassistant/components/aranet/sensor.py @@ -5,7 +5,7 @@ from __future__ import annotations from dataclasses import dataclass from typing import Any -from aranet4.client import Aranet4Advertisement +from aranet4.client import Aranet4Advertisement, Color from bleak.backends.device import BLEDevice from homeassistant.components.bluetooth.passive_update_processor import ( @@ -78,7 +78,7 @@ SENSOR_DESCRIPTIONS = { key="status", name="Threshold Level", device_class=SensorDeviceClass.ENUM, - options=["green", "yellow", "red"], + options=[status.name for status in Color], ), "co2": AranetSensorEntityDescription( key="co2", @@ -167,7 +167,10 @@ def sensor_update_to_bluetooth_data_update( val = getattr(adv.readings, key) if val == -1: continue - val *= desc.scale + if key == "status": + val = val.name # Use the name of the status + else: + val *= desc.scale data[tag] = val names[tag] = desc.name descs[tag] = desc