Cleanup modbus binary_sensor signature (#49809)

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
jan iversen 2021-05-05 00:29:18 +02:00 committed by GitHub
parent 98edd58c55
commit 2f89ba24b6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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