Quadratic Formula

quadratic-formula.py

import argparse

"""
Quadratic equation root calculator
"""


def get_arguments():
    parser = argparse.ArgumentParser(
        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
        description="%(prog)s is a tool for finding \
      the roots of a quadratic equation."
    )
    parser.add_argument('-a',
                        dest='a_var',
                        required=True,
                        type=float)
    parser.add_argument('-b',
                        dest='b_var',
                        required=True,
                        type=float)
    parser.add_argument('-c',
                        dest='c_var',
                        required=True,
                        type=float)
    args = parser.parse_args()
    return args


def roots(a, b, c):
    results = {}
    radical = (b**2 - 4*a*c)**0.5
    results['x_1'] = (-b + radical)/(2*a)
    results['x_2'] = (-b - radical)/(2*a)
    return results


if __name__ == '__main__':
    arguments = get_arguments()
    answers = roots(
        arguments.a_var, arguments.b_var, arguments.c_var)
    print(answers)