Untitled

import random
import os

# Listas de dados
nomes = ["Paulo", "Andre", "Renata", "Matheus", "Castanha"]
emails = ["paulo@gmail.com", "andre@uol.com", "renata@yahoo.com", "matheus@hotmail.com", "castanha@outlook.com"]
telefones = ["(48)98432-0047", "(48)98825-9878", "(48)98432-8598", "(48)98856-7777", "(48)98456-7898"]
cidades = ["São José", "Rio de Janeiro", "Belo Horizonte", "Porto Alegre", "Curitiba"]
estados = ["SC", "RJ", "MG", "RS", "PR"]

# Função menu
def show_menu():
    print("--------------------------------------------------------------------------------------")
    print("Bem-vindo(a) ao Gerador de Dados de Testes - Digite 'parar' para finalizar o programa.")
    print("--------------------------------------------------------------------------------------")
    print("Escolha uma ou mais opções a serem geradas aleatoriamente:")
    print("[1] - Nome")
    print("[2] - E-mail")
    print("[3] - Telefone")
    print("[4] - Cidade")
    print("[5] - Estado")
    print("----------------------------------------------------------------")
    
# Função para gerar dados aleatórios 
def gerador_dados(choices):
    dados_gerados = []
    if "1" in choices:
        dados_gerados.append(f"Nome: {random.choice(nomes)}")
    if "2" in choices:
        dados_gerados.append(f"E-mail: {random.choice(emails)}")
    if "3" in choices:
        dados_gerados.append(f"Telefone: {random.choice(telefones)}")
    if "4" in choices:
        dados_gerados.append(f"Cidade: {random.choice(cidades)}")
    if "5" in choices:
        dados_gerados.append(f"Estado: {random.choice(estados)}")
    return dados_gerados

# Função para limpar a tela
def clear_screen():
    if os.name == 'nt':
        os.system('cls')
    else:
        os.system('clear')

# Função principal
def main():
    while True:
        clear_screen()  # Limpa a tela antes de mostrar o menu
        show_menu()
        choices = input("Digite uma das opções 1,2,3,4: ").split(',')
        print("----------------------------------------------------------------")
        choices = [choice.strip() for choice in choices]
        
        if "parar" in choices:
            print("Finalizando o programa.")
            break
        
        data = gerador_dados(choices)
        
        if data:
            for item in data:
                print(item)
                
            save_to_file = input("Deseja salvar esses dados em um arquivo? (s/n): ").strip().lower()
           
            if save_to_file == 's':
                with open("dados.txt", "a") as file:
                    for item in data:
                        file.write(item + "\n")
                print("Dados salvos em 'dados.txt'.")

if __name__ == "__main__":
    main()