cuadratica

"""
Script que calcula las raíces de una ecuación cuadrática en el campo
de los nros reales
Ej:
>>> Ingrese el tèrmino cuadrático: 1
>>> Ingrese el término lineal: 10
>>> Ingrese el término independiente: 5
>>> No existen raíces en el campo de los nros reales
"""
import datetime



def ingresar_datos():
    a = float(input("Ingrese el término cuadrático: "))
    b = float(input("Ingrese el término lineal: "))
    c = float(input("Ingrese el término independiente: "))
    return [a,b,c]




def calcular_raices(a,b,c):
    from math import sqrt
    delta = b**2 - 4*a*c

    if not delta:
        raices = -1*b / (2*a)
        return raices
    elif delta > 0:
        raiz1 = (-1*b + sqrt(delta)) / (2*a)
        raiz2 = (-1*b - sqrt(delta)) / (2*a)
        return [raiz1, raiz2]
    else:
        return "No existen raices en el campo de los nros. reales"


def grabar_datos(a,b,c,raices):
    # escribo fecha y hora:
    f.write(datetime.datetime.now().strftime("%d/%m/%y %H:%M:%S") + "\n")
    f.write("Parámetros de la parábola:\n")
    f.write(f"Término cuadrático (a): {a}\n")
    f.write(f"Término lineal (b): {b}\n")
    f.write(f"Término independiente (c): {c}\n")
    if isinstance(raices, float):
        f.write("Existe una raíz: {:.3f}\n".format(raices))
    elif isinstance(raices, list):
        f.write("Existen dos raíces: {:.3f} y {:.3f}\n".format(raices[0], raices[1]))
    else:
        f.write(raices + "\n")
    f.write(30*"-"+"\n")




def imprimir(raices):
    if isinstance(raices, float):
        print("Existe una raíz: {:.3f}".format(raices))
    elif isinstance(raices, list):
        print("Existen dos raíces: {:.3f} y {:.3f}".format(raices[0], raices[1]))
    else:
        print(raices)


###  main  ##############
print("Script que calcula raíces de una ec.cuadrática en R")

while True:
    a,b,c = ingresar_datos()
    raices = calcular_raices(a,b,c)
    f = open("cuadratica.txt", "a")
    grabar_datos(a,b,c,raices)
    imprimir(raices)

    opcion = input("\nPresione cualquier tecla para continuar(o '1' para salir): ")
    if opcion == "1":
        print("Gracias por usar este programa...")
        f.close()
        break