evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Migrate integrations i-m to extend SensorEntity (#48213)

Browse source
This commit is contained in:
Erik Montnemery 2021-03-22 19:59:03 +01:00 • committed by GitHub
parent 64bc9a8196
commit fdf97eaca3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
62 changed files with 152 additions and 157 deletions
Download patch file Download diff file Expand all files Collapse all files

9
homeassistant/components/loopenergy/sensor.py
View file

@ -4,14 +4,13 @@ import logging
import pyloopenergy
import voluptuous as vol
from homeassistant.components.sensor import PLATFORM_SCHEMA
from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity
from homeassistant.const import (
CONF_UNIT_SYSTEM_IMPERIAL,
CONF_UNIT_SYSTEM_METRIC,
EVENT_HOMEASSISTANT_STOP,
)
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity import Entity
_LOGGER = logging.getLogger(__name__)
@ -82,7 +81,7 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
add_entities(sensors)
class LoopEnergyDevice(Entity):
class LoopEnergySensor(SensorEntity):
"""Implementation of an Loop Energy base sensor."""
def __init__(self, controller):
@ -116,7 +115,7 @@ class LoopEnergyDevice(Entity):
self.schedule_update_ha_state(True)
class LoopEnergyElec(LoopEnergyDevice):
class LoopEnergyElec(LoopEnergySensor):
"""Implementation of an Loop Energy Electricity sensor."""
def __init__(self, controller):
@ -133,7 +132,7 @@ class LoopEnergyElec(LoopEnergyDevice):
self._state = round(self._controller.electricity_useage, 2)
class LoopEnergyGas(LoopEnergyDevice):
class LoopEnergyGas(LoopEnergySensor):
"""Implementation of an Loop Energy Gas sensor."""
def __init__(self, controller):