Skip to content
Open
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 4 additions & 5 deletions custom_components/speedport/button.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import HomeAssistantType
from speedport import Speedport

from .const import DOMAIN
Expand All @@ -29,7 +28,7 @@ async def async_setup_entry(
class SpeedportReconnectButton(ButtonEntity, SpeedportEntity):
_attr_device_class = ButtonDeviceClass.RESTART

def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
"""Initialize the button entity."""
super().__init__(hass, speedport)
self._attr_name = "Reconnect"
Expand All @@ -44,7 +43,7 @@ async def async_press(self) -> None:
class SpeedportRebootButton(ButtonEntity, SpeedportEntity):
_attr_device_class = ButtonDeviceClass.RESTART

def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
"""Initialize the button entity."""
super().__init__(hass, speedport)
self._attr_name = "Reboot"
Expand All @@ -59,7 +58,7 @@ async def async_press(self) -> None:
class SpeedportWPSButton(ButtonEntity, SpeedportEntity):
_attr_device_class = ButtonDeviceClass.IDENTIFY

def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
"""Initialize the button entity."""
super().__init__(hass, speedport)
self._attr_name = "WPS on"
Expand All @@ -68,4 +67,4 @@ def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:

async def async_press(self) -> None:
"""Send out a restart command."""
await self._speedport.wps_on()
await self._speedport.wps_on()
10 changes: 5 additions & 5 deletions custom_components/speedport/device.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@
from datetime import timedelta

from homeassistant.core import callback
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.typing import HomeAssistantType
from homeassistant.helpers.update_coordinator import (
DataUpdateCoordinator,
CoordinatorEntity,
Expand All @@ -17,7 +17,7 @@


class SpeedportCoordinator(DataUpdateCoordinator[None]):
def __init__(self, hass: HomeAssistantType, device: Speedport):
def __init__(self, hass: HomeAssistant, device: Speedport):
"""Initialize my coordinator."""

super().__init__(
Expand All @@ -38,7 +38,7 @@ class SpeedportEntity(CoordinatorEntity[SpeedportCoordinator]):
_attr_has_entity_name = True

def __init__(
self, hass: HomeAssistantType, speedport: Speedport, description=None
self, hass: HomeAssistant, speedport: Speedport, description=None
) -> None:
coordinator = get_coordinator(hass, speedport)
super().__init__(coordinator)
Expand Down Expand Up @@ -66,7 +66,7 @@ def _handle_coordinator_update(self) -> None:


def get_coordinator(
hass: HomeAssistantType, speedport: Speedport
hass: HomeAssistant, speedport: Speedport
) -> SpeedportCoordinator:
coordinators = hass.data[DOMAIN]["coordinators"]
if speedport.serial_number in coordinators:
Expand All @@ -76,4 +76,4 @@ def get_coordinator(
else:
coordinator = SpeedportCoordinator(hass, speedport)
hass.data[DOMAIN]["coordinators"][speedport.serial_number] = coordinator
return coordinator
return coordinator
9 changes: 4 additions & 5 deletions custom_components/speedport/switch.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import HomeAssistantType
from speedport import Speedport

from .const import DOMAIN
Expand All @@ -34,7 +33,7 @@ async def async_setup_entry(
class SpeedportWifiSwitch(SwitchEntity, SpeedportEntity):
_attr_is_on: bool | None = False

def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
super().__init__(hass, speedport)
self._speedport: Speedport = speedport
self._attr_icon = "mdi:wifi"
Expand All @@ -57,7 +56,7 @@ async def async_turn_off(self, **kwargs: Any) -> None:
class SpeedportGuestWifiSwitch(SwitchEntity, SpeedportEntity):
_attr_is_on: bool | None = False

def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
super().__init__(hass, speedport)
self._speedport: Speedport = speedport
self._attr_icon = "mdi:wifi"
Expand All @@ -80,7 +79,7 @@ async def async_turn_off(self, **kwargs: Any) -> None:
class SpeedportOfficeWifiSwitch(SwitchEntity, SpeedportEntity):
_attr_is_on: bool | None = False

def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
super().__init__(hass, speedport)
self._speedport: Speedport = speedport
self._attr_icon = "mdi:wifi"
Expand All @@ -97,4 +96,4 @@ async def async_turn_on(self, **kwargs: Any) -> None:

async def async_turn_off(self, **kwargs: Any) -> None:
"""Turn off switch."""
await self._speedport.wifi_office_off()
await self._speedport.wifi_office_off()