Completed Enums exercise.

Author: anosidium

prep exercises/Enums exercise.py

@@ -0,0 +1,109 @@
+from dataclasses import dataclass
+from enum import Enum
+from typing import List
+import sys
+
+class OperatingSystem(Enum):
+    MACOS = "macOS"
+    ARCH = "Arch Linux"
+    UBUNTU = "Ubuntu"
+
+@dataclass(frozen=True)
+class Person:
+    name: str
+    age: int
+    preferred_operating_system: OperatingSystem
+
+
+@dataclass(frozen=True)
+class Laptop:
+    id: int
+    manufacturer: str
+    model: str
+    screen_size_in_inches: float
+    operating_system: OperatingSystem
+
+
+def find_possible_laptops(laptops: List[Laptop], person: Person) -> List[Laptop]:
+    possible_laptops = []
+    for laptop in laptops:
+        if laptop.operating_system == person.preferred_operating_system:
+            possible_laptops.append(laptop)
+    return possible_laptops
+
+
+people = [
+    Person(name="Imran", age=22, preferred_operating_system=OperatingSystem.UBUNTU),
+    Person(name="Eliza", age=34, preferred_operating_system=OperatingSystem.ARCH),
+]
+
+laptops = [
+    Laptop(id=1, manufacturer="Dell", model="XPS", screen_size_in_inches=13, operating_system=OperatingSystem.ARCH),
+    Laptop(id=2, manufacturer="Dell", model="XPS", screen_size_in_inches=15, operating_system=OperatingSystem.UBUNTU),
+    Laptop(id=3, manufacturer="Dell", model="XPS", screen_size_in_inches=15, operating_system=OperatingSystem.UBUNTU),
+    Laptop(id=4, manufacturer="Apple", model="macBook", screen_size_in_inches=13, operating_system=OperatingSystem.MACOS),
+]
+
+# for person in people:
+#     possible_laptops = find_possible_laptops(laptops, person)
+#     print(f"Possible laptops for {person.name}: {possible_laptops}")
+
+name = input("Please enter your name: ")
+
+def get_age() -> int:
+    while True:
+        try:
+            age_string = input("Please enter your age, you must be >= 18 years old: ")
+            age = int(age_string)
+
+            if age < 18:
+                print("You must be 18+.")
+                continue
+
+            return age
+        except ValueError:
+            print("Please enter a valid number.")
+
+age = get_age()
+
+def get_preferred_operating_system() -> OperatingSystem:
+    while True:
+        preferred_operating_system_input = input("Your preferred operating system (macOS, Ubuntu or Arch Linux): ")
+
+        try:
+            normalised_value = preferred_operating_system_input.strip()
+            operating_system = OperatingSystem(normalised_value)
+            return operating_system
+        except ValueError:
+            print(f"Sorry, we don't have {preferred_operating_system_input}. Please choose from macOS, Ubuntu or Arch Linux.")
+
+operating_system = get_preferred_operating_system()
+
+person = Person(name=name, age=age, preferred_operating_system=operating_system)
+people.append(person)
+
+def find_laptop() -> None:
+    matching_laptops = []
+    non_matching_laptops = []
+
+    for laptop in laptops:
+        if laptop.operating_system == person.preferred_operating_system:
+            matching_laptops.append(laptop)
+        else:
+            non_matching_laptops.append(laptop)
+
+    if matching_laptops:
+            count = len(matching_laptops)
+            print(f"Congratulations, we have found {count} matching laptops!")
+
+            other_laptops_count = len(non_matching_laptops)
+            
+            if other_laptops_count > count:
+                print(f"We have found {other_laptops_count} laptops with other operating systems. You might be interested in switching to a different operating system.")
+            
+            sys.exit(0)
+    else:
+        sys.exit("Sorry, we couldn't find any matching laptops!")
+        
+
+find_laptop()