calc.py

from art import logo


def add(n1, n2):
    return n1 + n2


def subtract(n1, n2):
    return n1 - n2


def multiply(n1, n2):
    return n1 * n2


def divide(n1, n2):
    return n1 / n2


operations = {"+": add, "-": subtract, "*": multiply, "/": divide}
ans = ""


def calculator():
    print(logo)
    num1 = float(input("What's the first number?: "))
    should_continue = True
    ans = ""

    while should_continue:
        operation_symbol = input("Pick an operation (+ - * /) ")
        num2 = float(input("What's the next number?: "))
        calculation_function = operations[operation_symbol]
        answer = calculation_function(num1, num2)
        print(f"{num1} {operation_symbol} {num2} = {answer}")

        ans = input(
            f"Type 'y' to continue calculating with {answer}, or type 'n' to start a new calculation, or type 'e' to exit "
        ).lower()
        if ans == 'y':
            num1 = answer
        elif ans == 'n':
            should_continue = False
            calculator()
        else:
            should_continue = False
            print("Thank you !")


calculator()