jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-30 16:57:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Simplify get_friendly_status()

Browse Source
This commit is contained in:
Parker Brown 2025-02-06 18:18:56 +00:00 committed by GitHub
parent cbb072ad43
commit cd71145af7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
homeassistant/components/aranet/sensor.py
View File

@ -5,7 +5,7 @@ from __future__ import annotations
from dataclasses import dataclass from dataclasses import dataclass
from typing import Any from typing import Any
from aranet4.client import Aranet4Advertisement from aranet4.client import Aranet4Advertisement, AranetType, Color
from bleak.backends.device import BLEDevice from bleak.backends.device import BLEDevice
from homeassistant.components.bluetooth.passive_update_processor import ( from homeassistant.components.bluetooth.passive_update_processor import (
@ -167,7 +167,7 @@ def sensor_update_to_bluetooth_data_update(
if val == -1: if val == -1:
continue continue
val = ( val = (
get_friendly_status(val.name, adv.readings.type.name) get_friendly_status(val.name, adv.readings.type)
if key == "status" if key == "status"
else val * desc.scale else val * desc.scale
) )
@ -224,24 +224,15 @@ class Aranet4BluetoothSensorEntity(
return self.processor.entity_data.get(self.entity_key) return self.processor.entity_data.get(self.entity_key)
def get_friendly_status(status: str, device_type: str) -> str: def get_friendly_status(status: Color, device_type: AranetType) -> str:
"""Convert device status color code to a more human-friendly status based on device type.""" """Map status code to human-readable status based on device type."""
base_status_map = { status_map = {
"ERROR": "Error", "ERROR": "Error",
"RED": "Unhealthy", "RED": "Unhealthy",
(device_type.ARANET4, "GREEN"): "Good",
(device_type.ARANET4, "YELLOW"): "Average",
(device_type.ARANET_RADON, "GREEN"): "Normal",
(device_type.ARANET_RADON, "YELLOW"): "Elevated",
} }
device_status_map = { return status_map.get((device_type, status), status_map.get(status, status))
"ARANET4": {
**base_status_map,
"GREEN": "Good",
"YELLOW": "Average",
},
"ARANET_RADON": {
**base_status_map,
"GREEN": "Normal",
"YELLOW": "Elevated",
},
}
return device_status_map.get(device_type, {}).get(status, status)