Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2024-12-22 11:13:34 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Add missing type hints to alarm module (#1111)

Browse Source
This commit is contained in:
Teemu R. 2024-09-10 18:20:00 +02:00 committed by GitHub
parent a967d5cd3a
commit 2a89e58ae0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
kasa/smart/modules/alarm.py
View File

@ -2,6 +2,8 @@
from __future__ import annotations from __future__ import annotations
from typing import Literal
from ...feature import Feature from ...feature import Feature
from ..smartmodule import SmartModule from ..smartmodule import SmartModule
@ -94,7 +96,7 @@ class Alarm(SmartModule):
) )
@property @property
def alarm_sound(self): def alarm_sound(self) -> str:
"""Return current alarm sound.""" """Return current alarm sound."""
return self.data["get_alarm_configure"]["type"] return self.data["get_alarm_configure"]["type"]
@ -113,11 +115,11 @@ class Alarm(SmartModule):
return self.data["get_support_alarm_type_list"]["alarm_type_list"] return self.data["get_support_alarm_type_list"]["alarm_type_list"]
@property @property
def alarm_volume(self): def alarm_volume(self) -> Literal["low", "normal", "high"]:
"""Return alarm volume.""" """Return alarm volume."""
return self.data["get_alarm_configure"]["volume"] return self.data["get_alarm_configure"]["volume"]
async def set_alarm_volume(self, volume: str): async def set_alarm_volume(self, volume: Literal["low", "normal", "high"]):
"""Set alarm volume.""" """Set alarm volume."""
payload = self.data["get_alarm_configure"].copy() payload = self.data["get_alarm_configure"].copy()
payload["volume"] = volume payload["volume"] = volume
@ -134,10 +136,10 @@ class Alarm(SmartModule):
src = self._device.sys_info["in_alarm_source"] src = self._device.sys_info["in_alarm_source"]
return src if src else None return src if src else None
async def play(self): async def play(self) -> dict:
"""Play alarm.""" """Play alarm."""
return await self.call("play_alarm") return await self.call("play_alarm")
async def stop(self): async def stop(self) -> dict:
"""Stop alarm.""" """Stop alarm."""
return await self.call("stop_alarm") return await self.call("stop_alarm")