From 96e034d478c87f57f5a6082e0f85139757380e80 Mon Sep 17 00:00:00 2001
From: Anatoly <akopyl@radner.ru>
Date: Sat, 2 Apr 2022 23:32:19 +0300
Subject: [PATCH] Multiple channel support

---
 grab_samples.py | 22 +++++++++++++++-------
 1 file changed, 15 insertions(+), 7 deletions(-)

diff --git a/grab_samples.py b/grab_samples.py
index c673822..3b164a3 100644
--- a/grab_samples.py
+++ b/grab_samples.py
@@ -6,12 +6,18 @@ load_dotenv()
 
 api_id = os.getenv('API_ID')
 api_hash = os.getenv('API_HASH')
-chat = 'tg1337const'
+chats = [
+  'tg1337const',
+  'cyberboec',
+  'Ateobreaking'
+]
 
 def cleanup(msg):
   unwanted_strings = [
     '[1337const | подписаться](https://t.me/tg1337const)',
-    '[1337const | Подписаться](https://t.me/tg1337const)'
+    '[1337const | Подписаться](https://t.me/tg1337const)',
+    '**',
+    '__'
   ]
 
   try:
@@ -25,10 +31,12 @@ def cleanup(msg):
 
 with TelegramClient('bot', api_id, api_hash) as client:
   f = open("data/data.txt", "w")
-  for message in client.iter_messages(chat):
-    try:
-      f.write(cleanup(message.text))
-    except:
-      pass
+
+  for chat in chats:
+    for message in client.iter_messages(chat):
+      try:
+        f.write(cleanup(message.text))
+      except:
+        pass
 
   f.close()