inverter_bot: setosp command

1 files changed, 91 insertions, 0 deletions

diff --git a/src/inverter_bot.py b/src/inverter_bot.py
index 0807b23..d4749d2 100755
--- a/src/inverter_bot.py
+++ b/src/inverter_bot.py
@@ -4,6 +4,7 @@ import re
 import datetime
 import json
 
+from enum import Enum
 from inverterd import Format, InverterError
 from html import escape
 from typing import Optional, Tuple, Union
@@ -51,7 +52,14 @@ flags_map = {
     'fault_code_record': 'FTCR',
 }
 logger = logging.getLogger(__name__)
+
 SETACMODE_STARTED, = range(1)
+SETOSP_STARTED, = range(1)
+
+
+class OutputSourcePriority(Enum):
+    SolarUtilityBattery = 'SUB'
+    SolarBatteryUtility = 'SBU'
 
 
 def monitor_charging(event: ChargingEvent, **kwargs) -> None:
@@ -283,6 +291,14 @@ def setacmode(mode: ACMode):
     inverter.exec('set-max-ac-charge-current', (0, a))
 
 
+def setosp(sp: OutputSourcePriority):
+    logger.debug(f'setosp: sp={sp}')
+    inverter.exec('set-output-charge-priority', (sp,))
+
+
+# /setacmode
+# ----------
+
 def setacmode_start(ctx: Context) -> None:
     if monitor.active_current is not None:
         raise RuntimeError('generator charging program is active')
@@ -338,6 +354,60 @@ def setacmode_cancel(ctx: Context):
     return ConversationHandler.END
 
 
+# /setosp
+# ------
+
+def setosp_start(ctx: Context) -> None:
+    buttons = []
+    for sp in OutputSourcePriority:
+        buttons.append(ctx.lang(f'setosp_{sp.value.lower()}'))
+    markup = ReplyKeyboardMarkup([buttons, [ctx.lang('cancel')]], one_time_keyboard=False)
+
+    ctx.reply(ctx.lang('select_ac_mode'), markup=markup)
+    return SETOSP_STARTED
+
+
+def setosp_input(ctx: Context):
+    selected_sp = None
+    for sp in OutputSourcePriority:
+        if ctx.text == ctx.lang(f'setosp_{sp.value.lower()}'):
+            selected_sp = sp
+            break
+
+    if selected_sp is None:
+        raise ValueError('invalid sp')
+
+    # apply the mode
+    setosp(selected_sp)
+
+    # reply to user
+    ctx.reply(ctx.lang('saved'), markup=IgnoreMarkup())
+
+    # notify other users
+    bot.notify_all(
+        lambda lang: bot.lang.get('osp_changed_notification', lang,
+                                  ctx.user.id, ctx.user.name,
+                                  bot.lang.get(str(selected_sp.value), lang)),
+        exclude=(ctx.user_id,)
+    )
+
+    bot.start(ctx)
+    return ConversationHandler.END
+
+
+def setosp_invalid(ctx: Context):
+    ctx.reply(ctx.lang('invalid_mode'), markup=IgnoreMarkup())
+    return SETOSP_STARTED
+
+
+def setosp_cancel(ctx: Context):
+    bot.start(ctx)
+    return ConversationHandler.END
+
+
+# other
+# -----
+
 def setbatuv(ctx: Context) -> None:
     try:
         v = float(ctx.args[0])
@@ -423,6 +493,7 @@ class InverterBot(Wrapper):
             done="Готово",
             unexpected_callback_data="Ошибка: неверные данные",
             select_ac_mode="Выберите режим:",
+            select_priortiy="Установите приоритет:",
             invalid_input="Неверное значение",
 
             flags_press_button='Нажмите кнопку для переключения настройки',
@@ -457,6 +528,9 @@ class InverterBot(Wrapper):
             setgenct_dv=f'напряжение отключения заряда, 48 {LT} DV {LT} 58',
             setgencc_a='максимальный ток заряда, допустимые значения: %s',
 
+            setosp_sub='Solar-Utility-Battery',
+            setosp_sbu='Solar-Battery-Utility',
+
             # monitor
             chrg_evt_started='✅ Начали заряжать от генератора.',
             chrg_evt_finished='✅ Зарядили. Генератор пора выключать.',
@@ -477,6 +551,7 @@ class InverterBot(Wrapper):
 
             # other notifications
             ac_mode_changed_notification='Пользователь <a href="tg://user?id=%d">%s</a> установил режим AC: <b>%s</b>.',
+            osp_changed_notification='Пользователь <a href="tg://user?id=%d">%s</a> установил приоритет источника питания нагрузки: <b>%s</b>.',
 
             bat_state_normal='Нормальный',
             bat_state_low='Низкий',
@@ -493,6 +568,7 @@ class InverterBot(Wrapper):
             done="Done",
             unexpected_callback_data="Unexpected callback data",
             select_ac_mode="Select AC input mode:",
+            select_priortiy="Select priority:",
             invalid_input="Invalid input",
 
             flags_press_button='Press a button to toggle a flag.',
@@ -527,6 +603,9 @@ class InverterBot(Wrapper):
             setgenct_dv=f'discharging voltage, 48 {LT} DV {LT} 58',
             setgencc_a='max charging current, allowed values: %s',
 
+            setosp_sub='Solar-Utility-Battery',
+            setosp_sbu='Solar-Battery-Utility',
+
             # monitor
             chrg_evt_started='✅ Started charging from AC.',
             chrg_evt_finished='✅ Finished charging, it\'s time to stop the generator.',
@@ -547,6 +626,7 @@ class InverterBot(Wrapper):
 
             # other notifications
             ac_mode_changed_notification='User <a href="tg://user?id=%d">%s</a> set AC mode to <b>%s</b>.',
+            osp_changed_notification='Пользователь <a href="tg://user?id=%d">%s</a> set output source priority: <b>%s</b>.',
 
             bat_state_normal='Normal',
             bat_state_low='Low',
@@ -585,6 +665,17 @@ class InverterBot(Wrapper):
             fallbacks=[MessageHandler(self.user_filter & cancel_filter, self.wrap(setacmode_cancel))]
         ))
 
+        self.add_handler(ConversationHandler(
+            entry_points=[CommandHandler('setosp', self.wrap(setosp_start), self.user_filter)],
+            states={
+                SETOSP_STARTED: [
+                    *[MessageHandler(text_filter(self.lang.all(f'setosp_{sp.value.lower()}')), self.wrap(setosp_input)) for sp in OutputSourcePriority],
+                    MessageHandler(self.user_filter & ~cancel_filter, self.wrap(setosp_invalid))
+                ]
+            },
+            fallbacks=[MessageHandler(self.user_filter & cancel_filter, self.wrap(setosp_cancel))]
+        ))
+
         super().run()
 
     def markup(self, ctx: Optional[Context]) -> Optional[ReplyKeyboardMarkup]: