From ac795d8b2c3b1dc73ed6f510f897ed364f9dc5bf Mon Sep 17 00:00:00 2001 From: Jules Dejaeghere Date: Mon, 5 May 2025 22:24:12 +0200 Subject: [PATCH] Sort imports --- irm_kmi_api/__init__.py | 9 ++++++--- irm_kmi_api/api.py | 12 ++++++------ irm_kmi_api/const.py | 2 +- irm_kmi_api/pollen.py | 4 ++-- tests/test_api.py | 3 ++- tests/test_daily_forecast.py | 2 +- tests/test_pollen.py | 2 +- tests/test_rain_graph.py | 2 +- 8 files changed, 20 insertions(+), 16 deletions(-) diff --git a/irm_kmi_api/__init__.py b/irm_kmi_api/__init__.py index 5d016fc..95f0762 100644 --- a/irm_kmi_api/__init__.py +++ b/irm_kmi_api/__init__.py @@ -1,6 +1,9 @@ -from .api import IrmKmiApiError, IrmKmiApiCommunicationError, IrmKmiApiClient, IrmKmiApiClientHa -from .data import Forecast, ConditionEvol, RadarStyle, PollenName, PollenLevel, WarningType, ExtendedForecast, \ - CurrentWeatherData, WarningData, RadarForecast, RadarAnimationData, AnimationFrameData +from .api import (IrmKmiApiClient, IrmKmiApiClientHa, + IrmKmiApiCommunicationError, IrmKmiApiError) +from .data import (AnimationFrameData, ConditionEvol, CurrentWeatherData, + ExtendedForecast, Forecast, PollenLevel, PollenName, + RadarAnimationData, RadarForecast, RadarStyle, WarningData, + WarningType) from .pollen import PollenParser from .rain_graph import RainGraph diff --git a/irm_kmi_api/api.py b/irm_kmi_api/api.py index b379033..492c594 100644 --- a/irm_kmi_api/api.py +++ b/irm_kmi_api/api.py @@ -8,17 +8,17 @@ import time import urllib.parse from datetime import datetime, timedelta from statistics import mean -from typing import List, Dict +from typing import Dict, List from zoneinfo import ZoneInfo import aiohttp import async_timeout -from .const import MAP_WARNING_ID_TO_SLUG as SLUG_MAP, WWEVOL_TO_ENUM_MAP -from .const import STYLE_TO_PARAM_MAP, WEEKDAYS -from .data import (AnimationFrameData, CurrentWeatherData, Forecast, - ExtendedForecast, RadarForecast, RadarAnimationData, - WarningData, RadarStyle, WarningType) +from .const import MAP_WARNING_ID_TO_SLUG as SLUG_MAP +from .const import STYLE_TO_PARAM_MAP, WEEKDAYS, WWEVOL_TO_ENUM_MAP +from .data import (AnimationFrameData, CurrentWeatherData, ExtendedForecast, + Forecast, RadarAnimationData, RadarForecast, RadarStyle, + WarningData, WarningType) from .pollen import PollenParser _LOGGER = logging.getLogger(__name__) diff --git a/irm_kmi_api/const.py b/irm_kmi_api/const.py index 5082348..1a0fbd2 100644 --- a/irm_kmi_api/const.py +++ b/irm_kmi_api/const.py @@ -1,6 +1,6 @@ from typing import Final -from .data import ConditionEvol, RadarStyle, PollenLevel, WarningType +from .data import ConditionEvol, PollenLevel, RadarStyle, WarningType POLLEN_LEVEL_TO_COLOR = { 'null': PollenLevel.GREEN, diff --git a/irm_kmi_api/pollen.py b/irm_kmi_api/pollen.py index f4a209e..66c51d6 100644 --- a/irm_kmi_api/pollen.py +++ b/irm_kmi_api/pollen.py @@ -1,10 +1,10 @@ """Parse pollen info from SVG from IRM KMI api""" import logging import xml.etree.ElementTree as ET -from typing import List, Dict +from typing import Dict, List from .const import POLLEN_LEVEL_TO_COLOR -from .data import PollenName, PollenLevel +from .data import PollenLevel, PollenName _LOGGER = logging.getLogger(__name__) diff --git a/tests/test_api.py b/tests/test_api.py index b1f366b..6136d3a 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -1,6 +1,7 @@ import json import time -from datetime import datetime as dt, timedelta +from datetime import datetime as dt +from datetime import timedelta from unittest.mock import AsyncMock, MagicMock from zoneinfo import ZoneInfo diff --git a/tests/test_daily_forecast.py b/tests/test_daily_forecast.py index 126ab5c..c5336bc 100644 --- a/tests/test_daily_forecast.py +++ b/tests/test_daily_forecast.py @@ -3,7 +3,7 @@ from zoneinfo import ZoneInfo from freezegun import freeze_time -from irm_kmi_api.data import ExtendedForecast, ConditionEvol +from irm_kmi_api.data import ConditionEvol, ExtendedForecast from tests.conftest import get_api_with_data from tests.const import ATTR_CONDITION_PARTLYCLOUDY diff --git a/tests/test_pollen.py b/tests/test_pollen.py index ce9751b..86720f3 100644 --- a/tests/test_pollen.py +++ b/tests/test_pollen.py @@ -1,7 +1,7 @@ import logging from unittest.mock import AsyncMock -from irm_kmi_api.data import PollenName, PollenLevel +from irm_kmi_api.data import PollenLevel, PollenName from irm_kmi_api.pollen import PollenParser from tests.conftest import get_api_with_data, load_fixture diff --git a/tests/test_rain_graph.py b/tests/test_rain_graph.py index 96c1e3e..c397008 100644 --- a/tests/test_rain_graph.py +++ b/tests/test_rain_graph.py @@ -3,7 +3,7 @@ import datetime import json from datetime import datetime as dt from datetime import timedelta -from unittest.mock import MagicMock, AsyncMock +from unittest.mock import AsyncMock, MagicMock from zoneinfo import ZoneInfo from irm_kmi_api.api import IrmKmiApiClientHa