Skip to content

Commit 328716b

Browse files
committed
Improve error handling in enums-exercise.py
- Add sys import for proper error handling - Change error output to stderr instead of stdout - Replace return statement with sys.exit(1) for proper program termination - Ensures invalid input causes program to exit with error code 1
1 parent 120cd4f commit 328716b

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

prep/enums-exercise.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import sys
12
from dataclasses import dataclass
23
from enum import Enum
34
from typing import List
@@ -44,8 +45,8 @@ def create_person_from_input(laptops: List[Laptop]) -> None:
4445
try:
4546
preferred_os = OperatingSystem[os_choice]
4647
except KeyError:
47-
print("Invalid operating system choice!")
48-
return
48+
print("Invalid operating system choice!", file=sys.stderr)
49+
sys.exit(1)
4950

5051
person = Person(name=name, age=age, preferred_operating_system=preferred_os)
5152
possible_laptops = find_possible_laptops(laptops, person)

0 commit comments

Comments
 (0)