diff --git a/homeassistant/components/fail2ban/sensor.py b/homeassistant/components/fail2ban/sensor.py index 5a7e1052b67..1063dfda08a 100644 --- a/homeassistant/components/fail2ban/sensor.py +++ b/homeassistant/components/fail2ban/sensor.py @@ -1,4 +1,6 @@ """Support for displaying IPs banned by fail2ban.""" +from __future__ import annotations + from datetime import timedelta import logging import os @@ -8,7 +10,10 @@ import voluptuous as vol from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity from homeassistant.const import CONF_FILE_PATH, CONF_NAME +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType _LOGGER = logging.getLogger(__name__) @@ -30,10 +35,15 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( ) -async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): +async def async_setup_platform( + hass: HomeAssistant, + config: ConfigType, + async_add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Set up the fail2ban sensor.""" - name = config.get(CONF_NAME) - jails = config.get(CONF_JAILS) + name = config[CONF_NAME] + jails = config[CONF_JAILS] log_file = config.get(CONF_FILE_PATH, DEFAULT_LOG) device_list = [] diff --git a/homeassistant/components/file/sensor.py b/homeassistant/components/file/sensor.py index 6a18fa4cc8b..b4822c7bcd5 100644 --- a/homeassistant/components/file/sensor.py +++ b/homeassistant/components/file/sensor.py @@ -1,4 +1,6 @@ """Support for sensor value(s) stored in local files.""" +from __future__ import annotations + import logging import os @@ -11,7 +13,10 @@ from homeassistant.const import ( CONF_UNIT_OF_MEASUREMENT, CONF_VALUE_TEMPLATE, ) +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType _LOGGER = logging.getLogger(__name__) @@ -29,10 +34,15 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( ) -async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): +async def async_setup_platform( + hass: HomeAssistant, + config: ConfigType, + async_add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Set up the file sensor.""" - file_path = config.get(CONF_FILE_PATH) - name = config.get(CONF_NAME) + file_path = config[CONF_FILE_PATH] + name = config[CONF_NAME] unit = config.get(CONF_UNIT_OF_MEASUREMENT) if (value_template := config.get(CONF_VALUE_TEMPLATE)) is not None: diff --git a/homeassistant/components/filesize/sensor.py b/homeassistant/components/filesize/sensor.py index dc44d3d8255..e409b346c6d 100644 --- a/homeassistant/components/filesize/sensor.py +++ b/homeassistant/components/filesize/sensor.py @@ -1,4 +1,6 @@ """Sensor for monitoring the size of a file.""" +from __future__ import annotations + import datetime import logging import os @@ -7,8 +9,11 @@ import voluptuous as vol from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity from homeassistant.const import DATA_MEGABYTES +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.reload import setup_reload_service +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import DOMAIN, PLATFORMS @@ -23,13 +28,18 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( ) -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 the file size sensor.""" setup_reload_service(hass, DOMAIN, PLATFORMS) sensors = [] - for path in config.get(CONF_FILE_PATHS): + for path in config[CONF_FILE_PATHS]: if not hass.config.is_allowed_path(path): _LOGGER.error("Filepath %s is not valid or allowed", path) continue diff --git a/homeassistant/components/fints/sensor.py b/homeassistant/components/fints/sensor.py index a7bec82cee0..4b6cb336bde 100644 --- a/homeassistant/components/fints/sensor.py +++ b/homeassistant/components/fints/sensor.py @@ -12,7 +12,10 @@ import voluptuous as vol from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity from homeassistant.const import CONF_NAME, CONF_PIN, CONF_URL, CONF_USERNAME +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType _LOGGER = logging.getLogger(__name__) @@ -51,7 +54,12 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( ) -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 the sensors. Login to the bank and get a list of existing accounts. Create a @@ -72,7 +80,7 @@ def setup_platform(hass, config, add_entities, discovery_info=None): client = FinTsClient(credentials, fints_name) balance_accounts, holdings_accounts = client.detect_accounts() - accounts = [] + accounts: list[SensorEntity] = [] for account in balance_accounts: if config[CONF_ACCOUNTS] and account.iban not in account_config: diff --git a/homeassistant/components/folder/sensor.py b/homeassistant/components/folder/sensor.py index c7257d40237..bfad45318e6 100644 --- a/homeassistant/components/folder/sensor.py +++ b/homeassistant/components/folder/sensor.py @@ -1,4 +1,6 @@ """Sensor for monitoring the contents of a folder.""" +from __future__ import annotations + from datetime import timedelta import glob import logging @@ -8,7 +10,10 @@ import voluptuous as vol from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity from homeassistant.const import DATA_MEGABYTES +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType _LOGGER = logging.getLogger(__name__) @@ -39,14 +44,19 @@ def get_size(files_list): return sum(size_list) -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 the folder sensor.""" - path = config.get(CONF_FOLDER_PATHS) + path = config[CONF_FOLDER_PATHS] if not hass.config.is_allowed_path(path): _LOGGER.error("Folder %s is not valid or allowed", path) else: - folder = Folder(path, config.get(CONF_FILTER)) + folder = Folder(path, config[CONF_FILTER]) add_entities([folder], True)