committed successfully

scripts/handler/handling.py

+from scripts.handler.input import Schema1
+from fastapi import HTTPException
+
+
+class Calculator:
+    @staticmethod
+    def calculation(request_data:Schema1):
+        if request_data is not None:
+            if request_data.operation == 1:
+                return "Addition:" + str(request_data.input1 + request_data.input2)
+            elif request_data.operation == 2:
+                return "Substraction:" + str(request_data.input1 - request_data.input2)
+            elif request_data.operation == 3:
+                return "Multiplication:" + str(request_data.input1 * request_data.input2)
+            elif request_data.operation == 4:
+                try:
+                    return "Division:" + str(request_data.input1 / request_data.input2)
+                except ZeroDivisionError:
+                    return "Cannot divide by zero"
+            else:
+                return "Invalid Operation"
+        else:
+            raise HTTPException(status_code=404, detail="data not found")

scripts/handler/input.py

+from pydantic import BaseModel
+
+class Schema1(BaseModel):
+    input1: int
+    input2: int
+    enter_ur_choice: str="1.addition 2.substraction 3.multiplication 4.division"
+    operation: int
\ No newline at end of file