deposit menu

unrealapex · unrealapex · commit 5e24f875485c · 2022-05-10T01:53:49.000-07:00
diff --git a/lab-11/atm.py b/lab-11/atm.py
@@ -30,11 +30,16 @@ def main_menu(id):
     selection = int(input("Enter your selection here: "))
     if (selection == 1):
         print(eval(f"account_{id}.get_balance()"))
-    else if(selection == 2):
+        get_id()
+    else if (selection == 2):
         withdraw_amount = float(prompt("Enter the amount of money you would like to withdraw: "))
         eval(f"account_{id}.withdraw({withdraw_amount})")
-        print(f"${withdraw_amount} was withdrawed successfully")
-        
-
+        print(f"${withdraw_amount} was withdrawed successfully\nYour new balance is {account_{id}.get_balance()}")
+        get_id()
+    else if (selection == 3):
+        deposit_amount  = float(prompt("Enter the amount of money you would like to deposit: "))
+        eval(f"account_{id}.deposit({deposit_amount})")
+        print(f"${deposit_amount} was deposited  successfully\nYour new balance is {account_{id}.get_balance()}")
+        get_id()
 
 get_id()
