131 lines
3.3 KiB
Python
131 lines
3.3 KiB
Python
"""Asuswrt status sensors."""
|
|
import logging
|
|
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
from . import DATA_ASUSWRT
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass, config, add_entities, discovery_info=None):
|
|
"""Set up the asuswrt sensors."""
|
|
if discovery_info is None:
|
|
return
|
|
|
|
api = hass.data[DATA_ASUSWRT]
|
|
|
|
devices = []
|
|
|
|
if 'download' in discovery_info:
|
|
devices.append(AsuswrtTotalRXSensor(api))
|
|
if 'upload' in discovery_info:
|
|
devices.append(AsuswrtTotalTXSensor(api))
|
|
if 'download_speed' in discovery_info:
|
|
devices.append(AsuswrtRXSensor(api))
|
|
if 'upload_speed' in discovery_info:
|
|
devices.append(AsuswrtTXSensor(api))
|
|
|
|
add_entities(devices)
|
|
|
|
|
|
class AsuswrtSensor(Entity):
|
|
"""Representation of a asuswrt sensor."""
|
|
|
|
_name = 'generic'
|
|
|
|
def __init__(self, api):
|
|
"""Initialize the sensor."""
|
|
self._api = api
|
|
self._state = None
|
|
self._rates = None
|
|
self._speed = None
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the sensor."""
|
|
return self._name
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the state of the sensor."""
|
|
return self._state
|
|
|
|
async def async_update(self):
|
|
"""Fetch status from asuswrt."""
|
|
self._rates = await self._api.async_get_bytes_total()
|
|
self._speed = await self._api.async_get_current_transfer_rates()
|
|
|
|
|
|
class AsuswrtRXSensor(AsuswrtSensor):
|
|
"""Representation of a asuswrt download speed sensor."""
|
|
|
|
_name = 'Asuswrt Download Speed'
|
|
_unit = 'Mbit/s'
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return the unit of measurement."""
|
|
return self._unit
|
|
|
|
async def async_update(self):
|
|
"""Fetch new state data for the sensor."""
|
|
await super().async_update()
|
|
if self._speed:
|
|
self._state = round(self._speed[0] / 125000, 2)
|
|
|
|
|
|
class AsuswrtTXSensor(AsuswrtSensor):
|
|
"""Representation of a asuswrt upload speed sensor."""
|
|
|
|
_name = 'Asuswrt Upload Speed'
|
|
_unit = 'Mbit/s'
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return the unit of measurement."""
|
|
return self._unit
|
|
|
|
async def async_update(self):
|
|
"""Fetch new state data for the sensor."""
|
|
await super().async_update()
|
|
if self._speed:
|
|
self._state = round(self._speed[1] / 125000, 2)
|
|
|
|
|
|
class AsuswrtTotalRXSensor(AsuswrtSensor):
|
|
"""Representation of a asuswrt total download sensor."""
|
|
|
|
_name = 'Asuswrt Download'
|
|
_unit = 'Gigabyte'
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return the unit of measurement."""
|
|
return self._unit
|
|
|
|
async def async_update(self):
|
|
"""Fetch new state data for the sensor."""
|
|
await super().async_update()
|
|
if self._rates:
|
|
self._state = round(self._rates[0] / 1000000000, 1)
|
|
|
|
|
|
class AsuswrtTotalTXSensor(AsuswrtSensor):
|
|
"""Representation of a asuswrt total upload sensor."""
|
|
|
|
_name = 'Asuswrt Upload'
|
|
_unit = 'Gigabyte'
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return the unit of measurement."""
|
|
return self._unit
|
|
|
|
async def async_update(self):
|
|
"""Fetch new state data for the sensor."""
|
|
await super().async_update()
|
|
if self._rates:
|
|
self._state = round(self._rates[1] / 1000000000, 1)
|