Cleanup modbus binary_sensor signature (#49809)
Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
parent
98edd58c55
commit
2f89ba24b6
1 changed files with 10 additions and 21 deletions
|
@ -91,17 +91,7 @@ async def async_setup_platform(
|
|||
hub = hass.data[MODBUS_DOMAIN][discovery_info[CONF_NAME]]
|
||||
if CONF_SCAN_INTERVAL not in entry:
|
||||
entry[CONF_SCAN_INTERVAL] = DEFAULT_SCAN_INTERVAL
|
||||
sensors.append(
|
||||
ModbusBinarySensor(
|
||||
hub,
|
||||
entry[CONF_NAME],
|
||||
entry.get(CONF_SLAVE),
|
||||
entry[CONF_ADDRESS],
|
||||
entry.get(CONF_DEVICE_CLASS),
|
||||
entry[CONF_INPUT_TYPE],
|
||||
entry[CONF_SCAN_INTERVAL],
|
||||
)
|
||||
)
|
||||
sensors.append(ModbusBinarySensor(hub, hass, entry))
|
||||
|
||||
async_add_entities(sensors)
|
||||
|
||||
|
@ -109,24 +99,23 @@ async def async_setup_platform(
|
|||
class ModbusBinarySensor(BinarySensorEntity):
|
||||
"""Modbus binary sensor."""
|
||||
|
||||
def __init__(
|
||||
self, hub, name, slave, address, device_class, input_type, scan_interval
|
||||
):
|
||||
def __init__(self, hub, hass, entry):
|
||||
"""Initialize the Modbus binary sensor."""
|
||||
self._hub = hub
|
||||
self._name = name
|
||||
self._slave = int(slave) if slave else None
|
||||
self._address = int(address)
|
||||
self._device_class = device_class
|
||||
self._input_type = input_type
|
||||
self._hass = hass
|
||||
self._name = entry[CONF_NAME]
|
||||
self._slave = entry.get(CONF_SLAVE)
|
||||
self._address = int(entry[CONF_ADDRESS])
|
||||
self._device_class = entry.get(CONF_DEVICE_CLASS)
|
||||
self._input_type = entry[CONF_INPUT_TYPE]
|
||||
self._value = None
|
||||
self._available = True
|
||||
self._scan_interval = timedelta(seconds=scan_interval)
|
||||
self._scan_interval = timedelta(seconds=entry[CONF_SCAN_INTERVAL])
|
||||
|
||||
async def async_added_to_hass(self):
|
||||
"""Handle entity which will be added."""
|
||||
async_track_time_interval(
|
||||
self.hass, lambda arg: self.update(), self._scan_interval
|
||||
self._hass, lambda arg: self.update(), self._scan_interval
|
||||
)
|
||||
|
||||
@property
|
||||
|
|
Loading…
Add table
Reference in a new issue