Multiple channel support

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,6 +31,8 @@ def cleanup(msg):
 
 with TelegramClient('bot', api_id, api_hash) as client:
   f = open("data/data.txt", "w")
+
+  for chat in chats:
     for message in client.iter_messages(chat):
       try:
         f.write(cleanup(message.text))