61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
from homeassistant.components.sensor import SensorEntity
|
|
|
|
from .const import DOMAIN, DATA_ADDERS
|
|
from .helpers import BrowserModEntity
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass, config_entry, async_add_entities, discoveryInfo=None
|
|
):
|
|
hass.data[DOMAIN][DATA_ADDERS]["sensor"] = async_add_entities
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
await async_setup_platform(hass, {}, async_add_entities)
|
|
|
|
|
|
class BrowserSensor(BrowserModEntity, SensorEntity):
|
|
def __init__(
|
|
self,
|
|
coordinator,
|
|
browserID,
|
|
parameter,
|
|
name,
|
|
unit_of_measurement=None,
|
|
device_class=None,
|
|
):
|
|
BrowserModEntity.__init__(self, coordinator, browserID, name)
|
|
SensorEntity.__init__(self)
|
|
self.parameter = parameter
|
|
self._device_class = device_class
|
|
self._unit_of_measurement = unit_of_measurement
|
|
|
|
@property
|
|
def native_value(self):
|
|
data = self._data
|
|
data = data.get("browser", {})
|
|
data = data.get(self.parameter, None)
|
|
return data
|
|
|
|
@property
|
|
def device_class(self):
|
|
return self._device_class
|
|
|
|
@property
|
|
def native_unit_of_measurement(self):
|
|
return self._unit_of_measurement
|
|
|
|
@property
|
|
def extra_state_attributes(self):
|
|
retval = super().extra_state_attributes
|
|
if self.parameter == "currentUser":
|
|
retval["userData"] = self._data.get("browser", {}).get("userData")
|
|
if self.parameter == "path":
|
|
retval["pathSegments"] = (
|
|
self._data.get("browser", {}).get("path", "").split("/")
|
|
)
|
|
if self.parameter == "userAgent":
|
|
retval["userAgent"] = self._data.get("browser", {}).get("userAgent")
|
|
|
|
return retval
|