Cleanup ADS constants and add type hints (#63390)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet
2022-01-14 12:58:12 +01:00
committed by GitHub
parent 55143e240b
commit 0042bb68d9
6 changed files with 50 additions and 54 deletions

View File

@@ -11,7 +11,14 @@ import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, StateType
from . import CONF_ADS_FACTOR, CONF_ADS_TYPE, CONF_ADS_VAR, STATE_KEY_STATE, AdsEntity
from . import (
ADS_TYPEMAP,
CONF_ADS_FACTOR,
CONF_ADS_TYPE,
CONF_ADS_VAR,
STATE_KEY_STATE,
AdsEntity,
)
DEFAULT_NAME = "ADS sensor"
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
@@ -67,7 +74,7 @@ class AdsSensor(AdsEntity, SensorEntity):
"""Register device notification."""
await self.async_initialize_device(
self._ads_var,
self._ads_hub.ADS_TYPEMAP[self._ads_type],
ADS_TYPEMAP[self._ads_type],
STATE_KEY_STATE,
self._factor,
)