cuadratica

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
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 a: 1
>>> Ingrese el término lineal b: 0
>>> Ingrese el término independiente c: 36
>>> No existen raíces
"""

from math import sqrt
from datetime import datetime


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


def calcular_raices(a,b,c):
    delta = b**2 - 4*a*c

    if delta > 0:
        raiz1 = (-1*b + sqrt(delta)) / (2*a)
        raiz2 = (-1*b - sqrt(delta)) / (2*a)
        return [raiz1, raiz2]

    elif delta < 0:
        return "No existen soluciones en el campo real"

    else:
        raiz = -1*b / (2*a)
        return raiz


def imprimir(raices):
    if isinstance(raices, float):
        print("Existe una raíz: {:.2f}".format(raices))

    elif isinstance(raices, list):
        print("Existen dos raíces: {:.2f} y {:.2f}".format(raices[0], raices[1]))

    else:
        print(raices)


def grabar_datos(a,b,c,raices):
    f = open("cuadratica.txt", "a")
    # escribo fecha y hora del cálculo:
    f.write(datetime.now().strftime('%d/%m/%y %H:%M:%S'))
    f.write("\nPará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: {:.2f}\n".format(raices))

    elif isinstance(raices, list):
        f.write("Existen dos raíces: {:.2f} y {:.2f}\n".format(raices[0], raices[1]))

    else:
        f.write(raices + "\n")
    f.write("------------------------------------------------\n")




#################################################

while True:
    print("\nCálculo de raíces de una ecuación cuadrática")
    a,b,c = ingresar_datos()
    raices = calcular_raices(a,b,c)
    imprimir(raices)
    grabar_datos(a,b,c,raices)

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