Skip to content

Commit

Permalink
Add basic type hints to nissan_leaf (#62904)
Browse files Browse the repository at this point in the history
  • Loading branch information
filcole authored Dec 28, 2021
1 parent d63b7bc commit 68acf13
Show file tree
Hide file tree
Showing 3 changed files with 35 additions and 5 deletions.
14 changes: 12 additions & 2 deletions homeassistant/components/nissan_leaf/binary_sensor.py
Original file line number Diff line number Diff line change
@@ -1,22 +1,32 @@
"""Plugged In Status Support for the Nissan Leaf."""
from __future__ import annotations

import logging

from homeassistant.components.binary_sensor import (
BinarySensorDeviceClass,
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

_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."""
if discovery_info is None:
return

devices = []
devices: list[LeafEntity] = []
for vin, datastore in hass.data[DATA_LEAF].items():
_LOGGER.debug("Adding binary_sensors for vin=%s", vin)
devices.append(LeafPluggedInSensor(datastore))
Expand Down
14 changes: 12 additions & 2 deletions homeassistant/components/nissan_leaf/sensor.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,14 @@
"""Battery Charge and Range Support for the Nissan Leaf."""
from __future__ import annotations

import logging

from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
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.typing import ConfigType, DiscoveryInfoType
from homeassistant.util.distance import LENGTH_KILOMETERS, LENGTH_MILES
from homeassistant.util.unit_system import IMPERIAL_SYSTEM, METRIC_SYSTEM

Expand All @@ -21,12 +26,17 @@
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."""
if discovery_info is None:
return

devices = []
devices: list[LeafEntity] = []
for vin, datastore in hass.data[DATA_LEAF].items():
_LOGGER.debug("Adding sensors for vin=%s", vin)
devices.append(LeafBatterySensor(datastore))
Expand Down
12 changes: 11 additions & 1 deletion homeassistant/components/nissan_leaf/switch.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,24 @@
"""Charge and Climate Control Support for the Nissan Leaf."""
from __future__ import annotations

import logging

from homeassistant.core import HomeAssistant
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

_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."""
if discovery_info is None:
return
Expand Down

0 comments on commit 68acf13

Please sign in to comment.