Merge pull request #364 from swar/telegram-notifications

Adding Telegram Notifications

config.yaml.default

@@ -57,7 +57,11 @@ notifications:
   notify_pushover: false
   pushover_user_key: xx
   pushover_api_key: xx
-
+
+  # TELEGRAM
+  notify_telegram: false
+  telegram_token: xxxxx
+
   # TWILIO
   notify_twilio: false
   twilio_account_sid: xxxxx

plotmanager/library/parse/configuration.py

@@ -70,7 +70,7 @@ def _get_notifications_settings(config):
         raise InvalidYAMLConfigException('Failed to find notifications parameter in the YAML.')
     notifications = config['notifications']
     expected_parameters = ['notify_discord', 'discord_webhook_url', 'notify_sound', 'song', 'notify_pushover',
-                           'pushover_user_key', 'pushover_api_key']
+                           'pushover_user_key', 'pushover_api_key', 'telegram_token']
     _check_parameters(parameter=notifications, expected_parameters=expected_parameters, parameter_type='notification')
     return notifications
 

plotmanager/library/utilities/configuration.py

@@ -27,6 +27,13 @@ def test_configuration(chia_location, notification_settings, instrumentation_set
             raise MissingImportError('Failed to find import "pushover". Be sure to run "pip install -r '
                                      'requirements-notification.txt".')
 
+    if instrumentation_settings.get('notify_telegram'):
+        try:
+            import telegram_notifier
+        except ImportError:
+            raise MissingImportError('Failed to find import "telegram_notifier". Be sure to run "pip install -r '
+                                     'requirements-notification.txt".')
+
     if instrumentation_settings.get('prometheus_enabled'):
         try:
             import prometheus_client

plotmanager/library/utilities/notifications.py

@@ -12,6 +12,11 @@ def _send_notifications(title, body, settings):
         import pushover
         client = pushover.Client(settings.get('pushover_user_key'), api_token=settings.get('pushover_api_key'))
         client.send_message(body, title=title)
+
+    if settings.get('notify_telegram') is True:
+        import telegram_notifier
+        notifier = telegram_notifier.TelegramNotifier(settings.get('telegram_token'), parse_mode="HTML")
+        notifier.send(body)
 
 
 def send_notifications(title, body, settings):

requirements-notification.txt

@@ -1,4 +1,5 @@
 discord_notify==1.0.0
 playsound==1.2.2
 prometheus-client==0.10.1
-python_pushover==0.4
+python_pushover==0.4
+telegram_notifier==0.2