add en(dis)bling alerts

Author: yungwine

modules/alert_bot.py

@@ -3,7 +3,7 @@
 import requests
 
 from modules.module import MtcModule
-from mypylib.mypylib import get_timestamp
+from mypylib.mypylib import get_timestamp, print_table, color_print
 from mytoncore import get_hostname
 from mytonctrl.utils import timestamp2utcdatetime
 
@@ -99,6 +99,8 @@ def send_message(self, text: str):
             raise Exception(f"send_message error: {response}")
 
     def send_alert(self, alert_name: str, *args, **kwargs):
+        if not self.alert_is_enabled(alert_name):
+            return
         last_sent = self.get_alert_sent(alert_name)
         time_ = timestamp2utcdatetime(int(time.time()))
         alert = ALERTS.get(alert_name)
@@ -136,15 +138,50 @@ def init(self):
         self.set_global_vars()
         self.inited = True
 
-    def set_alert_sent(self, alert_name: str):
+    def get_alert_from_db(self, alert_name: str):
         if 'alerts' not in self.ton.local.db:
             self.ton.local.db['alerts'] = {}
-        self.ton.local.db['alerts'][alert_name] = int(time.time())
+        if alert_name not in self.ton.local.db['alerts']:
+            self.ton.local.db['alerts'][alert_name] = {'sent': 0, 'enabled': True}
+        return self.ton.local.db['alerts'][alert_name]
+
+    def set_alert_sent(self, alert_name: str):
+        alert = self.get_alert_from_db(alert_name)
+        alert['sent'] = int(time.time())
 
     def get_alert_sent(self, alert_name: str):
-        if 'alerts' not in self.ton.local.db:
-            return 0
-        return self.ton.local.db['alerts'].get(alert_name, 0)
+        alert = self.get_alert_from_db(alert_name)
+        return alert.get('sent', 0)
+
+    def alert_is_enabled(self, alert_name: str):
+        alert = self.get_alert_from_db(alert_name)
+        return alert.get('enabled', True)  # default is True
+
+    def set_alert_enabled(self, alert_name: str, enabled: bool):
+        alert = self.get_alert_from_db(alert_name)
+        alert['enabled'] = enabled
+        self.ton.local.save()
+
+    def enable_alert(self, args):
+        if len(args) != 1:
+            raise Exception("Usage: enable_alert <alert_name>")
+        alert_name = args[0]
+        self.set_alert_enabled(alert_name, True)
+        color_print("enable_alert - {green}OK{endc}")
+
+    def disable_alert(self, args):
+        if len(args) != 1:
+            raise Exception("Usage: disable_alert <alert_name>")
+        alert_name = args[0]
+        self.set_alert_enabled(alert_name, False)
+        color_print("disable_alert - {green}OK{endc}")
+
+    def print_alerts(self, args):
+        table = [['Name', 'Enabled', 'Last sent']]
+        for alert_name in ALERTS:
+            alert = self.get_alert_from_db(alert_name)
+            table.append([alert_name, alert['enabled'], alert['sent']])
+        print_table(table)
 
     def check_db_usage(self):
         usage = self.ton.GetDbUsage()
@@ -226,4 +263,6 @@ def check_status(self):
         self.local.try_function(self.check_adnl_connection_failed)
 
     def add_console_commands(self, console):
-        ...
+        console.AddItem("enable_alert", self.enable_alert, self.local.translate("enable_alert_cmd"))
+        console.AddItem("disable_alert", self.disable_alert, self.local.translate("disable_alert_cmd"))
+        console.AddItem("list_alerts", self.print_alerts, self.local.translate("list_alerts_cmd"))

mytoncore/mytoncore.py

@@ -3089,6 +3089,9 @@ def using_validator(self):
 	def using_liteserver(self):
 		return self.get_mode_value('liteserver')
 
+	def using_alert_bot(self):
+		return self.get_mode_value('alert-bot')
+
 	def Tlb2Json(self, text):
 		# Заменить скобки
 		start = 0

mytonctrl/mytonctrl.py

@@ -135,6 +135,11 @@ def inject_globals(func):
 			module = ControllerModule(ton, local)
 			module.add_console_commands(console)
 
+	if ton.using_alert_bot():
+		from modules.alert_bot import AlertBotModule
+		module = AlertBotModule(ton, local)
+		module.add_console_commands(console)
+
 	console.AddItem("cleanup", inject_globals(cleanup_validator_db), local.translate("cleanup_cmd"))
 	console.AddItem("benchmark", inject_globals(run_benchmark), local.translate("benchmark_cmd"))
 	# console.AddItem("activate_ton_storage_provider", inject_globals(activate_ton_storage_provider), local.translate("activate_ton_storage_provider_cmd"))

mytonctrl/resources/translate.json

@@ -464,6 +464,21 @@
 		"ru": "Удалить пользовательский оверлей",
 		"zh_TW": "刪除自定義覆蓋"
 	},
+	"enable_alert_cmd": {
+		"en": "Enable specific Telegram Bot alert",
+		"ru": "Включить определенное оповещение через Telegram Bot",
+		"zh_TW": "啟用特定的 Telegram Bot 警報"
+	},
+	"disable_alert_cmd": {
+		"en": "Disable specific Telegram Bot alert",
+		"ru": "Отключить определенное оповещение через Telegram Bot",
+		"zh_TW": "禁用特定的 Telegram Bot 警報"
+	},
+	"list_alerts_cmd": {
+		"en": "List all available Telegram Bot alerts",
+		"ru": "Список всех доступных оповещений через Telegram Bot",
+		"zh_TW": "列出所有可用的 Telegram Bot 警報"
+	},
 	"cleanup_cmd": {
 		"en": "Clean node old logs and temp files",
 		"ru": "Очистить старые логи и временные файлы ноды",