Set enity_category to config for airgradient select entities (#118477)

pull/118845/head
Robert Resch 2024-05-30 14:20:02 +02:00 committed by Franck Nijhof
parent 50acc26812
commit 522152e7d2
No known key found for this signature in database
GPG Key ID: D62583BA8AB11CA3
2 changed files with 6 additions and 3 deletions

View File

@ -8,6 +8,7 @@ from airgradient.models import ConfigurationControl, TemperatureUnit
from homeassistant.components.select import SelectEntity, SelectEntityDescription from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ServiceValidationError from homeassistant.exceptions import ServiceValidationError
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -30,6 +31,7 @@ CONFIG_CONTROL_ENTITY = AirGradientSelectEntityDescription(
key="configuration_control", key="configuration_control",
translation_key="configuration_control", translation_key="configuration_control",
options=[x.value for x in ConfigurationControl], options=[x.value for x in ConfigurationControl],
entity_category=EntityCategory.CONFIG,
value_fn=lambda config: config.configuration_control, value_fn=lambda config: config.configuration_control,
set_value_fn=lambda client, value: client.set_configuration_control( set_value_fn=lambda client, value: client.set_configuration_control(
ConfigurationControl(value) ConfigurationControl(value)
@ -41,6 +43,7 @@ PROTECTED_SELECT_TYPES: tuple[AirGradientSelectEntityDescription, ...] = (
key="display_temperature_unit", key="display_temperature_unit",
translation_key="display_temperature_unit", translation_key="display_temperature_unit",
options=[x.value for x in TemperatureUnit], options=[x.value for x in TemperatureUnit],
entity_category=EntityCategory.CONFIG,
value_fn=lambda config: config.temperature_unit, value_fn=lambda config: config.temperature_unit,
set_value_fn=lambda client, value: client.set_temperature_unit( set_value_fn=lambda client, value: client.set_temperature_unit(
TemperatureUnit(value) TemperatureUnit(value)

View File

@ -16,7 +16,7 @@
'device_id': <ANY>, 'device_id': <ANY>,
'disabled_by': None, 'disabled_by': None,
'domain': 'select', 'domain': 'select',
'entity_category': None, 'entity_category': <EntityCategory.CONFIG: 'config'>,
'entity_id': 'select.airgradient_configuration_source', 'entity_id': 'select.airgradient_configuration_source',
'has_entity_name': True, 'has_entity_name': True,
'hidden_by': None, 'hidden_by': None,
@ -72,7 +72,7 @@
'device_id': <ANY>, 'device_id': <ANY>,
'disabled_by': None, 'disabled_by': None,
'domain': 'select', 'domain': 'select',
'entity_category': None, 'entity_category': <EntityCategory.CONFIG: 'config'>,
'entity_id': 'select.airgradient_display_temperature_unit', 'entity_id': 'select.airgradient_display_temperature_unit',
'has_entity_name': True, 'has_entity_name': True,
'hidden_by': None, 'hidden_by': None,
@ -128,7 +128,7 @@
'device_id': <ANY>, 'device_id': <ANY>,
'disabled_by': None, 'disabled_by': None,
'domain': 'select', 'domain': 'select',
'entity_category': None, 'entity_category': <EntityCategory.CONFIG: 'config'>,
'entity_id': 'select.airgradient_configuration_source', 'entity_id': 'select.airgradient_configuration_source',
'has_entity_name': True, 'has_entity_name': True,
'hidden_by': None, 'hidden_by': None,