# Importando as bibliotecas necessárias
import re

# Abrindo o arquivo lista.txt para leitura
with open("lista.txt", "r") as lista_arquivo:
    # Lendo o conteúdo do arquivo como uma string
    lista_texto = lista_arquivo.read()

# Expressão regular para capturar códigos IMDb
regex_imdb = r"(tt\d{7})"

# Encontrando todos os códigos IMDb na string
codigos_imdb = re.findall(regex_imdb, lista_texto)

# Combinando os códigos IMDb em uma string separada por vírgulas com espaço
codigos_imdb_virgula = ", ".join(codigos_imdb)

# Abrindo o arquivo imdb.txt para escrita
with open("imdb.txt", "w") as imdb_arquivo:
    # Escrevendo a string com os códigos IMDb separados por vírgula com espaço
    imdb_arquivo.write(codigos_imdb_virgula)

# Mensagem de sucesso
print("Códigos IMDb salvos com sucesso no arquivo imdb.txt!")