Introduce while loop

hariscats · hariscats · commit aa951c36882d · 2024-01-18T17:00:45.000-05:00
diff --git a/Ch03/try_except.py b/Ch03/try_except.py
@@ -1,9 +1,18 @@
-num_from_user = input("Enter a number: ")
+# Loop until the user enters a valid number
+while True:
+    # Get input from the user
+    user_input = input("Enter a number: ")
 
-try:
-    num = int(num_from_user)
-except ValueError:
-    print("You didn't enter a valid number.")
-    num = 0
+    try:
+        # Try to convert the input to an integer
+        number = int(user_input)
+    except ValueError:
+        # If the input can't be converted to an integer, print an error message and continue the loop
+        print("You didn't enter a valid number. Please try again.")
+        continue
 
-print(f"Your number squared is {num**2}")
+    # If the input was successfully converted to an integer, break the loop
+    break
+
+# Print the square of the number
+print(f"Your number squared is {number**2}")
