#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
from math import sqrt

"""
Programa que calcula las raíces de una parábola en el campo de los nros
 reales usando la ecuaciòn cuadrática:
 x1,2= [-b +/- raiz(b**2 - 4ac)] / (2a)
"""

def cargar_datos():
    a = float(input("Ingrese a: "))
    b = float(input("Ingrese b: "))
    c = float(input("Ingrese c: "))
    return [a,b,c]


def calcular_raices(a,b,c):
    delta = b ** 2 - 4 * a * c
    if not delta: # if delta == 0:
        raiz = -b / (2 * a)
        return raiz
    elif delta > 0:
        raiz1 = (-b + sqrt(delta)) / (2 * a)
        raiz2 = (-b - sqrt(delta)) / (2 * a)
        return [raiz1, raiz2]
    else:
        return "No existen raíces en el campo de los nros reales"


def imprimir_raices(raices):
    if isinstance(raices, float):
        print(f"Raiz: {raices}")
        f = open("salida.txt","a")
        f.write(f"Raiz: {raices}\n")
        f.close()
    elif isinstance(raices, list):
        print(f"Raiz 1: {raices[0]}, Raiz 2: {raices[1]}")
        f = open("salida.txt","a")
        f.write(f"Raiz 1: {raices[0]}, Raiz 2: {raices[1]}\n")
        f.close()
    else:
        print(raices)
        f = open("salida.txt","a")
        f.write(f"{raices}\n")
        f.close()



# título que aparece
# al comenzar a ejecutar el programa
print("Programa que calcula las raices de una parábola:")
print("------------------------------------------------")
while True:
    a,b,c = cargar_datos()
    raices = calcular_raices(a,b,c)
    imprimir_raices(raices)
    opcion = input("Presione cualquier tecla ('X' para salir): ")
    if opcion.casefold() == "x":
        print("Gracias por usar este programa...")
        break