38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
|
import logging
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
from homeassistant.const import STATE_UNAVAILABLE
|
||
|
|
|
||
|
|
from .helpers import setup_platform, BrowserModEntity
|
||
|
|
|
||
|
|
PLATFORM = 'sensor'
|
||
|
|
|
||
|
|
async def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
|
||
|
|
return setup_platform(hass, config, async_add_devices, PLATFORM, BrowserModSensor)
|
||
|
|
|
||
|
|
class BrowserModSensor(BrowserModEntity):
|
||
|
|
domain = PLATFORM
|
||
|
|
|
||
|
|
def __init__(self, hass, connection, deviceID, alias=None):
|
||
|
|
super().__init__(hass, connection, deviceID, alias)
|
||
|
|
self.last_seen = None
|
||
|
|
|
||
|
|
def updated(self):
|
||
|
|
self.last_seen = datetime.now()
|
||
|
|
self.schedule_update_ha_state()
|
||
|
|
|
||
|
|
@property
|
||
|
|
def state(self):
|
||
|
|
if not self.connection.connection:
|
||
|
|
return STATE_UNAVAILABLE
|
||
|
|
return len(self.connection.connection)
|
||
|
|
|
||
|
|
@property
|
||
|
|
def device_state_attributes(self):
|
||
|
|
return {
|
||
|
|
"type": "browser_mod",
|
||
|
|
"last_seen": self.last_seen,
|
||
|
|
"deviceID": self.deviceID,
|
||
|
|
**self.data
|
||
|
|
}
|