Reformat imports with ruff

custom_components/irm_kmi/__init__.py

@@ -8,9 +8,16 @@ from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryError
 from irm_kmi_api import RadarStyle
 
-from .const import (CONF_DARK_MODE, CONF_LANGUAGE_OVERRIDE, CONF_STYLE,
+from .const import (
-                    CONFIG_FLOW_VERSION, DOMAIN, PLATFORMS, CONF_USE_DEPRECATED_FORECAST,
+    CONF_DARK_MODE,
-                    OPTION_DEPRECATED_FORECAST_NOT_USED)
+    CONF_LANGUAGE_OVERRIDE,
+    CONF_STYLE,
+    CONF_USE_DEPRECATED_FORECAST,
+    CONFIG_FLOW_VERSION,
+    DOMAIN,
+    OPTION_DEPRECATED_FORECAST_NOT_USED,
+    PLATFORMS,
+)
 from .coordinator import IrmKmiCoordinator
 
 _LOGGER = logging.getLogger(__name__)

custom_components/irm_kmi/binary_sensor.py

@@ -2,8 +2,10 @@
 import logging
 
 from homeassistant.components import binary_sensor
-from homeassistant.components.binary_sensor import (BinarySensorDeviceClass,
+from homeassistant.components.binary_sensor import (
-                                                    BinarySensorEntity)
+    BinarySensorDeviceClass,
+    BinarySensorEntity,
+)
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback

custom_components/irm_kmi/camera.py

@@ -9,8 +9,8 @@ from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
-from .coordinator import IrmKmiCoordinator
 from .const import DOMAIN
+from .coordinator import IrmKmiCoordinator
 
 _LOGGER = logging.getLogger(__name__)
 

custom_components/irm_kmi/config_flow.py

@@ -9,17 +9,25 @@ from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE, CONF_ZONE
 from homeassistant.core import callback
 from homeassistant.data_entry_flow import FlowResult
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
-from homeassistant.helpers.selector import (EntitySelector,
+from homeassistant.helpers.selector import (
-                                            EntitySelectorConfig,
+    EntitySelector,
-                                            SelectSelector,
+    EntitySelectorConfig,
-                                            SelectSelectorConfig,
+    SelectSelector,
-                                            SelectSelectorMode)
+    SelectSelectorConfig,
+    SelectSelectorMode,
+)
 from irm_kmi_api import IrmKmiApiClient, RadarStyle
 
-from .const import (CONF_DARK_MODE, CONF_LANGUAGE_OVERRIDE,
+from .const import (
-                    CONF_LANGUAGE_OVERRIDE_OPTIONS, CONF_STYLE,
+    CONF_DARK_MODE,
-                    CONFIG_FLOW_VERSION, DOMAIN,
+    CONF_LANGUAGE_OVERRIDE,
-                    OUT_OF_BENELUX, USER_AGENT)
+    CONF_LANGUAGE_OVERRIDE_OPTIONS,
+    CONF_STYLE,
+    CONFIG_FLOW_VERSION,
+    DOMAIN,
+    OUT_OF_BENELUX,
+    USER_AGENT,
+)
 from .utils import get_config_value
 
 _LOGGER = logging.getLogger(__name__)

custom_components/irm_kmi/const.py

@@ -2,18 +2,25 @@
 from typing import Final
 
 from homeassistant.components.sensor import SensorDeviceClass
