move files, rename home package to homekit

1 files changed, 257 insertions, 0 deletions

diff --git a/bin/pump_bot.py b/bin/pump_bot.py
new file mode 100755
index 0000000..08d0dc6
--- /dev/null
+++ b/bin/pump_bot.py
@@ -0,0 +1,257 @@
+#!/usr/bin/env python3
+import __py_include
+
+from enum import Enum
+from typing import Optional
+from telegram import ReplyKeyboardMarkup, User
+from time import time
+from datetime import datetime
+
+from homekit.config import config, is_development_mode
+from homekit.telegram import bot
+from homekit.telegram._botutil import user_any_name
+from homekit.relay.sunxi_h3_client import RelayClient
+from homekit.api.types import BotType
+from homekit.mqtt import MqttNode, MqttWrapper, MqttPayload
+from homekit.mqtt.module.relay import MqttPowerStatusPayload, MqttRelayModule
+from homekit.mqtt.module.temphum import MqttTemphumDataPayload
+from homekit.mqtt.module.diagnostics import InitialDiagnosticsPayload, DiagnosticsPayload
+
+
+config.load_app('pump_bot')
+
+mqtt: Optional[MqttWrapper] = None
+mqtt_node: Optional[MqttNode] = None
+mqtt_relay_module: Optional[MqttRelayModule] = None
+time_format = '%d.%m.%Y, %H:%M:%S'
+
+watering_mcu_status = {
+    'last_time': 0,
+    'last_boot_time': 0,
+    'relay_opened': False,
+    'ambient_temp': 0.0,
+    'ambient_rh': 0.0,
+}
+
+bot.initialize()
+bot.lang.ru(
+    start_message="Выберите команду на клавиатуре",
+    unknown_command="Неизвестная команда",
+
+    enable="Включить",
+    enable_silently="Включить тихо",
+    enabled="Насос включен ✅",
+
+    disable="Выключить",
+    disable_silently="Выключить тихо",
+    disabled="Насос выключен ❌",
+
+    start_watering="Включить полив",
+    stop_watering="Отключить полив",
+
+    status="Статус насоса",
+    watering_status="Статус полива",
+
+    done="Готово 👌",
+    sent="Команда отправлена",
+
+    user_action_notification='Пользователь <a href="tg://user?id=%d">%s</a> <b>%s</b> насос.',
+    user_watering_notification='Пользователь <a href="tg://user?id=%d">%s</a> <b>%s</b> полив.',
+    user_action_on="включил",
+    user_action_off="выключил",
+    user_action_watering_on="включил",
+    user_action_watering_off="выключил",
+)
+bot.lang.en(
+    start_message="Select command on the keyboard",
+    unknown_command="Unknown command",
+
+    enable="Turn ON",
+    enable_silently="Turn ON silently",
+    enabled="The pump is turned ON ✅",
+
+    disable="Turn OFF",
+    disable_silently="Turn OFF silently",
+    disabled="The pump is turned OFF ❌",
+
+    start_watering="Start watering",
+    stop_watering="Stop watering",
+
+    status="Pump status",
+    watering_status="Watering status",
+
+    done="Done 👌",
+    sent="Request sent",
+
+    user_action_notification='User <a href="tg://user?id=%d">%s</a> turned the pump <b>%s</b>.',
+    user_watering_notification='User <a href="tg://user?id=%d">%s</a> <b>%s</b> the watering.',
+    user_action_on="ON",
+    user_action_off="OFF",
+    user_action_watering_on="started",
+    user_action_watering_off="stopped",
+)
+
+
+class UserAction(Enum):
+    ON = 'on'
+    OFF = 'off'
+    WATERING_ON = 'watering_on'
+    WATERING_OFF = 'watering_off'
+
+
+def get_relay() -> RelayClient:
+    relay = RelayClient(host=config['relay']['ip'], port=config['relay']['port'])
+    relay.connect()
+    return relay
+
+
+def on(ctx: bot.Context, silent=False) -> None:
+    get_relay().on()
+    ctx.reply(ctx.lang('done'))
+    if not silent:
+        notify(ctx.user, UserAction.ON)
+
+
+def off(ctx: bot.Context, silent=False) -> None:
+    get_relay().off()
+    ctx.reply(ctx.lang('done'))
+    if not silent:
+        notify(ctx.user, UserAction.OFF)
+
+
+def watering_on(ctx: bot.Context) -> None:
+    mqtt_relay_module.switchpower(True, config.get('mqtt_water_relay.secret'))
+    ctx.reply(ctx.lang('sent'))
+    notify(ctx.user, UserAction.WATERING_ON)
+
+
+def watering_off(ctx: bot.Context) -> None:
+    mqtt_relay_module.switchpower(False, config.get('mqtt_water_relay.secret'))
+    ctx.reply(ctx.lang('sent'))
+    notify(ctx.user, UserAction.WATERING_OFF)
+
+
+def notify(user: User, action: UserAction) -> None:
+    notification_key = 'user_watering_notification' if action in (UserAction.WATERING_ON, UserAction.WATERING_OFF) else 'user_action_notification'
+    def text_getter(lang: str):
+        action_name = bot.lang.get(f'user_action_{action.value}', lang)
+        user_name = user_any_name(user)
+        return 'ℹ ' + bot.lang.get(notification_key, lang,
+                                   user.id, user_name, action_name)
+
+    bot.notify_all(text_getter, exclude=(user.id,))
+
+
+@bot.handler(message='enable')
+def enable_handler(ctx: bot.Context) -> None:
+    on(ctx)
+
+
+@bot.handler(message='enable_silently')
+def enable_s_handler(ctx: bot.Context) -> None:
+    on(ctx, True)
+
+
+@bot.handler(message='disable')
+def disable_handler(ctx: bot.Context) -> None:
+    off(ctx)
+
+
+@bot.handler(message='start_watering')
+def start_watering(ctx: bot.Context) -> None:
+    watering_on(ctx)
+
+
+@bot.handler(message='stop_watering')
+def stop_watering(ctx: bot.Context) -> None:
+    watering_off(ctx)
+
+
+@bot.handler(message='disable_silently')
+def disable_s_handler(ctx: bot.Context) -> None:
+    off(ctx, True)
+
+
+@bot.handler(message='status')
+def status(ctx: bot.Context) -> None:
+    ctx.reply(
+        ctx.lang('enabled') if get_relay().status() == 'on' else ctx.lang('disabled')
+    )
+
+
+def _get_timestamp_as_string(timestamp: int) -> str:
+    if timestamp != 0:
+        return datetime.fromtimestamp(timestamp).strftime(time_format)
+    else:
+        return 'unknown'
+
+
+@bot.handler(message='watering_status')
+def watering_status(ctx: bot.Context) -> None:
+    buf = ''
+    if 0 < watering_mcu_status["last_time"] < time()-1800:
+        buf += '<b>WARNING! long time no reports from mcu! maybe something\'s wrong</b>\n'
+    buf += f'last report time: <b>{_get_timestamp_as_string(watering_mcu_status["last_time"])}</b>\n'
+    if watering_mcu_status["last_boot_time"] != 0:
+        buf += f'boot time: <b>{_get_timestamp_as_string(watering_mcu_status["last_boot_time"])}</b>\n'
+    buf += 'relay opened: <b>' + ('yes' if watering_mcu_status['relay_opened'] else 'no') + '</b>\n'
+    buf += f'ambient temp & humidity: <b>{watering_mcu_status["ambient_temp"]} °C, {watering_mcu_status["ambient_rh"]}%</b>'
+    ctx.reply(buf)
+
+
+@bot.defaultreplymarkup
+def markup(ctx: Optional[bot.Context]) -> Optional[ReplyKeyboardMarkup]:
+    buttons = []
+    if ctx.user_id in config['bot']['silent_users']:
+        buttons.append([ctx.lang('enable_silently'), ctx.lang('disable_silently')])
+    buttons.append([ctx.lang('enable'), ctx.lang('disable'), ctx.lang('status')],)
+    buttons.append([ctx.lang('start_watering'), ctx.lang('stop_watering'), ctx.lang('watering_status')])
+
+    return ReplyKeyboardMarkup(buttons, one_time_keyboard=False)
+
+
+def mqtt_payload_callback(mqtt_node: MqttNode, payload: MqttPayload):
+    global watering_mcu_status
+
+    types_the_node_can_send = (
+        InitialDiagnosticsPayload,
+        DiagnosticsPayload,
+        MqttTemphumDataPayload,
+        MqttPowerStatusPayload
+    )
+    for cl in types_the_node_can_send:
+        if isinstance(payload, cl):
+            watering_mcu_status['last_time'] = int(time())
+            break
+
+    if isinstance(payload, InitialDiagnosticsPayload):
+        watering_mcu_status['last_boot_time'] = int(time())
+
+    elif isinstance(payload, MqttTemphumDataPayload):
+        watering_mcu_status['ambient_temp'] = payload.temp
+        watering_mcu_status['ambient_rh'] = payload.rh
+
+    elif isinstance(payload, MqttPowerStatusPayload):
+        watering_mcu_status['relay_opened'] = payload.opened
+
+
+if __name__ == '__main__':
+    mqtt = MqttWrapper()
+    mqtt_node = MqttNode(node_id=config.get('mqtt_water_relay.node_id'))
+    if is_development_mode():
+        mqtt_node.load_module('diagnostics')
+
+    mqtt_node.load_module('temphum')
+    mqtt_relay_module = mqtt_node.load_module('relay')
+
+    mqtt_node.add_payload_callback(mqtt_payload_callback)
+
+    mqtt.connect_and_loop(loop_forever=False)
+
+    bot.enable_logging(BotType.PUMP)
+    bot.run()
+
+    try:
+        mqtt.disconnect()
+    except:
+        pass