Adding division

bmm21002 · Feb 12, 2025 · bdbade8 · bdbade8
1 parent 77a0afa
commit bdbade8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/hw01/main.py b/hw01/main.py
@@ -7,10 +7,13 @@ def sub(x,y):
 def mult(x,y):
     return x*y
 
+def div(x,y):
+    return x/y
+
 def calc():
     while(True):
-        operation = input("Enter operation from [+,-,*]: ")
-        if operation =="+" or operation == "-" or operation == "*":
+        operation = input("Enter operation from [+,-,*,/]: ")
+        if operation =="+" or operation == "-" or operation == "*" or operation == "/":
             break
         print("Enter valid operation")
 
@@ -26,6 +29,8 @@ def calc():
     if operation == "*":
         print(mult(x,y))
 
+    if operation == "/":
+        print(div(x,y))
 
 if __name__ == "__main__":
     calc()