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

Fix: Shelly Gen2 - filter unsupported sensors (#57065)

Browse Source
This commit is contained in:
Shay Levy 2021-10-04 23:46:46 +03:00 committed by GitHub
parent 08cebb247f
commit c8dc5d15ee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 31 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
homeassistant/components/shelly/binary_sensor.py
View File

@ -122,34 +122,28 @@ REST_SENSORS: Final = {
RPC_SENSORS: Final = { RPC_SENSORS: Final = {
"input": RpcAttributeDescription( "input": RpcAttributeDescription(
key="input", key="input",
sub_key="state",
name="Input", name="Input",
value=lambda status, _: status["state"],
device_class=DEVICE_CLASS_POWER, device_class=DEVICE_CLASS_POWER,
default_enabled=False, default_enabled=False,
removal_condition=is_rpc_momentary_input, removal_condition=is_rpc_momentary_input,
), ),
"cloud": RpcAttributeDescription( "cloud": RpcAttributeDescription(
key="cloud", key="cloud",
sub_key="connected",
name="Cloud", name="Cloud",
value=lambda status, _: status["connected"],
device_class=DEVICE_CLASS_CONNECTIVITY, device_class=DEVICE_CLASS_CONNECTIVITY,
default_enabled=False, default_enabled=False,
), ),
"fwupdate": RpcAttributeDescription( "fwupdate": RpcAttributeDescription(
key="sys", key="sys",
sub_key="available_updates",
name="Firmware Update", name="Firmware Update",
device_class=DEVICE_CLASS_UPDATE, device_class=DEVICE_CLASS_UPDATE,
value=lambda status, _: status["available_updates"],
default_enabled=False, default_enabled=False,
extra_state_attributes=lambda status: { extra_state_attributes=lambda status: {
"latest_stable_version": status["available_updates"].get( "latest_stable_version": status.get("stable", {"version": ""})["version"],
"stable", "beta_version": status.get("beta", {"version": ""})["version"],
{"version": ""},
)["version"],
"beta_version": status["available_updates"].get(
"beta",
{"version": ""},
)["version"],
}, },
), ),
} }

19
homeassistant/components/shelly/entity.py
View File

@ -166,6 +166,10 @@ async def async_setup_entry_rpc(
key_instances = get_rpc_key_instances(wrapper.device.status, description.key) key_instances = get_rpc_key_instances(wrapper.device.status, description.key)
for key in key_instances: for key in key_instances:
# Filter non-existing sensors
if description.sub_key not in wrapper.device.status[key]:
continue
# Filter and remove entities that according to settings should not create an entity # Filter and remove entities that according to settings should not create an entity
if description.removal_condition and description.removal_condition( if description.removal_condition and description.removal_condition(
wrapper.device.config, key wrapper.device.config, key
@ -240,10 +244,11 @@ class RpcAttributeDescription:
"""Class to describe a RPC sensor.""" """Class to describe a RPC sensor."""
key: str key: str
sub_key: str
name: str name: str
icon: str | None = None icon: str | None = None
unit: str | None = None unit: str | None = None
value: Callable[[dict, Any], Any] | None = None value: Callable[[Any, Any], Any] | None = None
device_class: str | None = None device_class: str | None = None
state_class: str | None = None state_class: str | None = None
default_enabled: bool = True default_enabled: bool = True
@ -549,6 +554,7 @@ class ShellyRpcAttributeEntity(ShellyRpcEntity, entity.Entity):
) -> None: ) -> None:
"""Initialize sensor.""" """Initialize sensor."""
super().__init__(wrapper, key) super().__init__(wrapper, key)
self.sub_key = description.sub_key
self.attribute = attribute self.attribute = attribute
self.description = description self.description = description
@ -564,8 +570,11 @@ class ShellyRpcAttributeEntity(ShellyRpcEntity, entity.Entity):
"""Value of sensor.""" """Value of sensor."""
if callable(self.description.value): if callable(self.description.value):
self._last_value = self.description.value( self._last_value = self.description.value(
self.wrapper.device.status[self.key], self._last_value self.wrapper.device.status[self.key][self.sub_key], self._last_value
) )
else:
self._last_value = self.wrapper.device.status[self.key][self.sub_key]
return self._last_value return self._last_value
@property @property
@ -576,7 +585,9 @@ class ShellyRpcAttributeEntity(ShellyRpcEntity, entity.Entity):
if not available or not self.description.available: if not available or not self.description.available:
return available return available
return self.description.available(self.wrapper.device.status[self.key]) return self.description.available(
self.wrapper.device.status[self.key][self.sub_key]
)
@property @property
def extra_state_attributes(self) -> dict[str, Any] | None: def extra_state_attributes(self) -> dict[str, Any] | None:
@ -585,7 +596,7 @@ class ShellyRpcAttributeEntity(ShellyRpcEntity, entity.Entity):
return None return None
return self.description.extra_state_attributes( return self.description.extra_state_attributes(
self.wrapper.device.status[self.key] self.wrapper.device.status[self.key][self.sub_key]
) )

17
homeassistant/components/shelly/sensor.py
View File

@ -242,51 +242,56 @@ REST_SENSORS: Final = {
RPC_SENSORS: Final = { RPC_SENSORS: Final = {
"power": RpcAttributeDescription( "power": RpcAttributeDescription(
key="switch", key="switch",
sub_key="apower",
name="Power", name="Power",
unit=POWER_WATT, unit=POWER_WATT,
value=lambda status, _: round(float(status["apower"]), 1), value=lambda status, _: round(float(status), 1),
device_class=sensor.DEVICE_CLASS_POWER, device_class=sensor.DEVICE_CLASS_POWER,
state_class=sensor.STATE_CLASS_MEASUREMENT, state_class=sensor.STATE_CLASS_MEASUREMENT,
), ),
"voltage": RpcAttributeDescription( "voltage": RpcAttributeDescription(
key="switch", key="switch",
sub_key="voltage",
name="Voltage", name="Voltage",
unit=ELECTRIC_POTENTIAL_VOLT, unit=ELECTRIC_POTENTIAL_VOLT,
value=lambda status, _: round(float(status["voltage"]), 1), value=lambda status, _: round(float(status), 1),
device_class=sensor.DEVICE_CLASS_VOLTAGE, device_class=sensor.DEVICE_CLASS_VOLTAGE,
state_class=sensor.STATE_CLASS_MEASUREMENT, state_class=sensor.STATE_CLASS_MEASUREMENT,
default_enabled=False, default_enabled=False,
), ),
"energy": RpcAttributeDescription( "energy": RpcAttributeDescription(
key="switch", key="switch",
sub_key="aenergy",
name="Energy", name="Energy",
unit=ENERGY_KILO_WATT_HOUR, unit=ENERGY_KILO_WATT_HOUR,
value=lambda status, _: round(status["aenergy"]["total"] / 1000, 2), value=lambda status, _: round(status["total"] / 1000, 2),
device_class=sensor.DEVICE_CLASS_ENERGY, device_class=sensor.DEVICE_CLASS_ENERGY,
state_class=sensor.STATE_CLASS_TOTAL_INCREASING, state_class=sensor.STATE_CLASS_TOTAL_INCREASING,
), ),
"temperature": RpcAttributeDescription( "temperature": RpcAttributeDescription(
key="switch", key="switch",
sub_key="temperature",
name="Temperature", name="Temperature",
unit=TEMP_CELSIUS, unit=TEMP_CELSIUS,
value=lambda status, _: round(status["temperature"]["tC"], 1), value=lambda status, _: round(status["tC"], 1),
device_class=sensor.DEVICE_CLASS_TEMPERATURE, device_class=sensor.DEVICE_CLASS_TEMPERATURE,
state_class=sensor.STATE_CLASS_MEASUREMENT, state_class=sensor.STATE_CLASS_MEASUREMENT,
default_enabled=False, default_enabled=False,
), ),
"rssi": RpcAttributeDescription( "rssi": RpcAttributeDescription(
key="wifi", key="wifi",
sub_key="rssi",
name="RSSI", name="RSSI",
unit=SIGNAL_STRENGTH_DECIBELS_MILLIWATT, unit=SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
value=lambda status, _: status["rssi"],
device_class=sensor.DEVICE_CLASS_SIGNAL_STRENGTH, device_class=sensor.DEVICE_CLASS_SIGNAL_STRENGTH,
state_class=sensor.STATE_CLASS_MEASUREMENT, state_class=sensor.STATE_CLASS_MEASUREMENT,
default_enabled=False, default_enabled=False,
), ),
"uptime": RpcAttributeDescription( "uptime": RpcAttributeDescription(
key="sys", key="sys",
sub_key="uptime",
name="Uptime", name="Uptime",
value=lambda status, last: get_device_uptime(status["uptime"], last), value=get_device_uptime,
device_class=sensor.DEVICE_CLASS_TIMESTAMP, device_class=sensor.DEVICE_CLASS_TIMESTAMP,
default_enabled=False, default_enabled=False,
), ),