# Lesson 3 Task 1: calculator with functions
# Author: Stanislav Mykhailenko
# License: Unlicense

# Return codes:
# 0 - OK
# 1 - Invalid numbers entered
# 2 - Division by zero
# 3 - Attempting to use a root with non-natural degree
# 4 - Attempting to extract a root of a negative number with an even degree
# 5 - Invalid operation entered

import sys

def add (firstNumber, secondNumber): return firstNumber + secondNumber
def substract(firstNumber, secondNumber): return firstNumber - secondNumber
def multiply(firstNumber, secondNumber): return firstNumber * secondNumber
def divide(firstNumber, secondNumber):
	if secondNumber == 0:
		print("Division by zero.")
		sys.exit(2)
	else:
		return firstNumber / secondNumber
def square(number): return number**2

def isNatural(number): return False if number < 1 or not number.is_integer() else True
def isEven(number): return True if (number % 2) == 0 else False
def isNegative(number): return True if number < 0 else False

def root(number, degree):
	if not isNatural(degree):
		print("Can't extract a root with non-natural degree.")
		sys.exit(3)
	elif isEven(degree) and isNegative(number):
		print("Can't extract a root of a negative number with an even degree.")
		sys.exit(4)
	else:
		if number < 0:
			return -abs(number ** (1 / degree))
		else:
			return number ** (1 / degree)

try:
	firstNumber = float(input("Enter the first number: "))
	secondNumber = float(input("Enter the second number: "))
except ValueError:
	print("Invalid numbers entered.")
	sys.exit(1)

print(
'''
Valid operations:
Addition: +
Subtraction: - or −
Multiplication: * or ×
Division: / or ÷
Root: root or √ (extracted from the first number, with the second number being its degree)
Square: square or ² (of both numbers)''', end="\n\n"
)

operation = input("Enter operation: ")

if operation == "+":
	print(add(firstNumber, secondNumber))
elif operation == "-" or operation == "−": # ASCII hyphen-minus or Unicode minus
	print(substract(firstNumber, secondNumber))
elif operation == "*" or operation == "×": # ASCII asterisk or Unicode multiplication sign
	print(multiply(firstNumber, secondNumber))
elif operation == "/" or operation == "÷": # ASCII slash or Unicode division sign
	print(divide(firstNumber, secondNumber))
elif operation == "root" or operation == "√": # Unicode radical symbol
	print(root(firstNumber, secondNumber))
elif operation == "square" or operation == "²": # Unicode superscript 2
	print("Number A square: " + str(square(firstNumber)) + "\nNumber B square: " + str(square(secondNumber)))
else:
	print("Invalid operation.")
	sys.exit(5)