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

Update Roborock entity categories (#100316)

Browse Source
This commit is contained in:
Paulus Schoutsen 2023-09-13 13:43:28 -04:00 committed by GitHub
parent 0d33cba823
commit 23a891ebb1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
homeassistant/components/roborock/select.py
View File

@ -7,6 +7,7 @@ from roborock.roborock_typing import RoborockCommand
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.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.util import slugify from homeassistant.util import slugify
@ -43,6 +44,7 @@ SELECT_DESCRIPTIONS: list[RoborockSelectDescription] = [
translation_key="mop_intensity", translation_key="mop_intensity",
api_command=RoborockCommand.SET_WATER_BOX_CUSTOM_MODE, api_command=RoborockCommand.SET_WATER_BOX_CUSTOM_MODE,
value_fn=lambda data: data.water_box_mode.name, value_fn=lambda data: data.water_box_mode.name,
entity_category=EntityCategory.CONFIG,
options_lambda=lambda data: data.water_box_mode.keys() options_lambda=lambda data: data.water_box_mode.keys()
if data.water_box_mode if data.water_box_mode
else None, else None,
@ -53,6 +55,7 @@ SELECT_DESCRIPTIONS: list[RoborockSelectDescription] = [
translation_key="mop_mode", translation_key="mop_mode",
api_command=RoborockCommand.SET_MOP_MODE, api_command=RoborockCommand.SET_MOP_MODE,
value_fn=lambda data: data.mop_mode.name, value_fn=lambda data: data.mop_mode.name,
entity_category=EntityCategory.CONFIG,
options_lambda=lambda data: data.mop_mode.keys() if data.mop_mode else None, options_lambda=lambda data: data.mop_mode.keys() if data.mop_mode else None,
parameter_lambda=lambda key, status: [status.mop_mode.as_dict().get(key)], parameter_lambda=lambda key, status: [status.mop_mode.as_dict().get(key)],
), ),

4
homeassistant/components/roborock/sensor.py
View File

@ -91,6 +91,7 @@ SENSOR_DESCRIPTIONS = [
translation_key="cleaning_time", translation_key="cleaning_time",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
value_fn=lambda data: data.status.clean_time, value_fn=lambda data: data.status.clean_time,
entity_category=EntityCategory.DIAGNOSTIC,
), ),
RoborockSensorDescription( RoborockSensorDescription(
native_unit_of_measurement=UnitOfTime.SECONDS, native_unit_of_measurement=UnitOfTime.SECONDS,
@ -99,6 +100,7 @@ SENSOR_DESCRIPTIONS = [
icon="mdi:history", icon="mdi:history",
device_class=SensorDeviceClass.DURATION, device_class=SensorDeviceClass.DURATION,
value_fn=lambda data: data.clean_summary.clean_time, value_fn=lambda data: data.clean_summary.clean_time,
entity_category=EntityCategory.DIAGNOSTIC,
), ),
RoborockSensorDescription( RoborockSensorDescription(
key="status", key="status",
@ -114,6 +116,7 @@ SENSOR_DESCRIPTIONS = [
icon="mdi:texture-box", icon="mdi:texture-box",
translation_key="cleaning_area", translation_key="cleaning_area",
value_fn=lambda data: data.status.square_meter_clean_area, value_fn=lambda data: data.status.square_meter_clean_area,
entity_category=EntityCategory.DIAGNOSTIC,
native_unit_of_measurement=AREA_SQUARE_METERS, native_unit_of_measurement=AREA_SQUARE_METERS,
), ),
RoborockSensorDescription( RoborockSensorDescription(
@ -121,6 +124,7 @@ SENSOR_DESCRIPTIONS = [
icon="mdi:texture-box", icon="mdi:texture-box",
translation_key="total_cleaning_area", translation_key="total_cleaning_area",
value_fn=lambda data: data.clean_summary.square_meter_clean_area, value_fn=lambda data: data.clean_summary.square_meter_clean_area,
entity_category=EntityCategory.DIAGNOSTIC,
native_unit_of_measurement=AREA_SQUARE_METERS, native_unit_of_measurement=AREA_SQUARE_METERS,
), ),
RoborockSensorDescription( RoborockSensorDescription(