-from homeassistant.components.weather import (ATTR_CONDITION_CLEAR_NIGHT,
+from homeassistant.components.weather import (
-                                              ATTR_CONDITION_CLOUDY,
+      ATTR_CONDITION_CLEAR_NIGHT,
-                                              ATTR_CONDITION_FOG,
+      ATTR_CONDITION_CLOUDY,
-                                              ATTR_CONDITION_LIGHTNING_RAINY,
+      ATTR_CONDITION_FOG,
-                                              ATTR_CONDITION_PARTLYCLOUDY,
+      ATTR_CONDITION_LIGHTNING_RAINY,
-                                              ATTR_CONDITION_POURING,
+      ATTR_CONDITION_PARTLYCLOUDY,
-                                              ATTR_CONDITION_RAINY,
+      ATTR_CONDITION_POURING,
-                                              ATTR_CONDITION_SNOWY,
+      ATTR_CONDITION_RAINY,
-                                              ATTR_CONDITION_SNOWY_RAINY,
+      ATTR_CONDITION_SNOWY,
-                                              ATTR_CONDITION_SUNNY)
+      ATTR_CONDITION_SNOWY_RAINY,
-from homeassistant.const import (DEGREE, Platform, UnitOfPressure, UnitOfSpeed,
+      ATTR_CONDITION_SUNNY,
-                                 UnitOfTemperature)
+)
+from homeassistant.const import (
+      DEGREE,
+      Platform,
+      UnitOfPressure,
+      UnitOfSpeed,
+      UnitOfTemperature,
+)
 from irm_kmi_api import PollenName
 
 DOMAIN: Final = 'irm_kmi'

custom_components/irm_kmi/coordinator.py

@@ -10,14 +10,28 @@ from homeassistant.helpers import issue_registry
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
 from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
 from homeassistant.helpers.update_coordinator import (
-    TimestampDataUpdateCoordinator, UpdateFailed)
+    TimestampDataUpdateCoordinator,
+    UpdateFailed,
+)
 from homeassistant.util import dt
 from homeassistant.util.dt import utcnow
-from irm_kmi_api import IrmKmiApiClientHa, IrmKmiApiError, PollenParser, RainGraph, RadarStyle
+from irm_kmi_api import (
+    IrmKmiApiClientHa,
+    IrmKmiApiError,
+    PollenParser,
+    RadarStyle,
+    RainGraph,
+)
 
-from .const import CONF_DARK_MODE, CONF_STYLE, DOMAIN, IRM_KMI_NAME
+from .const import (
+    CONF_DARK_MODE,
+    CONF_STYLE,
+    DOMAIN,
+    IRM_KMI_NAME,
+    OUT_OF_BENELUX,
+    USER_AGENT,
+)
 from .const import IRM_KMI_TO_HA_CONDITION_MAP as CDT_MAP
-from .const import OUT_OF_BENELUX, USER_AGENT
 from .data import ProcessedCoordinatorData
 from .utils import disable_from_config, get_config_value, preferred_language
 

custom_components/irm_kmi/data.py

@@ -1,7 +1,7 @@
 from typing import List, TypedDict
 
 from homeassistant.components.weather import Forecast
-from irm_kmi_api import CurrentWeatherData, ExtendedForecast, WarningData, RainGraph
+from irm_kmi_api import CurrentWeatherData, ExtendedForecast, RainGraph, WarningData
 
 
 class ProcessedCoordinatorData(TypedDict, total=False):

custom_components/irm_kmi/repairs.py

@@ -11,8 +11,14 @@ from homeassistant.helpers.selector import SelectSelector, SelectSelectorConfig
 from irm_kmi_api import IrmKmiApiClient
 
 from . import async_reload_entry
-from .const import (OUT_OF_BENELUX, REPAIR_OPT_DELETE, REPAIR_OPT_MOVE,
+from .const import (
-                    REPAIR_OPTIONS, REPAIR_SOLUTION, USER_AGENT)
+    OUT_OF_BENELUX,
+    REPAIR_OPT_DELETE,
+    REPAIR_OPT_MOVE,
+    REPAIR_OPTIONS,
+    REPAIR_SOLUTION,
+    USER_AGENT,
+)
 from .utils import modify_from_config
 
 _LOGGER = logging.getLogger(__name__)

custom_components/irm_kmi/sensor.py

@@ -9,12 +9,17 @@ from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from homeassistant.util import dt
-from irm_kmi_api import ExtendedForecast, PollenParser, PollenName, RadarForecast
+from irm_kmi_api import ExtendedForecast, PollenName, PollenParser, RadarForecast
 
