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')
chat = 'tg1337const'

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 message in client.iter_messages(chat):
    try:
      f.write(cleanup(message.text))
    except:
      pass

  f.close()