Refactoring and cleanup

This commit is contained in:
2022-07-25 22:03:54 +00:00
parent ae916a3900
commit 947735292a
16 changed files with 132 additions and 144 deletions

View File

@@ -1,7 +1,7 @@
from homeassistant.components.binary_sensor import BinarySensorEntity
from .const import DOMAIN, DATA_ADDERS
from .helpers import BrowserModEntity
from .const import DATA_BROWSERS, DOMAIN, DATA_ADDERS
from .entities import BrowserModEntity
async def async_setup_platform(
@@ -22,10 +22,7 @@ class BrowserBinarySensor(BrowserModEntity, BinarySensorEntity):
@property
def is_on(self):
data = self._data
data = data.get("browser", {})
data = data.get(self.parameter, None)
return data
return self._data.get(DATA_BROWSERS, {}).get(self.parameter, None)
class ActivityBinarySensor(BrowserModEntity, BinarySensorEntity):
@@ -47,6 +44,4 @@ class ActivityBinarySensor(BrowserModEntity, BinarySensorEntity):
@property
def is_on(self):
data = self._data
data = data.get("activity", False)
return data
return self._data.get("activity", False)