+from .const import (
+    CURRENT_WEATHER_SENSOR_CLASS,
+    CURRENT_WEATHER_SENSOR_ICON,
+    CURRENT_WEATHER_SENSOR_UNITS,
+    CURRENT_WEATHER_SENSORS,
+    DOMAIN,
+    POLLEN_TO_ICON_MAP,
+)
 from .coordinator import IrmKmiCoordinator
-from .const import (CURRENT_WEATHER_SENSOR_CLASS, CURRENT_WEATHER_SENSOR_ICON,
-                    CURRENT_WEATHER_SENSOR_UNITS, CURRENT_WEATHER_SENSORS,
-                    POLLEN_TO_ICON_MAP, DOMAIN)
 
 _LOGGER = logging.getLogger(__name__)
 

custom_components/irm_kmi/weather.py

@@ -1,14 +1,21 @@
 """Support for IRM KMI weather."""
 import logging
 from datetime import datetime
-from typing import List, Dict
+from typing import Dict, List
 
 import voluptuous as vol
-from homeassistant.components.weather import (Forecast, WeatherEntity,
+from homeassistant.components.weather import (
-                                              WeatherEntityFeature)
+    Forecast,
+    WeatherEntity,
+    WeatherEntityFeature,
+)
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import (UnitOfPrecipitationDepth, UnitOfPressure,
+from homeassistant.const import (
-                                 UnitOfSpeed, UnitOfTemperature)
+    UnitOfPrecipitationDepth,
+    UnitOfPressure,
+    UnitOfSpeed,
+    UnitOfTemperature,
+)
 from homeassistant.core import HomeAssistant, SupportsResponse
 from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers import entity_platform

tests/conftest.py

@@ -8,14 +8,24 @@ from unittest.mock import MagicMock, patch
 
 import pytest
 from homeassistant.const import CONF_ZONE
-from irm_kmi_api import IrmKmiApiClientHa, IrmKmiApiError, AnimationFrameData, RadarAnimationData, RadarStyle
+from irm_kmi_api import (
-from pytest_homeassistant_custom_component.common import (MockConfigEntry,
+    AnimationFrameData,
-                                                          load_fixture)
+    IrmKmiApiClientHa,
+    IrmKmiApiError,
+    RadarAnimationData,
+    RadarStyle,
+)
+from pytest_homeassistant_custom_component.common import MockConfigEntry, load_fixture
 
 from custom_components.irm_kmi.const import (
-    CONF_DARK_MODE, CONF_LANGUAGE_OVERRIDE, CONF_STYLE,
+    CONF_DARK_MODE,
-    CONF_USE_DEPRECATED_FORECAST, DOMAIN, IRM_KMI_TO_HA_CONDITION_MAP,
+    CONF_LANGUAGE_OVERRIDE,
-    OPTION_DEPRECATED_FORECAST_NOT_USED)
+    CONF_STYLE,
+    CONF_USE_DEPRECATED_FORECAST,
+    DOMAIN,
+    IRM_KMI_TO_HA_CONDITION_MAP,
+    OPTION_DEPRECATED_FORECAST_NOT_USED,
+)
 
 
 def get_api_data(fixture: str) -> dict:

tests/test_config_flow.py

