sq

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
"""
Script que calcula las raices de una ec. cuadrática en el campo de los
nros. reales.
Ej:
>>> Ingrese el tèrmino cuadrático (a): 1
>>> Ingrese el tèrmino lineal (b): 10
>>> Ingrese el tèrmino independiente (c): 5
>>> No existen raìces en el campo de los nros reales
"""

import datetime  # para guardar fecha y hora

########  funciones ###############

def ingresar_datos():
    """ Función que retorna los parámetros ingresados
    por el usario"""
    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):
    from math import sqrt
    delta = b**2 -4*a*c

    if delta == 0:
        raiz = -1*b / (2*a)
        return raiz

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

    else:
        return "No existen raìces en el campo de los nros reales"



def grabar_datos(a,b,c,raices):
    """
    Guardo en el archivo 'cuadratica.txt' fecha, hora, parámetros y
    raíces de cada cálculo efectuado
    """
    f = open("cuadratica.txt", "a")
    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"a: {a}\n")
    f.write(f"b: {b}\n")
    f.write(f"c: {c}\n")
    if isinstance(raices, float):
        f.write("Existe una solución: {:.2f}".format(raices) + "\n")
    elif isinstance(raices, list):
        f.write("Existen dos soluciones: {:.2f} y {:.2f}".format(raices[0],raices[1]) + "\n")
    else:
        f.write(raices + "\n")
    f.close()





def imprimir(raices):
    if isinstance(raices, float):
        print("Existe una solución: {:.2f}".format(raices))
    elif isinstance(raices, list):
        print("Existen dos soluciones: {:.2f} y {:.2f}".format(raices[0],raices[1]))
    else:
        print(raices)



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

print("Programa que calcula las raíces de una parábola")

while True:

    a,b,c = ingresar_datos()
    raices = calcular_raices(a,b,c)
    grabar_datos(a,b,c,raices)
    imprimir(raices)

    opcion = input("Presione cualquier tecla o '1' para salir: ")
    if opcion == "1":
        print("Gracias por utilizar este programa...")
        break