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

Use entity class vars in SolarEdge (#51123)

Browse Source
This commit is contained in:
Franck Nijhof 2021-05-28 08:24:55 +02:00 committed by GitHub
parent eb66f8ef6d
commit e9b09325c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
homeassistant/components/solaredge/sensor.py
View File

@ -127,20 +127,9 @@ class SolarEdgeSensor(CoordinatorEntity, SensorEntity):
self.sensor_key = sensor_key
self.data_service = data_service
@property
def unit_of_measurement(self) -> str | None:
"""Return the unit of measurement."""
return SENSOR_TYPES[self.sensor_key][2]
@property
def name(self) -> str:
"""Return the name."""
return f"{self.platform_name} ({SENSOR_TYPES[self.sensor_key][1]})"
@property
def icon(self) -> str | None:
"""Return the sensor icon."""
return SENSOR_TYPES[self.sensor_key][3]
self._attr_unit_of_measurement = SENSOR_TYPES[sensor_key][2]
self._attr_name = f"{platform_name} ({SENSOR_TYPES[sensor_key][1]})"
self._attr_icon = SENSOR_TYPES[sensor_key][3]
class SolarEdgeOverviewSensor(SolarEdgeSensor):
@ -202,6 +191,7 @@ class SolarEdgeEnergyDetailsSensor(SolarEdgeSensor):
super().__init__(platform_name, sensor_key, data_service)
self._json_key = SENSOR_TYPES[self.sensor_key][0]
self._attr_unit_of_measurement = data_service.unit
@property
def extra_state_attributes(self) -> dict[str, Any]:
@ -213,15 +203,12 @@ class SolarEdgeEnergyDetailsSensor(SolarEdgeSensor):
"""Return the state of the sensor."""
return self.data_service.data.get(self._json_key)
@property
def unit_of_measurement(self) -> str | None:
"""Return the unit of measurement."""
return self.data_service.unit
class SolarEdgePowerFlowSensor(SolarEdgeSensor):
"""Representation of an SolarEdge Monitoring API power flow sensor."""
_attr_device_class = DEVICE_CLASS_POWER
def __init__(
self, platform_name: str, sensor_key: str, data_service: SolarEdgeDataService
) -> None:
@ -229,11 +216,7 @@ class SolarEdgePowerFlowSensor(SolarEdgeSensor):
super().__init__(platform_name, sensor_key, data_service)
self._json_key = SENSOR_TYPES[self.sensor_key][0]
@property
def device_class(self) -> str:
"""Device Class."""
return DEVICE_CLASS_POWER
self._attr_unit_of_measurement = data_service.unit
@property
def extra_state_attributes(self) -> dict[str, Any]:
@ -245,15 +228,12 @@ class SolarEdgePowerFlowSensor(SolarEdgeSensor):
"""Return the state of the sensor."""
return self.data_service.data.get(self._json_key)
@property
def unit_of_measurement(self) -> str | None:
"""Return the unit of measurement."""
return self.data_service.unit
class SolarEdgeStorageLevelSensor(SolarEdgeSensor):
"""Representation of an SolarEdge Monitoring API storage level sensor."""
_attr_device_class = DEVICE_CLASS_BATTERY
def __init__(
self, platform_name: str, sensor_key: str, data_service: SolarEdgeDataService
) -> None:
@ -262,11 +242,6 @@ class SolarEdgeStorageLevelSensor(SolarEdgeSensor):
self._json_key = SENSOR_TYPES[self.sensor_key][0]
@property
def device_class(self) -> str:
"""Return the device_class of the device."""
return DEVICE_CLASS_BATTERY
@property
def state(self) -> str | None:
"""Return the state of the sensor."""