From e44f9442e7bab474fc65689e3bde4613078c0e87 Mon Sep 17 00:00:00 2001
From: Gurulhu <lucasag@hotmail.com.br>
Date: Wed, 1 Jun 2022 16:18:36 -0300
Subject: [PATCH] Fixed imports for Telegram storage and enabled multiple
 chat-ids

fixed configuration files to reflect new changes.
changed chat-id to chat-ids to better suit multiple IDs.
---
 pystemon.yaml                       |  8 +++++---
 pystemon/storage/telegramstorage.py | 16 ++++++++--------
 2 files changed, 13 insertions(+), 11 deletions(-)

diff --git a/pystemon.yaml b/pystemon.yaml
index be91b8b..4275861 100644
--- a/pystemon.yaml
+++ b/pystemon.yaml
@@ -58,8 +58,10 @@ storage:
   telegram:
     storage-classname:    TelegramStorage
     enable: no            # Enable this alerting engine
+    save: yes
     token: 0              # see https://core.telegram.org/bots/api#authorizing-your-bot
-    chat-id: 0
+    chat-ids:
+      - 0
 
 email:
   alert: no             # Enable/disable email alerts
@@ -178,7 +180,7 @@ site:
     download-url: 'http://paste.org.ru/?{id}'
     pastie-classname: PastiePasteOrgRu
     throttling: 5000
- 
+
   kpaste.net:
     enable: no
     archive-url: 'http://kpaste.net/'
@@ -206,7 +208,7 @@ site:
     archive-regex: '<td><a href="(\d+)" title='
     download-url: 'http://pastebin.gr/paste.php?download&id={id}'
     throttling: 5000
-    
+
   pastebin.pl:
     enable: no
     archive-url: 'https://pastebin.pl/lists'
diff --git a/pystemon/storage/telegramstorage.py b/pystemon/storage/telegramstorage.py
index 96e555a..aa9aa20 100644
--- a/pystemon/storage/telegramstorage.py
+++ b/pystemon/storage/telegramstorage.py
@@ -1,6 +1,7 @@
 
 import logging.handlers
 import requests
+from pystemon.storage import PastieStorage
 
 logger = logging.getLogger('pystemon')
 
@@ -8,7 +9,7 @@ class TelegramStorage(PastieStorage):
 
     def __init_storage__(self, **kwargs):
         self.token = kwargs.get('token')
-        self.chat_id = kwargs.get('chat-id')
+        self.chat_ids = kwargs.get('chat-ids')
 
     def __save_pastie__(self, pastie):
         if pastie.matched:
@@ -26,10 +27,9 @@ def __save_pastie__(self, pastie):
         '''.format(site=pastie.site.name, url=pastie.public_url, matches=pastie.matches_to_regex(), content=pastie.pastie_content.decode('utf8'))
 
             url = 'https://api.telegram.org/bot{0}/sendMessage'.format(self.token)
-            try:
-                logger.debug('Sending message to telegram {} for pastie_id {}'.format(url, pastie.id))
-                requests.post(url, data={'chat_id': self.chat_id, 'text': message})
-            except Exception as e:
-                logger.warning("Failed to alert through telegram: {0}".format(e))
-
-
+            for chat_id in self.chat_ids:
+                try:
+                    logger.debug('Sending message to telegram {} for pastie_id {}'.format(url, pastie.id))
+                    requests.post(url, data={'chat_id': chat_id, 'text': message})
+                except Exception as e:
+                    logger.warning("Failed to alert through telegram: {0}".format(e))