Migrate integrations f-h to extend SensorEntity (#48212)

This commit is contained in:
Erik Montnemery 2021-03-22 19:45:17 +01:00 committed by GitHub
parent a49989241a
commit 339a56e434
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
61 changed files with 154 additions and 167 deletions

View file

@ -4,6 +4,7 @@ from __future__ import annotations
from datetime import timedelta
from typing import Any
from homeassistant.components.sensor import SensorEntity
from homeassistant.const import DEVICE_CLASS_BATTERY, PERCENTAGE
from homeassistant.helpers.typing import ConfigType, HomeAssistantType
import homeassistant.util.dt as dt_util
@ -38,7 +39,7 @@ async def async_setup_platform(
async_add_entities(sensors + issues, update_before_add=True)
class GeniusBattery(GeniusDevice):
class GeniusBattery(GeniusDevice, SensorEntity):
"""Representation of a Genius Hub sensor."""
def __init__(self, broker, device, state_attr) -> None:
@ -88,7 +89,7 @@ class GeniusBattery(GeniusDevice):
return level if level != 255 else 0
class GeniusIssue(GeniusEntity):
class GeniusIssue(GeniusEntity, SensorEntity):
"""Representation of a Genius Hub sensor."""
def __init__(self, broker, level) -> None: