Use shorthand attributes in Octoprint (#99623)

This commit is contained in:
Joost Lekkerkerker 2023-09-04 19:00:19 +02:00 committed by GitHub
parent 2391087836
commit cb5d4ee6fa
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 16 deletions

View file

@ -52,11 +52,7 @@ class OctoPrintBinarySensorBase(
self._device_id = device_id
self._attr_name = f"OctoPrint {sensor_type}"
self._attr_unique_id = f"{sensor_type}-{device_id}"
@property
def device_info(self):
"""Device info."""
return self.coordinator.device_info
self._attr_device_info = coordinator.device_info
@property
def is_on(self):

View file

@ -5,7 +5,6 @@ from homeassistant.components.button import ButtonEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
@ -53,11 +52,7 @@ class OctoprintButton(CoordinatorEntity[OctoprintDataUpdateCoordinator], ButtonE
self._device_id = device_id
self._attr_name = f"OctoPrint {button_type}"
self._attr_unique_id = f"{button_type}-{device_id}"
@property
def device_info(self) -> DeviceInfo:
"""Device info."""
return self.coordinator.device_info
self._attr_device_info = coordinator.device_info
@property
def available(self) -> bool:

View file

@ -104,11 +104,7 @@ class OctoPrintSensorBase(
self._device_id = device_id
self._attr_name = f"OctoPrint {sensor_type}"
self._attr_unique_id = f"{sensor_type}-{device_id}"
@property
def device_info(self):
"""Device info."""
return self.coordinator.device_info
self._attr_device_info = coordinator.device_info
class OctoPrintStatusSensor(OctoPrintSensorBase):