interface.py

from pacote_modulos import *


def interface():
    """Interface do programa
    """

    escolha = None
    while escolha != 2:
        print("""
=-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-=
=                        Bem vindo ao Py.Bank                        =
=-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-=
[ 1 ] Ver Carteira de Investimento
[ 2 ] Sair
=-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-=
            """)  # menu
        escolha = verifica_int("Escolha sua opção: ")  # A função vai verificar se o número digitado é um int
        if escolha == 1:
            # Aqui o programa irá pedi a URL e verificará se ela existe e funciona.
            url = str(input("Por favor, digite a url da sua carteira: "))
            if verifica_url(url):
                # Caso exista, o módulo que desenvolve o excel com as informações será chamado
                nome_do_arquivo_excel = input("Insira um nome para o arquivo com seu dashboard: ")
                carteira = scrapping.encontra_ativos(url)
                carteira_cotacao = cotacao.cotacao(carteira)
                dashboard.dashboard(carteira_cotacao, nome_do_arquivo_excel)
                # Vai salvar um arquivo .xlsx
        elif escolha == 2:
            break
        else:
            print("Por favor, escolha uma opção válida!")
    print("Obrigado, Volte Sempre!!")


def verifica_int(msg):
    """-> Verifica se o caracter passado no input é um int.

    Args:
        msg (string): Mensagem personalizada solicitando o usuário que digite um número inteiro.
 
    Returns:
        int : número inteiro  
    """  
    while True:
        try:  # Ele tenta transformar a resposta do usuário em inteiro
            inteiro = int(input(msg))
        except:  # Se houver erro, a looping continua até digitar um int
            print('ERRO! por favor, digite um inteiro válido!')

            print('''
=-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-=
[ 1 ] Ver Carteira de Investimento
[ 2 ] Sair
=-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-==-=-=
            ''')
        else:
            break  # Se ele tiver sucesso, o programa segue normalmente.
    return inteiro


def verifica_url(url):
    """Verifica se a url possui uma carteira correta.
 
    Args:
        url (string): url que se deseja verificar se possui uma carteira correta.
 
    Returns:
        bool: Retorna um False ou um True
    """

    try:  # O programa tenta abrir e fechar o arquivo que o usuário passou.
        scrapping.encontra_ativos(url)
    except:  # Se não consegui, ele retorna falso e volta para o menu.
        print(f"Não foi encontrada uma carteira na url passada.")
        return False
    return True