From 434c7bec090f33507df0b29ee06774e589a2c260 Mon Sep 17 00:00:00 2001
From: Stanislav Mykhailenko <mykhailenkostanislavzno@gmail.com>
Date: Wed, 19 Oct 2022 20:06:42 +0300
Subject: [PATCH] Make Lesson 1 Task 2 use only one operation, added support
 for any nth roots

---
 Lesson_1/task2.py | 56 ++++++++++++++++++++++++++---------------------
 1 file changed, 31 insertions(+), 25 deletions(-)

diff --git a/Lesson_1/task2.py b/Lesson_1/task2.py
index 964369e..094ffe3 100644
--- a/Lesson_1/task2.py
+++ b/Lesson_1/task2.py
@@ -6,10 +6,11 @@
 # 0 - OK
 # 1 - Invalid numbers entered
 # 2 - Division by zero
-# 3 - Invalid first operation entered
-# 4 - Square root of a negative number calculation
+# 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, math
+import sys
 
 try:
 	numberA = float(input("Enter number A: "))
@@ -18,37 +19,42 @@ except ValueError:
 	print("Invalid numbers entered.")
 	sys.exit(1)
 
-operation = input("Enter operation (+, − or -, × or *, ÷ or /): ")
+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"
+)
 
-# The other part of the task is not clearly defined. It asks to implement square root of a number, or its square, but it can only be done with one number. Assuming it is the result of the first operation and saving it…
-result: float
+operation = input("Enter operation: ")
 
 if operation == "+":
-	result = numberA + numberB
+	print(numberA + numberB)
 elif operation == "-" or operation == "−": # ASCII hyphen-minus or Unicode minus
-	result = numberA - numberB
+	print(numberA - numberB)
 elif operation == "*" or operation == "×": # ASCII asterisk or Unicode multiplication sign
-	result = numberA * numberB
+	print(numberA * numberB)
 elif operation == "/" or operation == "÷": # ASCII slash or Unicode division sign
 	if numberB == 0:
 		print("Division by zero.")
 		sys.exit(2)
 	else:
-		result = numberA / numberB
-else:
-	print("Invalid operation.")
-	sys.exit(3)
-
-print("Result: " + str(result))
-
-# …and asking to do sqrt or square with it.
-secondOperation = input("Do anything else with the result (√ or sqrt, ² or square, any other input will be ignored)? ")
-
-if secondOperation == "√" or secondOperation == "sqrt": # Unicode radical symbol
-	if result < 0:
-		print("Can't get a square root of a negative number.")
+		print(numberA / numberB)
+elif operation == "root" or operation == "√": # Unicode radical symbol
+	if numberB < 1 or not numberB.is_integer():
+		print("Can't extract a root with non-natural degree.")
+		sys.exit(3)
+	elif (numberB % 2) == 0 and numberA < 0:
+		print("Can't extract a root of a negative number with an even degree.")
 		sys.exit(4)
 	else:
-		print(math.sqrt(result))
-elif secondOperation == "²" or secondOperation == "square": # Unicode superscript 2
-	print(math.pow(result, 2))
+		print(numberA ** (1 / numberB))
+elif operation == "square" or operation == "²": # Unicode superscript 2
+	print("Number A square: " + str(numberA ** 2) + "\nNumber B square: " + str(numberB ** 2))
+else:
+	print("Invalid operation.")
+	sys.exit(5)