python-kasa/kasa/smart/modules/temperaturesensor.py
Steven B f259a8f162
Make module names consistent and remove redundant module casting (#909)
Address the inconsistent naming of smart modules by removing all "Module" suffixes and aligning filenames with class names. Removes the casting of modules to the correct module type now that is is redundant. Update the adding of iot modules to use the ModuleName class rather than a free string.
2024-05-11 19:28:18 +01:00

78 lines
2.5 KiB
Python

"""Implementation of temperature module."""
from __future__ import annotations
from typing import TYPE_CHECKING, Literal
from ...feature import Feature
from ..smartmodule import SmartModule
if TYPE_CHECKING:
from ..smartdevice import SmartDevice
class TemperatureSensor(SmartModule):
"""Implementation of temperature module."""
REQUIRED_COMPONENT = "temperature"
QUERY_GETTER_NAME = "get_comfort_temp_config"
def __init__(self, device: SmartDevice, module: str):
super().__init__(device, module)
self._add_feature(
Feature(
device,
id="temperature",
name="Temperature",
container=self,
attribute_getter="temperature",
icon="mdi:thermometer",
category=Feature.Category.Primary,
)
)
if "current_temp_exception" in device.sys_info:
self._add_feature(
Feature(
device,
id="temperature_warning",
name="Temperature warning",
container=self,
attribute_getter="temperature_warning",
type=Feature.Type.BinarySensor,
icon="mdi:alert",
category=Feature.Category.Debug,
)
)
self._add_feature(
Feature(
device,
id="temperature_unit",
name="Temperature unit",
container=self,
attribute_getter="temperature_unit",
attribute_setter="set_temperature_unit",
type=Feature.Type.Choice,
choices=["celsius", "fahrenheit"],
)
)
# TODO: use temperature_unit for feature creation
@property
def temperature(self):
"""Return current humidity in percentage."""
return self._device.sys_info["current_temp"]
@property
def temperature_warning(self) -> bool:
"""Return True if temperature is outside of the wanted range."""
return self._device.sys_info.get("current_temp_exception", 0) != 0
@property
def temperature_unit(self):
"""Return current temperature unit."""
return self._device.sys_info["temp_unit"]
async def set_temperature_unit(self, unit: Literal["celsius", "fahrenheit"]):
"""Set the device temperature unit."""
return await self.call("set_temperature_unit", {"temp_unit": unit})