Add basic type hints to nissan_leaf (#62904)

This commit is contained in:
Phil Cole 2021-12-28 07:56:07 +00:00 committed by GitHub
parent d63b7bc5f1
commit 68acf13f48
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 35 additions and 5 deletions

View File

@ -1,22 +1,32 @@
"""Plugged In Status Support for the Nissan Leaf.""" """Plugged In Status Support for the Nissan Leaf."""
from __future__ import annotations
import logging import logging
from homeassistant.components.binary_sensor import ( from homeassistant.components.binary_sensor import (
BinarySensorDeviceClass, BinarySensorDeviceClass,
BinarySensorEntity, BinarySensorEntity,
) )
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import DATA_CHARGING, DATA_LEAF, DATA_PLUGGED_IN, LeafEntity from . import DATA_CHARGING, DATA_LEAF, DATA_PLUGGED_IN, LeafEntity
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
def setup_platform(hass, config, add_entities, discovery_info=None): def setup_platform(
hass: HomeAssistant,
config: ConfigType,
add_entities: AddEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Set up of a Nissan Leaf binary sensor.""" """Set up of a Nissan Leaf binary sensor."""
if discovery_info is None: if discovery_info is None:
return return
devices = [] devices: list[LeafEntity] = []
for vin, datastore in hass.data[DATA_LEAF].items(): for vin, datastore in hass.data[DATA_LEAF].items():
_LOGGER.debug("Adding binary_sensors for vin=%s", vin) _LOGGER.debug("Adding binary_sensors for vin=%s", vin)
devices.append(LeafPluggedInSensor(datastore)) devices.append(LeafPluggedInSensor(datastore))

View File

@ -1,9 +1,14 @@
"""Battery Charge and Range Support for the Nissan Leaf.""" """Battery Charge and Range Support for the Nissan Leaf."""
from __future__ import annotations
import logging import logging
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import PERCENTAGE from homeassistant.const import PERCENTAGE
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.icon import icon_for_battery_level from homeassistant.helpers.icon import icon_for_battery_level
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util.distance import LENGTH_KILOMETERS, LENGTH_MILES from homeassistant.util.distance import LENGTH_KILOMETERS, LENGTH_MILES
from homeassistant.util.unit_system import IMPERIAL_SYSTEM, METRIC_SYSTEM from homeassistant.util.unit_system import IMPERIAL_SYSTEM, METRIC_SYSTEM
@ -21,12 +26,17 @@ _LOGGER = logging.getLogger(__name__)
ICON_RANGE = "mdi:speedometer" ICON_RANGE = "mdi:speedometer"
def setup_platform(hass, config, add_devices, discovery_info=None): def setup_platform(
hass: HomeAssistant,
config: ConfigType,
add_devices: AddEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Sensors setup.""" """Sensors setup."""
if discovery_info is None: if discovery_info is None:
return return
devices = [] devices: list[LeafEntity] = []
for vin, datastore in hass.data[DATA_LEAF].items(): for vin, datastore in hass.data[DATA_LEAF].items():
_LOGGER.debug("Adding sensors for vin=%s", vin) _LOGGER.debug("Adding sensors for vin=%s", vin)
devices.append(LeafBatterySensor(datastore)) devices.append(LeafBatterySensor(datastore))

View File

@ -1,14 +1,24 @@
"""Charge and Climate Control Support for the Nissan Leaf.""" """Charge and Climate Control Support for the Nissan Leaf."""
from __future__ import annotations
import logging import logging
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import ToggleEntity from homeassistant.helpers.entity import ToggleEntity
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import DATA_CLIMATE, DATA_LEAF, LeafEntity from . import DATA_CLIMATE, DATA_LEAF, LeafEntity
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
def setup_platform(hass, config, add_devices, discovery_info=None): def setup_platform(
hass: HomeAssistant,
config: ConfigType,
add_devices: AddEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Nissan Leaf switch platform setup.""" """Nissan Leaf switch platform setup."""
if discovery_info is None: if discovery_info is None:
return return