Skip to content

Commit 0b0f6c3

Browse files
committed
feat: add validation for age
1 parent f019d5d commit 0b0f6c3

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

Perp-exercises/enum8.1.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,11 +112,19 @@ def read_operating_system() -> OperatingSystem:
112112
3: OperatingSystem.MACOS,
113113
}
114114
return os_map[choice]
115-
115+
def read_valid_number(message:str)->int:
116+
117+
while True:
118+
try:
119+
num = int(input(message))
120+
return num
121+
except ValueError:
122+
print("Please enter a valid number.")
123+
116124

117125
def main() -> None:
118126
name: str = input("Name: ")
119-
age: int = int(input("Age: "))
127+
age: int = read_valid_number("Age: ")
120128
preferred_operating_system: OperatingSystem = read_operating_system()
121129
person1: Person = Person(name, age, preferred_operating_system)
122130

0 commit comments

Comments
 (0)