import os
from dotenv import load_dotenv
from telethon.sync import TelegramClient

load_dotenv()

api_id = os.getenv('API_ID')
api_hash = os.getenv('API_HASH')
chats = [
  'tg1337const',
  'cyberboec',
  'Ateobreaking'
]

def cleanup(msg):
  unwanted_strings = [
    '[1337const | подписаться](https://t.me/tg1337const)',
    '[1337const | Подписаться](https://t.me/tg1337const)',
    '**',
    '__'
  ]

  try:
    for unwanted in unwanted_strings:
      msg = msg.replace(unwanted, '')

    return msg
  except:
    return 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))
      except:
        pass

  f.close()