Add entity categories for appropriate RainMachine entities (#58107)

This commit is contained in:
Aaron Bach 2021-10-20 15:57:28 -06:00 committed by GitHub
parent a824fa9a7b
commit 745e42621b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 0 deletions

View File

@ -7,6 +7,7 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntityDescription,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -50,24 +51,28 @@ BINARY_SENSOR_DESCRIPTIONS = (
key=TYPE_FREEZE,
name="Freeze Restrictions",
icon="mdi:cancel",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
api_category=DATA_RESTRICTIONS_CURRENT,
),
RainMachineBinarySensorDescription(
key=TYPE_FREEZE_PROTECTION,
name="Freeze Protection",
icon="mdi:weather-snowy",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
api_category=DATA_RESTRICTIONS_UNIVERSAL,
),
RainMachineBinarySensorDescription(
key=TYPE_HOT_DAYS,
name="Extra Water on Hot Days",
icon="mdi:thermometer-lines",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
api_category=DATA_RESTRICTIONS_UNIVERSAL,
),
RainMachineBinarySensorDescription(
key=TYPE_HOURLY,
name="Hourly Restrictions",
icon="mdi:cancel",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
entity_registry_enabled_default=False,
api_category=DATA_RESTRICTIONS_CURRENT,
),
@ -75,6 +80,7 @@ BINARY_SENSOR_DESCRIPTIONS = (
key=TYPE_MONTH,
name="Month Restrictions",
icon="mdi:cancel",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
entity_registry_enabled_default=False,
api_category=DATA_RESTRICTIONS_CURRENT,
),
@ -82,6 +88,7 @@ BINARY_SENSOR_DESCRIPTIONS = (
key=TYPE_RAINDELAY,
name="Rain Delay Restrictions",
icon="mdi:cancel",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
entity_registry_enabled_default=False,
api_category=DATA_RESTRICTIONS_CURRENT,
),
@ -89,6 +96,7 @@ BINARY_SENSOR_DESCRIPTIONS = (
key=TYPE_RAINSENSOR,
name="Rain Sensor Restrictions",
icon="mdi:cancel",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
entity_registry_enabled_default=False,
api_category=DATA_RESTRICTIONS_CURRENT,
),
@ -96,6 +104,7 @@ BINARY_SENSOR_DESCRIPTIONS = (
key=TYPE_WEEKDAY,
name="Weekday Restrictions",
icon="mdi:cancel",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
entity_registry_enabled_default=False,
api_category=DATA_RESTRICTIONS_CURRENT,
),

View File

@ -13,6 +13,7 @@ from homeassistant.components.sensor import (
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
DEVICE_CLASS_TEMPERATURE,
ENTITY_CATEGORY_DIAGNOSTIC,
TEMP_CELSIUS,
VOLUME_CUBIC_METERS,
)
@ -49,6 +50,7 @@ SENSOR_DESCRIPTIONS = (
name="Flow Sensor Clicks per Cubic Meter",
icon="mdi:water-pump",
native_unit_of_measurement=f"clicks/{VOLUME_CUBIC_METERS}",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
entity_registry_enabled_default=False,
state_class=STATE_CLASS_MEASUREMENT,
api_category=DATA_PROVISION_SETTINGS,
@ -57,6 +59,7 @@ SENSOR_DESCRIPTIONS = (
key=TYPE_FLOW_SENSOR_CONSUMED_LITERS,
name="Flow Sensor Consumed Liters",
icon="mdi:water-pump",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
native_unit_of_measurement="liter",
entity_registry_enabled_default=False,
state_class=STATE_CLASS_TOTAL_INCREASING,
@ -66,6 +69,7 @@ SENSOR_DESCRIPTIONS = (
key=TYPE_FLOW_SENSOR_START_INDEX,
name="Flow Sensor Start Index",
icon="mdi:water-pump",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
native_unit_of_measurement="index",
entity_registry_enabled_default=False,
api_category=DATA_PROVISION_SETTINGS,
@ -74,6 +78,7 @@ SENSOR_DESCRIPTIONS = (
key=TYPE_FLOW_SENSOR_WATERING_CLICKS,
name="Flow Sensor Clicks",
icon="mdi:water-pump",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
native_unit_of_measurement="clicks",
entity_registry_enabled_default=False,
state_class=STATE_CLASS_MEASUREMENT,
@ -83,6 +88,7 @@ SENSOR_DESCRIPTIONS = (
key=TYPE_FREEZE_TEMP,
name="Freeze Protect Temperature",
icon="mdi:thermometer",
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
native_unit_of_measurement=TEMP_CELSIUS,
device_class=DEVICE_CLASS_TEMPERATURE,
state_class=STATE_CLASS_MEASUREMENT,