from math import log, sqrt, pow
from fractions import Fraction
from decimal import Decimal
from time import sleep

print('Bem vindo(a) ao 1º Programão Geral feito por Nícolas Malaquias\n')
while True:
    opcao = int(input('''[1] Equação do Segundo Grau
[2] Logaritmos
[3] Sair
Sua opção: '''))
    if opcao not in (1, 2, 3, 1001):
        print('Opção invalida! Tente novamente.')

    if opcao == 1:  # Equação do 2º Grau
        def valores(a, b, c):  # Definindo os valores para fazer os calculos
            """
            :param a: Primeiro valor: a²
            :param b: Segundo valor: b
            :param c: Terceiro valor: Número independente.
            :return:
            """
            delta = pow(b, 2) - 4 * a * c

            if delta < 0:
                print('Não há raizes para Δ < 0')
            else:
                bhaskara1 = -(b + sqrt(delta)) / (2 * a)
                bhaskara2 = -(b - sqrt(delta)) / (2 * a)
                if bhaskara1 < 1:
                    bhaskara1 = Fraction(Decimal(bhaskara1))
                if bhaskara2 < 1:
                    bhaskara2 = Fraction(Decimal(bhaskara2))
                print('-=' * 30)
                print(f'O Δ é {delta}\nx1 = {bhaskara1}\nx2 = {bhaskara2}')
                print('-=' * 30)

        while True:
            try:
                valores(a=float(input('Defina o valor de A: ')),
                        b=float(input('Defina o valor de B: ')),
                        c=float(input('Defina o valor de C: ')))
            except ValueError:
                print('Algo deu errado, tente novamente')
            resp = str(input('Deseja continuar? [S/N]: ')).strip().upper()[0]
            while resp not in 'SN':
                resp = str(input('ERRO. Deseja continuar? [S/N]: ')).strip().upper()[0]
            if resp == 'N':
                break
        print('-='*30)

    sleep(0.5)
    if opcao == 2:  # Logaritmos
        def logaritmo(numero, base):
            """
            :param numero: Informe um número
            :param base: Essa é a base ne, uai
            :return: nada aqui.
            """
            if base == 0:
                base = 10
            print('-=' * 30)
            print(f'O logaritmo de {numero} é: {log(numero, base):.4f}')
            print('-=' * 30)


        print('Programa de Logaritmos com bases diferentes!\n')
        while True:
            try:
                logaritmo(numero=int(input('Qual é o número que desejas saber: ')),
                          base=int(input('Qual é a base (Digite 0 para base 10): ')))
            except ZeroDivisionError:
                print('A base não pode ser 1, escolha outro número')
            resp = str(input('Deseja continuar? [S/N]: ')).strip().upper()[0]
            while resp not in 'SN':
                resp = str(input('ERRO. Deseja continuar? [S/N]: ')).strip().upper()[0]
            if resp == 'N':
                break
        sleep(0.5)
    elif opcao == 3:
        break
    elif opcao == 1001:
        print('POGCHAMP EASTER EGG')
        print('-='*30)


print('Obrigado por usar o programa, volte sempre!')