@@ -12,8 +12,12 @@ from pytest_homeassistant_custom_component.common import MockConfigEntry
 
 from custom_components.irm_kmi import async_migrate_entry
 from custom_components.irm_kmi.const import (
-    CONF_DARK_MODE, CONF_LANGUAGE_OVERRIDE, CONF_STYLE,
+    CONF_DARK_MODE,
-    CONFIG_FLOW_VERSION, DOMAIN)
+    CONF_LANGUAGE_OVERRIDE,
+    CONF_STYLE,
+    CONFIG_FLOW_VERSION,
+    DOMAIN,
+)
 
 
 async def test_full_user_flow(

tests/test_current_weather_sensors.py

@@ -7,9 +7,11 @@ from irm_kmi_api.data import CurrentWeatherData
 from pytest_homeassistant_custom_component.common import MockConfigEntry
 
 from custom_components.irm_kmi import IrmKmiCoordinator
-from custom_components.irm_kmi.const import (CURRENT_WEATHER_SENSOR_CLASS,
+from custom_components.irm_kmi.const import (
-                                             CURRENT_WEATHER_SENSOR_UNITS,
+    CURRENT_WEATHER_SENSOR_CLASS,
-                                             CURRENT_WEATHER_SENSORS)
+    CURRENT_WEATHER_SENSOR_UNITS,
+    CURRENT_WEATHER_SENSORS,
+)
 from custom_components.irm_kmi.data import ProcessedCoordinatorData
 from custom_components.irm_kmi.sensor import IrmKmiCurrentRainfall
 from tests.conftest import get_api_with_data

tests/test_init.py

@@ -11,8 +11,12 @@ from pytest_homeassistant_custom_component.common import MockConfigEntry
 
 from custom_components.irm_kmi import async_migrate_entry
 from custom_components.irm_kmi.const import (
-    CONF_DARK_MODE, CONF_LANGUAGE_OVERRIDE, CONF_STYLE,
+    CONF_DARK_MODE,
-    CONFIG_FLOW_VERSION, DOMAIN)
+    CONF_LANGUAGE_OVERRIDE,
+    CONF_STYLE,
+    CONFIG_FLOW_VERSION,
+    DOMAIN,
+)
 
 
 async def test_load_unload_config_entry(

tests/test_repairs.py

@@ -5,14 +5,18 @@ from unittest.mock import AsyncMock, MagicMock
 from homeassistant.core import HomeAssistant
 from homeassistant.data_entry_flow import FlowResultType
 from homeassistant.helpers import issue_registry
-from pytest_homeassistant_custom_component.common import (MockConfigEntry,
+from pytest_homeassistant_custom_component.common import MockConfigEntry, load_fixture
-                                                          load_fixture)
 
 from custom_components.irm_kmi import DOMAIN, IrmKmiCoordinator
-from custom_components.irm_kmi.const import (REPAIR_OPT_DELETE,
+from custom_components.irm_kmi.const import (
-                                             REPAIR_OPT_MOVE, REPAIR_SOLUTION)
+    REPAIR_OPT_DELETE,
-from custom_components.irm_kmi.repairs import (OutOfBeneluxRepairFlow,
+    REPAIR_OPT_MOVE,
-                                               async_create_fix_flow)
+    REPAIR_SOLUTION,
+)
+from custom_components.irm_kmi.repairs import (
+    OutOfBeneluxRepairFlow,
+    async_create_fix_flow,
+)
 
 _LOGGER = logging.getLogger(__name__)
 

tests/test_sensors.py

@@ -8,8 +8,7 @@ from pytest_homeassistant_custom_component.common import MockConfigEntry
 from custom_components.irm_kmi import IrmKmiCoordinator
 from custom_components.irm_kmi.binary_sensor import IrmKmiWarning
 from custom_components.irm_kmi.const import CONF_LANGUAGE_OVERRIDE
-from custom_components.irm_kmi.sensor import (IrmKmiNextSunMove,
+from custom_components.irm_kmi.sensor import IrmKmiNextSunMove, IrmKmiNextWarning
-                                              IrmKmiNextWarning)
 from tests.conftest import get_api_with_data, get_radar_animation_data
 
 

tests/test_weather.py

@@ -1,16 +1,15 @@
 import json
 from datetime import datetime
-from typing import List, Dict
+from typing import Dict, List
 
 from freezegun import freeze_time
 from homeassistant.components.weather import Forecast
 from homeassistant.core import HomeAssistant
 from irm_kmi_api.data import RadarForecast
-from pytest_homeassistant_custom_component.common import (MockConfigEntry,
+from pytest_homeassistant_custom_component.common import MockConfigEntry, load_fixture
-                                                          load_fixture)
 
-from custom_components.irm_kmi.weather import IrmKmiCoordinator, IrmKmiWeather
 from custom_components.irm_kmi.data import ProcessedCoordinatorData
+from custom_components.irm_kmi.weather import IrmKmiCoordinator, IrmKmiWeather
 from tests.conftest import get_api_with_data