From caff3e0630b97c3cf9281f2135fadb74ccd9e23d Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Mon, 10 Jan 2022 23:42:39 +0100 Subject: [PATCH] Add type hints to get_scanner/async_get_scanner (#63836) Co-authored-by: epenet --- homeassistant/components/arris_tg2492lg/device_tracker.py | 4 +++- homeassistant/components/aruba/device_tracker.py | 6 +++++- homeassistant/components/bbox/device_tracker.py | 4 +++- homeassistant/components/bt_home_hub_5/device_tracker.py | 6 +++++- homeassistant/components/bt_smarthub/device_tracker.py | 6 +++++- homeassistant/components/cisco_ios/device_tracker.py | 6 +++++- .../components/cisco_mobility_express/device_tracker.py | 6 +++++- homeassistant/components/cppm_tracker/device_tracker.py | 6 +++++- homeassistant/components/ddwrt/device_tracker.py | 6 +++++- homeassistant/components/fortios/device_tracker.py | 6 +++++- homeassistant/components/hitron_coda/device_tracker.py | 6 +++++- homeassistant/components/linksys_smart/device_tracker.py | 6 +++++- homeassistant/components/luci/device_tracker.py | 6 +++++- .../components/quantum_gateway/device_tracker.py | 6 +++++- homeassistant/components/sky_hub/device_tracker.py | 8 +++++++- homeassistant/components/snmp/device_tracker.py | 6 +++++- homeassistant/components/swisscom/device_tracker.py | 6 +++++- homeassistant/components/synology_srm/device_tracker.py | 4 +++- homeassistant/components/tado/device_tracker.py | 6 +++++- homeassistant/components/thomson/device_tracker.py | 6 +++++- homeassistant/components/tomato/device_tracker.py | 6 +++++- homeassistant/components/upc_connect/device_tracker.py | 6 +++++- homeassistant/components/xiaomi/device_tracker.py | 6 +++++- homeassistant/components/xiaomi_miio/device_tracker.py | 6 +++++- 24 files changed, 116 insertions(+), 24 deletions(-) diff --git a/homeassistant/components/arris_tg2492lg/device_tracker.py b/homeassistant/components/arris_tg2492lg/device_tracker.py index d0f15499d20..fbeb36366fe 100644 --- a/homeassistant/components/arris_tg2492lg/device_tracker.py +++ b/homeassistant/components/arris_tg2492lg/device_tracker.py @@ -10,7 +10,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST, CONF_PASSWORD +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType DEFAULT_HOST = "192.168.178.1" @@ -22,7 +24,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Return the Arris device scanner.""" conf = config[DOMAIN] url = f"http://{conf[CONF_HOST]}" diff --git a/homeassistant/components/aruba/device_tracker.py b/homeassistant/components/aruba/device_tracker.py index 721585fa391..ecdcc5f70f2 100644 --- a/homeassistant/components/aruba/device_tracker.py +++ b/homeassistant/components/aruba/device_tracker.py @@ -1,4 +1,6 @@ """Support for Aruba Access Points.""" +from __future__ import annotations + import logging import re @@ -11,7 +13,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -30,7 +34,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a Aruba scanner.""" scanner = ArubaDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/bbox/device_tracker.py b/homeassistant/components/bbox/device_tracker.py index 68f24d03ed0..ab8e7a72441 100644 --- a/homeassistant/components/bbox/device_tracker.py +++ b/homeassistant/components/bbox/device_tracker.py @@ -14,7 +14,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType from homeassistant.util import Throttle import homeassistant.util.dt as dt_util @@ -29,7 +31,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a Bbox scanner.""" scanner = BboxDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/bt_home_hub_5/device_tracker.py b/homeassistant/components/bt_home_hub_5/device_tracker.py index ec852f0c31b..2cef6e9ba41 100644 --- a/homeassistant/components/bt_home_hub_5/device_tracker.py +++ b/homeassistant/components/bt_home_hub_5/device_tracker.py @@ -1,4 +1,6 @@ """Support for BT Home Hub 5.""" +from __future__ import annotations + import logging import bthomehub5_devicelist @@ -10,7 +12,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -21,7 +25,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Return a BT Home Hub 5 scanner if successful.""" scanner = BTHomeHub5DeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/bt_smarthub/device_tracker.py b/homeassistant/components/bt_smarthub/device_tracker.py index a4c8717e444..0d50b09affa 100644 --- a/homeassistant/components/bt_smarthub/device_tracker.py +++ b/homeassistant/components/bt_smarthub/device_tracker.py @@ -1,4 +1,6 @@ """Support for BT Smart Hub (Sometimes referred to as BT Home Hub 6).""" +from __future__ import annotations + from collections import namedtuple import logging @@ -11,7 +13,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -26,7 +30,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Return a BT Smart Hub scanner if successful.""" info = config[DOMAIN] smarthub_client = BTSmartHub( diff --git a/homeassistant/components/cisco_ios/device_tracker.py b/homeassistant/components/cisco_ios/device_tracker.py index 9861f657ff6..dc9fe07aa53 100644 --- a/homeassistant/components/cisco_ios/device_tracker.py +++ b/homeassistant/components/cisco_ios/device_tracker.py @@ -1,4 +1,6 @@ """Support for Cisco IOS Routers.""" +from __future__ import annotations + import logging import re @@ -11,7 +13,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_PORT, CONF_USERNAME +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -27,7 +31,7 @@ PLATFORM_SCHEMA = vol.All( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a Cisco scanner.""" scanner = CiscoDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/cisco_mobility_express/device_tracker.py b/homeassistant/components/cisco_mobility_express/device_tracker.py index 5854f4a72a1..d207922f6a5 100644 --- a/homeassistant/components/cisco_mobility_express/device_tracker.py +++ b/homeassistant/components/cisco_mobility_express/device_tracker.py @@ -1,4 +1,6 @@ """Support for Cisco Mobility Express.""" +from __future__ import annotations + import logging from ciscomobilityexpress.ciscome import CiscoMobilityExpress @@ -16,7 +18,9 @@ from homeassistant.const import ( CONF_USERNAME, CONF_VERIFY_SSL, ) +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -34,7 +38,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a Cisco ME scanner.""" config = config[DOMAIN] diff --git a/homeassistant/components/cppm_tracker/device_tracker.py b/homeassistant/components/cppm_tracker/device_tracker.py index 7bd2be96030..8984da6beed 100644 --- a/homeassistant/components/cppm_tracker/device_tracker.py +++ b/homeassistant/components/cppm_tracker/device_tracker.py @@ -1,4 +1,6 @@ """Support for ClearPass Policy Manager.""" +from __future__ import annotations + from datetime import timedelta import logging @@ -11,7 +13,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_API_KEY, CONF_CLIENT_ID, CONF_HOST +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType SCAN_INTERVAL = timedelta(seconds=120) @@ -28,7 +32,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( _LOGGER = logging.getLogger(__name__) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Initialize Scanner.""" data = { diff --git a/homeassistant/components/ddwrt/device_tracker.py b/homeassistant/components/ddwrt/device_tracker.py index 8b7e7ed8c04..0947b755470 100644 --- a/homeassistant/components/ddwrt/device_tracker.py +++ b/homeassistant/components/ddwrt/device_tracker.py @@ -1,4 +1,6 @@ """Support for DD-WRT routers.""" +from __future__ import annotations + from http import HTTPStatus import logging import re @@ -18,7 +20,9 @@ from homeassistant.const import ( CONF_USERNAME, CONF_VERIFY_SSL, ) +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -42,7 +46,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a DD-WRT scanner.""" try: return DdWrtDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/fortios/device_tracker.py b/homeassistant/components/fortios/device_tracker.py index e160268e3fc..0e2d45db537 100644 --- a/homeassistant/components/fortios/device_tracker.py +++ b/homeassistant/components/fortios/device_tracker.py @@ -3,6 +3,8 @@ Support to use FortiOS device like FortiGate as device tracker. This component is part of the device_tracker platform. """ +from __future__ import annotations + import logging from awesomeversion import AwesomeVersion @@ -15,7 +17,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST, CONF_TOKEN, CONF_VERIFY_SSL +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) DEFAULT_VERIFY_SSL = False @@ -30,7 +34,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a FortiOSDeviceScanner.""" host = config[DOMAIN][CONF_HOST] verify_ssl = config[DOMAIN][CONF_VERIFY_SSL] diff --git a/homeassistant/components/hitron_coda/device_tracker.py b/homeassistant/components/hitron_coda/device_tracker.py index ac362f173e4..749347e82fe 100644 --- a/homeassistant/components/hitron_coda/device_tracker.py +++ b/homeassistant/components/hitron_coda/device_tracker.py @@ -1,4 +1,6 @@ """Support for the Hitron CODA-4582U, provided by Rogers.""" +from __future__ import annotations + from collections import namedtuple from http import HTTPStatus import logging @@ -12,7 +14,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_TYPE, CONF_USERNAME +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -28,7 +32,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(_hass, config): +def get_scanner(_hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a Hitron CODA-4582U scanner.""" scanner = HitronCODADeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/linksys_smart/device_tracker.py b/homeassistant/components/linksys_smart/device_tracker.py index c43de3141f7..8b296404532 100644 --- a/homeassistant/components/linksys_smart/device_tracker.py +++ b/homeassistant/components/linksys_smart/device_tracker.py @@ -1,4 +1,6 @@ """Support for Linksys Smart Wifi routers.""" +from __future__ import annotations + from http import HTTPStatus import logging @@ -11,7 +13,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType DEFAULT_TIMEOUT = 10 @@ -20,7 +24,7 @@ _LOGGER = logging.getLogger(__name__) PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend({vol.Required(CONF_HOST): cv.string}) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a Linksys AP scanner.""" try: return LinksysSmartWifiDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/luci/device_tracker.py b/homeassistant/components/luci/device_tracker.py index 7f70e793e60..c57e235a8d2 100644 --- a/homeassistant/components/luci/device_tracker.py +++ b/homeassistant/components/luci/device_tracker.py @@ -1,4 +1,6 @@ """Support for OpenWRT (luci) routers.""" +from __future__ import annotations + import logging from openwrt_luci_rpc import OpenWrtRpc @@ -16,7 +18,9 @@ from homeassistant.const import ( CONF_USERNAME, CONF_VERIFY_SSL, ) +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -34,7 +38,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a Luci scanner.""" scanner = LuciDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/quantum_gateway/device_tracker.py b/homeassistant/components/quantum_gateway/device_tracker.py index 228f6a5eab0..b9be96ecaea 100644 --- a/homeassistant/components/quantum_gateway/device_tracker.py +++ b/homeassistant/components/quantum_gateway/device_tracker.py @@ -1,4 +1,6 @@ """Support for Verizon FiOS Quantum Gateways.""" +from __future__ import annotations + import logging from quantum_gateway import QuantumGatewayScanner @@ -11,7 +13,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_SSL +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -26,7 +30,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a Quantum Gateway scanner.""" scanner = QuantumGatewayDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/sky_hub/device_tracker.py b/homeassistant/components/sky_hub/device_tracker.py index 8333e8e0cda..731baddcdb4 100644 --- a/homeassistant/components/sky_hub/device_tracker.py +++ b/homeassistant/components/sky_hub/device_tracker.py @@ -1,4 +1,6 @@ """Support for Sky Hub.""" +from __future__ import annotations + import logging from pyskyqhub.skyq_hub import SkyQHub @@ -10,15 +12,19 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST +from homeassistant.core import HomeAssistant from homeassistant.helpers.aiohttp_client import async_get_clientsession import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend({vol.Optional(CONF_HOST): cv.string}) -async def async_get_scanner(hass, config): +async def async_get_scanner( + hass: HomeAssistant, config: ConfigType +) -> DeviceScanner | None: """Return a Sky Hub scanner if successful.""" host = config[DOMAIN].get(CONF_HOST, "192.168.1.254") websession = async_get_clientsession(hass) diff --git a/homeassistant/components/snmp/device_tracker.py b/homeassistant/components/snmp/device_tracker.py index aeaf3c72e0f..c20dd58a752 100644 --- a/homeassistant/components/snmp/device_tracker.py +++ b/homeassistant/components/snmp/device_tracker.py @@ -1,4 +1,6 @@ """Support for fetching WiFi associations through SNMP.""" +from __future__ import annotations + import binascii import logging @@ -12,7 +14,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType from .const import ( CONF_AUTH_KEY, @@ -35,7 +39,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return an SNMP scanner.""" scanner = SnmpScanner(config[DOMAIN]) diff --git a/homeassistant/components/swisscom/device_tracker.py b/homeassistant/components/swisscom/device_tracker.py index 228f69d4c6c..3c8e7341be2 100644 --- a/homeassistant/components/swisscom/device_tracker.py +++ b/homeassistant/components/swisscom/device_tracker.py @@ -1,4 +1,6 @@ """Support for Swisscom routers (Internet-Box).""" +from __future__ import annotations + from contextlib import suppress import logging @@ -12,7 +14,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -23,7 +27,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Return the Swisscom device scanner.""" scanner = SwisscomDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/synology_srm/device_tracker.py b/homeassistant/components/synology_srm/device_tracker.py index 5981686e9b4..49369741218 100644 --- a/homeassistant/components/synology_srm/device_tracker.py +++ b/homeassistant/components/synology_srm/device_tracker.py @@ -19,7 +19,9 @@ from homeassistant.const import ( CONF_USERNAME, CONF_VERIFY_SSL, ) +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -68,7 +70,7 @@ ATTRIBUTE_ALIAS = { } -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return Synology SRM scanner.""" scanner = SynologySrmDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/tado/device_tracker.py b/homeassistant/components/tado/device_tracker.py index 977be7d226a..5992e0d7758 100644 --- a/homeassistant/components/tado/device_tracker.py +++ b/homeassistant/components/tado/device_tracker.py @@ -1,4 +1,6 @@ """Support for Tado Smart device trackers.""" +from __future__ import annotations + import asyncio from collections import namedtuple from datetime import timedelta @@ -15,8 +17,10 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_PASSWORD, CONF_USERNAME +from homeassistant.core import HomeAssistant from homeassistant.helpers.aiohttp_client import async_create_clientsession import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType from homeassistant.util import Throttle _LOGGER = logging.getLogger(__name__) @@ -34,7 +38,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Return a Tado scanner.""" scanner = TadoDeviceScanner(hass, config[DOMAIN]) return scanner if scanner.success_init else None diff --git a/homeassistant/components/thomson/device_tracker.py b/homeassistant/components/thomson/device_tracker.py index 8c52a6669dc..c04636e622a 100644 --- a/homeassistant/components/thomson/device_tracker.py +++ b/homeassistant/components/thomson/device_tracker.py @@ -1,4 +1,6 @@ """Support for THOMSON routers.""" +from __future__ import annotations + import logging import re import telnetlib @@ -11,7 +13,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -34,7 +38,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a THOMSON scanner.""" scanner = ThomsonDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/tomato/device_tracker.py b/homeassistant/components/tomato/device_tracker.py index 9799e1a4a87..e48b61ead4f 100644 --- a/homeassistant/components/tomato/device_tracker.py +++ b/homeassistant/components/tomato/device_tracker.py @@ -1,4 +1,6 @@ """Support for Tomato routers.""" +from __future__ import annotations + from http import HTTPStatus import json import logging @@ -20,7 +22,9 @@ from homeassistant.const import ( CONF_USERNAME, CONF_VERIFY_SSL, ) +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType CONF_HTTP_ID = "http_id" @@ -39,7 +43,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and returns a Tomato scanner.""" return TomatoDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/upc_connect/device_tracker.py b/homeassistant/components/upc_connect/device_tracker.py index f95f3276852..9908520342a 100644 --- a/homeassistant/components/upc_connect/device_tracker.py +++ b/homeassistant/components/upc_connect/device_tracker.py @@ -13,7 +13,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST, CONF_PASSWORD, EVENT_HOMEASSISTANT_STOP +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -27,7 +29,9 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -async def async_get_scanner(hass, config): +async def async_get_scanner( + hass: HomeAssistant, config: ConfigType +) -> DeviceScanner | None: """Return the UPC device scanner.""" conf = config[DOMAIN] session = hass.helpers.aiohttp_client.async_get_clientsession() diff --git a/homeassistant/components/xiaomi/device_tracker.py b/homeassistant/components/xiaomi/device_tracker.py index c1e38f64c53..c21f21e1f9b 100644 --- a/homeassistant/components/xiaomi/device_tracker.py +++ b/homeassistant/components/xiaomi/device_tracker.py @@ -1,4 +1,6 @@ """Support for Xiaomi Mi routers.""" +from __future__ import annotations + from http import HTTPStatus import logging @@ -11,7 +13,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -24,7 +28,7 @@ PLATFORM_SCHEMA = PARENT_PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Validate the configuration and return a Xiaomi Device Scanner.""" scanner = XiaomiDeviceScanner(config[DOMAIN]) diff --git a/homeassistant/components/xiaomi_miio/device_tracker.py b/homeassistant/components/xiaomi_miio/device_tracker.py index a6c1c7e5a28..3f819d7ab7d 100644 --- a/homeassistant/components/xiaomi_miio/device_tracker.py +++ b/homeassistant/components/xiaomi_miio/device_tracker.py @@ -1,4 +1,6 @@ """Support for Xiaomi Mi WiFi Repeater 2.""" +from __future__ import annotations + import logging from miio import DeviceException, WifiRepeater @@ -10,7 +12,9 @@ from homeassistant.components.device_tracker import ( DeviceScanner, ) from homeassistant.const import CONF_HOST, CONF_TOKEN +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.typing import ConfigType _LOGGER = logging.getLogger(__name__) @@ -22,7 +26,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( ) -def get_scanner(hass, config): +def get_scanner(hass: HomeAssistant, config: ConfigType) -> DeviceScanner | None: """Return a Xiaomi MiIO device scanner.""" scanner = None host = config[DOMAIN][CONF_HOST]