Small ruff-fixes

Author: bouwew

custom_components/plugwise_usb/__init__.py

@@ -11,7 +11,7 @@
 from plugwise_usb.api import NodeEvent
 from plugwise_usb.exceptions import StickError
 
-from .const import (
+from .const import CONF_USB_PATH, NODES, PLUGWISE_USB_PLATFORMS, STICK
     CONF_USB_PATH,
     NODES,
     PLUGWISE_USB_PLATFORMS,

custom_components/plugwise_usb/binary_sensor.py

@@ -16,7 +16,7 @@
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from plugwise_usb.api import NodeEvent, NodeFeature
 
-from .const import DOMAIN, NODES, STICK, UNSUB_NODE_LOADED
+from .const import NODES, STICK, UNSUB_NODE_LOADED
 from .entity import PlugwiseUSBEntity, PlugwiseUSBEntityDescription
 
 _LOGGER = logging.getLogger(__name__)

custom_components/plugwise_usb/coordinator.py

@@ -1,6 +1,5 @@
 """DataUpdateCoordinator for Plugwise USB-Stick."""
 
-import asyncio
 from collections import Counter
 from datetime import timedelta
 import logging
@@ -53,7 +52,7 @@ def __init__(
     async def async_node_update(self) -> dict[NodeFeature, Any]:
         """Request status update for Plugwise Node."""
         states: dict[NodeFeature, Any] = {}
-        
+
         # Only unique features
         freq_features = Counter(self.async_contexts())
         features = tuple(freq_features.keys())
@@ -63,7 +62,7 @@ async def async_node_update(self) -> dict[NodeFeature, Any]:
         except StickError as err:
             raise ConfigEntryError from err
         except (StickTimeout, NodeTimeout) as err:
-            raise asyncio.TimeoutError from err
+            raise TimeoutError from err
         except NodeError as err:
             raise UpdateFailed from err
 

custom_components/plugwise_usb/entity.py

@@ -12,6 +12,7 @@
     CoordinatorEntity,
 )
 from plugwise_usb.api import PUSHING_FEATURES, NodeFeature, NodeInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
 from .const import DOMAIN
 from .coordinator import PlugwiseUSBDataUpdateCoordinator

custom_components/plugwise_usb/sensor.py

@@ -26,7 +26,7 @@
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from plugwise_usb.api import NodeEvent, NodeFeature
 
-from .const import DOMAIN, NODES, STICK, UNSUB_NODE_LOADED
+from .const import NODES, STICK, UNSUB_NODE_LOADED
 from .entity import PlugwiseUSBEntity, PlugwiseUSBEntityDescription
 
 _LOGGER = logging.getLogger(__name__)

custom_components/plugwise_usb/switch.py

@@ -16,7 +16,7 @@
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from plugwise_usb.api import NodeEvent, NodeFeature
 
-from .const import DOMAIN, NODES, STICK, UNSUB_NODE_LOADED
+from .const import NODES, STICK, UNSUB_NODE_LOADED
 from .coordinator import PlugwiseUSBDataUpdateCoordinator
 from .entity import PlugwiseUSBEntity, PlugwiseUSBEntityDescription
 

tests/components/plugwise_usb/test_config_flow.py

@@ -48,7 +48,7 @@ async def test_user_flow_select(hass):
         usb_mock.return_value.initialize = AsyncMock(return_value=None)
         usb_mock.return_value.disconnect = AsyncMock(return_value=None)
         usb_mock.return_value.mac_stick = MagicMock(return_value="01:23:45:67:AB")
-    
+
         result = await hass.config_entries.flow.async_configure(
             result["flow_id"], user_input={CONF_USB_PATH: port_select}
         )