fix mypy

aviate-labs · Dec 1, 2023 · dc6a8b7 · dc6a8b7
1 parent 6e85ad6
commit dc6a8b7
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/node_monitor/bot_slack.py b/node_monitor/bot_slack.py
@@ -28,13 +28,13 @@ def send_message(
 
 
     def send_messages(
-            self, slack_channel_names: List[str],
+            self, slack_channel_names: List[str], subject: str,
             message: str) -> None | SlackApiError:
         """Send a message to multiple Slack channels."""
         # Propagate the last error that occurs
         err = None
         for slack_channel_name in slack_channel_names:
-            this_err = self.send_message(slack_channel_name, message)
+            this_err = self.send_message(slack_channel_name, subject, message)
             if this_err is not None:
                 err = this_err
         return err
diff --git a/node_monitor/bot_telegram.py b/node_monitor/bot_telegram.py
@@ -30,13 +30,13 @@ def send_message(
 
 
     def send_messages(
-            self, chat_ids: List[str], message: str
+            self, chat_ids: List[str], subject: str, message: str
             ) -> None | requests.exceptions.HTTPError:
         """Send a message to multiple Telegram chats."""
         # Propagate the last error that occurs
         err = None
         for chat_id in chat_ids:
-            this_err = self.send_message(chat_id, message)
+            this_err = self.send_message(chat_id, subject, message)
             if this_err is not None:
                 err = this_err
         return err

diff --git a/node_monitor/node_monitor.py b/node_monitor/node_monitor.py
@@ -125,12 +125,12 @@ def broadcaster(node_provider_id: str,
                 if self.slack_bot:
                     channels = slack_channels.get(node_provider_id, [])
                     if channels:
-                        err1 = self.slack_bot.send_messages(channels, message)
+                        err1 = self.slack_bot.send_messages(channels, subject, message)
             if preferences['notify_telegram'] == True:
                 if self.telegram_bot:
                     chats = telegram_chats.get(node_provider_id, [])
                     if chats:
-                        err2 = self.telegram_bot.send_messages(chats, message)
+                        err2 = self.telegram_bot.send_messages(chats, subject, message)
             return None
 
         return broadcaster