Assistants.climate

  1"""
  2Ассистент контроля температуры и влажности.
  3
  4Периодически опрашивает датчик температуры/влажности и отправляет
  5уведомления через бота при выходе показателей за установленные пороги.
  6"""
  7
  8from langfuse.decorators import observe
  9from loguru import logger
 10
 11from Assistants.assistant import Assistant
 12from Controllers.climate import ClimateController
 13from Messengers.bot import Bot
 14
 15
 16class ClimateAssistant(Assistant):
 17    """Контролирует температуру и влажность в помещении.
 18
 19    При выходе температуры или влажности за заданные пороги
 20    отправляет предупреждение через Telegram-бота.
 21
 22    Attributes:
 23        temp_threshold_low: Нижний порог температуры.
 24        temp_threshold_high: Верхний порог температуры.
 25        humidity_threshold_low: Нижний порог влажности.
 26        humidity_threshold_high: Верхний порог влажности.
 27        rounding: Количество знаков после запятой для округления.
 28        temperature_controller: Контроллер датчика температуры/влажности.
 29        bot: Бот для отправки уведомлений.
 30    """
 31
 32    def __init__(
 33        self,
 34        name,
 35        temperature_controller: ClimateController,
 36        temperature_threshold_low=23,
 37        temperature_threshold_high=25,
 38        humidity_threshold_low=25,
 39        humidity_threshold_high=50,
 40        rounding=0,
 41        bot: Bot = None,
 42    ):
 43        """Инициализирует ассистента контроля климата.
 44
 45        Args:
 46            name: Имя ассистента.
 47            temperature_controller: Контроллер датчика температуры/влажности.
 48            temperature_threshold_low: Нижний порог температуры.
 49            temperature_threshold_high: Верхний порог температуры.
 50            humidity_threshold_low: Нижний порог влажности.
 51            humidity_threshold_high: Верхний порог влажности.
 52            rounding: Количество знаков после запятой для округления.
 53            bot: Бот для отправки уведомлений.
 54        """
 55        super().__init__(name)
 56        self.temp_threshold_low = temperature_threshold_low
 57        self.temp_threshold_high = temperature_threshold_high
 58        self.humidity_threshold_low = humidity_threshold_low
 59        self.humidity_threshold_high = humidity_threshold_high
 60        self.rounding = rounding
 61        self.temperature_controller = temperature_controller
 62        self.bot = bot
 63
 64    def handle_high_temp(self, temp):
 65        """Обрабатывает превышение верхнего порога температуры.
 66
 67        Args:
 68            temp: Текущее значение температуры.
 69        """
 70        logger.warning(f"Температура превысила порог: {temp}°C > {self.temp_threshold_high}°C")
 71        if self.bot:
 72            self.bot.send_message(
 73                f"Температура превысила порог: {temp}°C больше {self.temp_threshold_high}°C"
 74            )
 75        # Дополнительные действия, например, отправка уведомления
 76
 77    def handle_low_temp(self, temp):
 78        """Обрабатывает понижение температуры ниже порога.
 79
 80        Args:
 81            temp: Текущее значение температуры.
 82        """
 83        logger.warning(f"Температура ниже, чем порог: {temp}°C < {self.temp_threshold_low}°C")
 84        if self.bot:
 85            self.bot.send_message(
 86                f"Температура ниже, чем порог: {temp}°C меньше {self.temp_threshold_low}°C"
 87            )
 88        # Дополнительные действия, например, отправка уведомления
 89
 90    def handle_high_humidity(self, humidity):
 91        """Обрабатывает превышение верхнего порога влажности.
 92
 93        Args:
 94            humidity: Текущее значение влажности.
 95        """
 96        logger.warning(f"Влажность превысила порог: {humidity}% > {self.humidity_threshold_high}%")
 97        if self.bot:
 98            self.bot.send_message(
 99                f"Влажность превысила порог: {humidity}% больше {self.humidity_threshold_high}%"
100            )
101        # Дополнительные действия, например, отправка уведомления
102
103    def handle_low_humidity(self, temp):
104        """Обрабатывает понижение влажности ниже порога.
105
106        Args:
107            temp: Текущее значение влажности.
108        """
109        logger.warning(f"Влажность ниже, чем порог: {temp}% < {self.humidity_threshold_low}%")
110        if self.bot:
111            self.bot.send_message(
112                f"Влажность ниже, чем порог: {temp}% меньше {self.humidity_threshold_low}%"
113            )
114        # Дополнительные действия, например, отправка уведомления
115
116    @observe(name=f"Цикл контроля температуры и влажности {__qualname__}")
117    def run(self):
118        """Выполняет один цикл контроля температуры и влажности.
119
120        Получает показания с датчика, округляет их и сравнивает
121        с заданными порогами. При выходе за пределы вызывает
122        соответствующие обработчики.
123        """
124        logger.debug(f'Запуск цикла контроля температуры и влажности для "{self.name}"...')
125        state = self.temperature_controller.get_state()
126        temp = state["temperature"]
127        humidity = state["humidity"]
128        rounded_temp = round(temp, self.rounding)
129        rounded_humidity = round(humidity, self.rounding)
130        rounded_temp = int(temp) if self.rounding == 0 else rounded_temp
131        rounded_humidity = int(humidity) if self.rounding == 0 else rounded_humidity
132        if rounded_temp > self.temp_threshold_high:
133            self.handle_high_temp(rounded_temp)
134        elif rounded_temp < self.temp_threshold_low:
135            self.handle_low_temp(rounded_temp)
136        if rounded_humidity > self.humidity_threshold_high:
137            self.handle_high_humidity(rounded_humidity)
138        elif rounded_humidity < self.humidity_threshold_low:
139            self.handle_low_humidity(rounded_humidity)