From 4fad49ec5796da877e18f66c152a12ef3faa46d6 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 13 Jan 2022 02:30:10 +0100 Subject: [PATCH] Add setup type hints to proximity (#63990) Co-authored-by: epenet --- homeassistant/components/proximity/__init__.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/proximity/__init__.py b/homeassistant/components/proximity/__init__.py index eb64de56df5..da4392ccc09 100644 --- a/homeassistant/components/proximity/__init__.py +++ b/homeassistant/components/proximity/__init__.py @@ -15,9 +15,11 @@ from homeassistant.const import ( LENGTH_MILES, LENGTH_YARD, ) +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv from homeassistant.helpers.entity import Entity from homeassistant.helpers.event import track_state_change +from homeassistant.helpers.typing import ConfigType from homeassistant.util.distance import convert from homeassistant.util.location import distance @@ -64,7 +66,9 @@ CONFIG_SCHEMA = vol.Schema( ) -def setup_proximity_component(hass, name, config): +def setup_proximity_component( + hass: HomeAssistant, name: str, config: ConfigType +) -> bool: """Set up the individual proximity component.""" ignored_zones = config.get(CONF_IGNORED_ZONES) proximity_devices = config.get(CONF_DEVICES) @@ -75,7 +79,7 @@ def setup_proximity_component(hass, name, config): ) zone_id = f"zone.{config.get(CONF_ZONE)}" - proximity = Proximity( + proximity = Proximity( # type:ignore[no-untyped-call] hass, proximity_zone, DEFAULT_DIST_TO_ZONE, @@ -96,7 +100,7 @@ def setup_proximity_component(hass, name, config): return True -def setup(hass, config): +def setup(hass: HomeAssistant, config: ConfigType) -> bool: """Get the zones and offsets from configuration.yaml.""" for zone, proximity_config in config[DOMAIN].items(): setup_proximity_component(hass, zone, proximity_config)