choice_selection.py file for choice selection

scripts/services/choice_selection.py

+#
+from scripts.core.handlers import Rectangle, Circle, Square
+
+
+def get_rectangle(calculation_choice):
+    rect_length = int(input("Enter the length"))
+    rect_breadth = int(input("Enter the breadth"))
+    if calculation_choice == 'area':
+        print(f'Area : {Rectangle.Area(rect_length, rect_breadth).display_area()}')
+        return 0
+    elif calculation_choice == 'perimeter':
+        print(f'Perimeter : {Rectangle.Perimeter(rect_length, rect_breadth).display_perimeter()}')
+        return 0
+    else:
+        return 1
+
+
+def get_circle(calculation_choice):
+    circle_radius = int(input("Enter the radius"))
+    if calculation_choice == 'area':
+        print(f'Area : {Circle.Area(circle_radius).display_area()}')
+        return 0
+    elif calculation_choice == 'perimeter':
+        print(f'Perimeter : {Circle.Perimeter(circle_radius).display_perimeter()}')
+        return 0
+    else:
+        return 1
+
+
+def get_square(calculation_choice):
+    square_side = int(input("Enter the side"))
+    if calculation_choice == 'area':
+        print(f'Area : {Square.Area(square_side).display_area()}')
+        return 0
+    elif calculation_choice == 'perimeter':
+        print(f'Perimeter : {Square.Perimeter(square_side).display_perimeter()}')
+        return 0
+    else:
+        return 1

scripts/services/main_service.py

 # Importing modules rectangle, circle, square from scripts
-from scripts.core.handlers import Circle, Rectangle, Square
+from scripts.services.choice_selection import get_rectangle, get_circle, get_square
 
 
 # core code
@@ -15,34 +15,15 @@ def shapes():
 
         # choice is rectangle
         if shape_choice == 'rectangle':
-            rect_length = int(input("Enter the length"))
-            rect_breadth = int(input("Enter the breadth"))
-            if calculation_choice == 'area':
-                print(f'Area : {Rectangle.Area(rect_length, rect_breadth).display_area()}')
-            elif calculation_choice == 'perimeter':
-                print(f'Perimeter : {Rectangle.Perimeter(rect_length, rect_breadth).display_perimeter()}')
-            else:
-                error = 1
+            error = get_rectangle(calculation_choice)
 
         # choice is circle
         elif shape_choice == 'circle':
-            circle_radius = int(input("Enter the radius"))
-            if calculation_choice == 'area':
-                print(f'Area : {Circle.Area(circle_radius).display_area()}')
-            elif calculation_choice == 'perimeter':
-                print(f'Perimeter : {Circle.Perimeter(circle_radius).display_perimeter()}')
-            else:
-                error = 1
+            error = get_circle(calculation_choice)
 
         # choice is square
         elif shape_choice == 'square':
-            square_side = int(input("Enter the side"))
-            if calculation_choice == 'area':
-                print(f'Area : {Square.Area(square_side).display_area()}')
-            elif calculation_choice == 'perimeter':
-                print(f'Perimeter : {Square.Perimeter(square_side).display_perimeter()}')
-            else:
-                error = 1
+            error = get_square(calculation_choice)
 
         # invalid choice
         else: