From 6019ad2cb29de6fb520e2469db33bfba121bb946 Mon Sep 17 00:00:00 2001
From: Evgeny Zinoviev <me@ch1p.io>
Date: Tue, 16 Nov 2021 03:14:18 +0300
Subject: report user requests to solarmon api

---
 src/inverter-bot | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

(limited to 'src/inverter-bot')

diff --git a/src/inverter-bot b/src/inverter-bot
index 8d7203a..744536c 100755
--- a/src/inverter-bot
+++ b/src/inverter-bot
@@ -3,6 +3,7 @@ import logging
 import re
 import datetime
 import json
+import solarmon_api
 
 from typing import Optional, Tuple
 from argparse import ArgumentParser
@@ -29,6 +30,7 @@ from telegram.error import TimedOut
 
 monitor: Optional[InverterMonitor] = None
 updater: Optional[Updater] = None
+solarmon: Optional[solarmon_api.Client] = None
 notify_to: list[int] = []
 LT = escape('<=')
 flags_map = {
@@ -154,6 +156,11 @@ def beautify_table(s):
     return '\n'.join(lines)
 
 
+def solarmon_report(update: Update, message: str = None) -> None:
+    if message is None:
+        message = update.message.text
+    solarmon.log_bot_request(solarmon_api.BotType.INVERTER, update.message.chat_id, message)
+
 #
 # command/message handlers
 #
@@ -197,6 +204,8 @@ def msg_status(update: Update, context: CallbackContext) -> None:
     except Exception as e:
         handle_exc(update, e)
 
+    solarmon_report(update)
+
 
 def msg_generation(update: Update, context: CallbackContext) -> None:
     try:
@@ -236,6 +245,8 @@ def msg_generation(update: Update, context: CallbackContext) -> None:
     except Exception as e:
         handle_exc(update, e)
 
+    solarmon_report(update)
+
 
 def msg_all(update: Update, context: CallbackContext) -> None:
     reply(update, "Command not recognized. Please try again.")
@@ -258,6 +269,8 @@ def on_set_ac_charging_current(update: Update, context: CallbackContext) -> None
         })
         reply(update, usage)
 
+    solarmon_report(update)
+
 
 def on_set_ac_charging_thresholds(update: Update, context: CallbackContext) -> None:
     try:
@@ -278,6 +291,8 @@ def on_set_ac_charging_thresholds(update: Update, context: CallbackContext) -> N
         })
         reply(update, usage)
 
+    solarmon_report(update)
+
 
 def on_set_battery_under_voltage(update: Update, context: CallbackContext) -> None:
     try:
@@ -296,6 +311,8 @@ def on_set_battery_under_voltage(update: Update, context: CallbackContext) -> No
         })
         reply(update, usage)
 
+    solarmon_report(update)
+
 
 def build_flags_keyboard(flags: dict) -> Tuple[str, InlineKeyboardMarkup]:
     keyboard = []
@@ -313,6 +330,7 @@ def on_flags(update: Update, context: CallbackContext) -> None:
     flags = inverter.exec('get-flags')['data']
     text, markup = build_flags_keyboard(flags)
     reply(update, text, reply_markup=markup)
+    solarmon_report(update)
 
 
 def on_status(update: Update, context: CallbackContext) -> None:
@@ -322,6 +340,8 @@ def on_status(update: Update, context: CallbackContext) -> None:
     except Exception as e:
         handle_exc(update, e)
 
+    solarmon_report(update)
+
 
 def on_config(update: Update, context: CallbackContext) -> None:
     try:
@@ -330,6 +350,8 @@ def on_config(update: Update, context: CallbackContext) -> None:
     except Exception as e:
         handle_exc(update, e)
 
+    solarmon_report(update)
+
 
 def on_errors(update: Update, context: CallbackContext) -> None:
     try:
@@ -338,6 +360,8 @@ def on_errors(update: Update, context: CallbackContext) -> None:
     except Exception as e:
         handle_exc(update, e)
 
+    solarmon_report(update)
+
 
 def on_button(update: Update, context: CallbackContext) -> None:
     query = update.callback_query
@@ -370,6 +394,8 @@ def on_button(update: Update, context: CallbackContext) -> None:
         text, markup = build_flags_keyboard(flags)
         query.edit_message_text(text, reply_markup=markup)
 
+        solarmon_report(update, message=query.data)
+
     else:
         query.answer('unexpected callback data')
 
@@ -432,6 +458,7 @@ if __name__ == '__main__':
     parser.add_argument('--inverterd-host', default='127.0.0.1', type=str)
     parser.add_argument('--inverterd-port', default=8305, type=int)
     parser.add_argument('--verbose', action='store_true')
+    parser.add_argument('--solarmon-api-token', type=str, required=True)
     args = parser.parse_args()
 
     whitelist = list(map(lambda x: int(x), args.users_whitelist))
@@ -474,6 +501,10 @@ if __name__ == '__main__':
 
     dispatcher.add_handler(MessageHandler(Filters.all & user_filter, msg_all))
 
+    # create api client instance
+    solarmon = solarmon_api.Client(args.solarmon_api_token, timeout=3)
+    solarmon.enable_async()
+
     # start the bot
     updater.start_polling()
 
-- 
cgit v1.2.3