Assistants.dog_water

  1#!/usr/bin/env python3
  2"""
  3Ассистент контроля миски с водой для собаки.
  4
  5Включает подогрев миски (через розетку) при понижении температуры
  6окружающего воздуха ниже заданного порога.
  7"""
  8
  9import time
 10
 11from e_lib.assistants import Assistant
 12from langfuse import observe
 13from loguru import logger
 14
 15from Controllers import SmartPlugController
 16from DataSources import OpenWeatherDataSource
 17from Messengers.bot import Bot
 18
 19
 20class DogWaterAssistant(Assistant):
 21    """Контролирует подогрев воды в миске для собаки.
 22
 23    Получает текущую температуру воздуха через OpenWeather и
 24    включает/выключает розетку с подогревом миски в зависимости
 25    от порога температуры.
 26
 27    Attributes:
 28        smart_plug: Контроллер розетки для подогрева.
 29        openweather: Источник данных о погоде.
 30        temp_threshold: Порог температуры для включения подогрева.
 31        bot: Бот для отправки уведомлений.
 32    """
 33
 34    def __init__(
 35        self,
 36        name: str,
 37        smart_plug: SmartPlugController,
 38        openweather: OpenWeatherDataSource,
 39        temp_threshold,
 40        bot: Bot = None,
 41    ):
 42        """Инициализирует ассистента контроля миски с водой.
 43
 44        Args:
 45            name: Имя ассистента.
 46            smart_plug: Контроллер розетки для подогрева.
 47            openweather: Источник данных о погоде.
 48            temp_threshold: Порог температуры для включения подогрева.
 49            bot: Бот для отправки уведомлений.
 50        """
 51        super().__init__(name)
 52
 53        self.smart_plug = smart_plug
 54        self.openweather = openweather
 55        self.temp_threshold = temp_threshold
 56        self.bot = bot
 57
 58    @observe(name=f"Получение состояния {__qualname__}")
 59    def get_state(self):
 60        """Получает состояние розетки с повторными попытками.
 61
 62        Проверяет, что состояние розетки консистентно: если включена —
 63        должно быть потребление тока, если выключена — тока быть не должно.
 64
 65        Returns:
 66            dict: Состояние розетки или None при ошибке.
 67        """
 68        retry_count = self.smart_plug.state_retry_count
 69        retry_delay = self.smart_plug.state_retry_delay
 70
 71        for attempt in range(retry_count):
 72            state = self.smart_plug.get_state()
 73            if (
 74                state is not None
 75                and state["power"]
 76                and state.get("amperage") is not None
 77                and state.get("amperage") > 0
 78            ):
 79                return state
 80            if (
 81                state is not None
 82                and not state["power"]
 83                and state.get("amperage") is not None
 84                and state.get("amperage") == 0
 85            ):
 86                return state
 87            logger.debug(
 88                f"Попытка {attempt + 1} не удалась, повтор через {retry_delay} секунд..."
 89            )
 90            time.sleep(retry_delay)
 91        if state is not None and state["power"]:
 92            logger.warning(
 93                "Розетка включена, но потребление тока отсутствует или недоступно после нескольких попыток."
 94            )
 95        if state is not None and not state["power"]:
 96            logger.warning(
 97                "Розетка выключена, но потребление тока присутствует или недоступно после нескольких попыток."
 98            )
 99
100        return state
101
102    @observe(name=f"Цикл контроля для {__qualname__}")
103    def run(self):
104        """Выполняет один цикл контроля подогрева миски.
105
106        Получает текущую температуру через OpenWeather, сравнивает
107        с порогом и при необходимости включает или выключает розетку.
108
109        Returns:
110            bool: True если цикл выполнен успешно, False при ошибке.
111        """
112        logger.debug(f'Запуск цикла контроля температуры для "{self.name}"...')
113
114        # Получаем текущую температуру
115        temperature = self.openweather.get_current_temperature()
116
117        if temperature is None:
118            logger.error(
119                f'Не удалось получить температуру, пропускаем цикл "{self.name}"'
120            )
121            return False
122
123        # Определяем нужное действие
124        should_turn_on = temperature <= self.temp_threshold
125
126        state = self.smart_plug.get_state()
127        if state is None:
128            logger.error(
129                f'Не удалось получить состояние розетки, пропускаем цикл "{self.name}"'
130            )
131            return False
132        if state["power"] != should_turn_on:
133            success = self.smart_plug.control_plug(should_turn_on)
134            if not success:
135                logger.error(
136                    f'Цикл "{self.name}" завершен с ошибкой управления розеткой'
137                )
138                return success
139            state = self.get_state()
140            logger.info(
141                f"Температура {temperature}°C (порог: {self.temp_threshold}°C). "
142                + str(self.smart_plug)
143            )
144            if self.bot:
145                self.bot.send_message(
146                    text=f"Температура {temperature}°C (порог: {self.temp_threshold}°C). "
147                    + str(self.smart_plug)
148                )
149
150        logger.debug(
151            f'Цикл "{self.name}" завершен: температура {temperature}°C (порог: {self.temp_threshold}°C). '
152            + str(self.smart_plug)
153        )
154
155        return True