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

Add device info to transmission (#84660)

Browse Source
This commit is contained in:
avee87 2022-12-31 12:20:44 +00:00 committed by GitHub
parent fd78373b5c
commit 2dd9229dc4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/transmission/sensor.py
View File

@ -9,7 +9,9 @@ from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, STATE_IDLE, UnitOfDataRate from homeassistant.const import CONF_NAME, STATE_IDLE, UnitOfDataRate
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import TransmissionClient from . import TransmissionClient
@ -58,6 +60,12 @@ class TransmissionSensor(SensorEntity):
self._name = sensor_name self._name = sensor_name
self._sub_type = sub_type self._sub_type = sub_type
self._state = None self._state = None
self._attr_device_info = DeviceInfo(
entry_type=DeviceEntryType.SERVICE,
identifiers={(DOMAIN, tm_client.config_entry.entry_id)},
manufacturer="Transmission",
name=client_name,
)
@property @property
def name(self): def name(self):

12
homeassistant/components/transmission/switch.py
View File

@ -6,7 +6,9 @@ from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME, STATE_OFF, STATE_ON from homeassistant.const import CONF_NAME, STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import DOMAIN, SWITCH_TYPES from .const import DOMAIN, SWITCH_TYPES
@ -36,15 +38,21 @@ class TransmissionSwitch(SwitchEntity):
_attr_should_poll = False _attr_should_poll = False
def __init__(self, switch_type, switch_name, tm_client, name): def __init__(self, switch_type, switch_name, tm_client, client_name):
"""Initialize the Transmission switch.""" """Initialize the Transmission switch."""
self._name = switch_name self._name = switch_name
self.client_name = name self.client_name = client_name
self.type = switch_type self.type = switch_type
self._tm_client = tm_client self._tm_client = tm_client
self._state = STATE_OFF self._state = STATE_OFF
self._data = None self._data = None
self.unsub_update = None self.unsub_update = None
self._attr_device_info = DeviceInfo(
entry_type=DeviceEntryType.SERVICE,
identifiers={(DOMAIN, tm_client.config_entry.entry_id)},
manufacturer="Transmission",
name=client_name,
)
@property @property
def name(self): def name(self):