Make device port configurable (#471)

This commit is contained in:
Viktar Karpach
2023-07-09 18:55:27 -05:00
committed by GitHub
parent 6199521269
commit 9b039d8374
11 changed files with 67 additions and 26 deletions

View File

@@ -1,6 +1,6 @@
"""Module for smart plugs (HS100, HS110, ..)."""
import logging
from typing import Any, Dict
from typing import Any, Dict, Optional
from kasa.modules import Antitheft, Cloud, Schedule, Time, Usage
from kasa.smartdevice import DeviceType, SmartDevice, requires_update
@@ -37,8 +37,8 @@ class SmartPlug(SmartDevice):
For more examples, see the :class:`SmartDevice` class.
"""
def __init__(self, host: str) -> None:
super().__init__(host)
def __init__(self, host: str, *, port: Optional[int] = None) -> None:
super().__init__(host, port=port)
self._device_type = DeviceType.Plug
self.add_module("schedule", Schedule(self, "schedule"))
self.add_module("usage", Usage(self, "schedule"))