refactor bot setup

Author: yungwine

modules/__init__.py

@@ -59,8 +59,6 @@ class Setting:
     'defaultCustomOverlaysUrl': Setting(None, 'https://ton-blockchain.github.io/fallback_custom_overlays.json', 'Default custom overlays config url'),
     'debug': Setting(None, False, 'Debug mtc console mode. Prints Traceback on errors'),
     'subscribe_tg_channel': Setting('validator', False, 'Disables warning about subscribing to the `TON STATUS` channel'),
-    'BotToken': Setting('alert-bot', None, 'Alerting Telegram bot token'),
-    'ChatId': Setting('alert-bot', None, 'Alerting Telegram chat id'),
     'auto_backup': Setting('validator', None, 'Make validator backup every election'),
     'auto_backup_path': Setting('validator', '/tmp/mytoncore/auto_backups/', 'Path to store auto-backups'),
     'prometheus_url': Setting('prometheus', None, 'Prometheus pushgateway url'),

modules/alert_bot.py

@@ -256,6 +256,22 @@ def test_alert(self, args):
             self.init()
         self.send_message('Test alert')
 
+    def setup_alert_bot(self, args):
+        if len(args) != 2:
+            raise Exception("Usage: setup_alert_bot <bot_token> <chat_id>")
+        self.token = args[0]
+        self.chat_id = args[1]
+        init_alerts()
+        try:
+            self.send_welcome_message()
+            self.ton.local.db['BotToken'] = args[0]
+            self.ton.local.db['ChatId'] = args[1]
+            color_print("setup_alert_bot - {green}OK{endc}")
+        except Exception as e:
+            self.local.add_log(f"Error while sending welcome message: {e}", "error")
+            self.local.add_log(f"If you want the bot to write to a multi-person chat group, make sure the bot is added to that chat group. If it is not - do it and run the command `setup_alert_bot <bot_token> <chat_id>` again.", "info")
+            color_print("setup_alert_bot - {red}Error{endc}")
+
     def send_welcome_message(self):
         message = f"""
 This is alert bot. You have connected validator with ADNL <code>{self.ton.GetAdnlAddr()}</code>.
@@ -275,20 +291,6 @@ def send_welcome_message(self):
 """
         self.send_message(text=message, disable_web_page_preview=True)
 
-    def on_set_chat_id(self, chat_id):
-        self.token = self.ton.local.db.get("BotToken")
-        if self.token is None:
-            raise Exception("BotToken is not set")
-        self.chat_id = chat_id
-        init_alerts()
-        try:
-            self.send_welcome_message()
-            return True
-        except Exception as e:
-            self.local.add_log(f"Error while sending welcome message: {e}", "error")
-            self.local.add_log(f"If you want the bot to write to a multi-person chat group, make sure the bot is added to that chat group. If it is not - do it and run the command `set ChatId <ChatId>` again.", "info")
-            return False
-
     def check_db_usage(self):
         if time.time() - self.last_db_check < 600:
             return
@@ -452,3 +454,4 @@ def add_console_commands(self, console):
         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"))
         console.AddItem("test_alert", self.test_alert, self.local.translate("test_alert_cmd"))
+        console.AddItem("setup_alert_bot", self.setup_alert_bot, self.local.translate("setup_alert_bot_cmd"))

mytonctrl/mytonctrl.py

@@ -896,10 +896,6 @@ def SetSettings(local, ton, args):
 		color_print(f"{{red}} Error: set {name} ... is deprecated and does not work {{endc}}."
 					f"\nInstead, use {{bold}}enable_mode {mode_name}{{endc}}")
 		return
-	if name == 'ChatId':
-		from modules.alert_bot import AlertBotModule
-		if not AlertBotModule(ton, local).on_set_chat_id(value):
-			return
 	force = False
 	if len(args) > 2:
 		if args[2] == "--force":

mytonctrl/resources/translate.json

@@ -499,6 +499,11 @@
 		"ru": "Отправить тестовое оповещение через Telegram Bot",
 		"zh_TW": "通過 Telegram Bot 發送測試警報"
 	},
+	"setup_alert_bot_cmd": {
+		"en": "Setup Telegram Bot for alerts",
+		"ru": "Настроить Telegram Bot для оповещений",
+		"zh_TW": "設置 Telegram Bot 以接收警報"
+	},
 	"benchmark_cmd": {
 		"en": "Run benchmark",
 		"ru": "Запустить бенчмарк",