wip:publish branch

Author: Droid-An

.gitignore

@@ -0,0 +1,3 @@
+.DS_Store
+**/.DS_Store
+.venv

laptop_alloctaion.py

@@ -0,0 +1,73 @@
+from dataclasses import dataclass
+from enum import Enum
+from typing import List
+
+
+class OperatingSystem(Enum):
+    MACOS = "macOS"
+    ARCH = "Arch Linux"
+    UBUNTU = "Ubuntu"
+
+
+@dataclass(frozen=True)
+class Person:
+    name: str
+    age: int
+    # Sorted in order of preference, most preferred is first.
+    preferred_operating_system: List[OperatingSystem]
+
+
+@dataclass(frozen=True)
+class Laptop:
+    id: int
+    manufacturer: str
+    model: str
+    screen_size_in_inches: float
+    operating_system: OperatingSystem
+
+
+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,
+    ),
+]
+
+
+def allocate_laptops(
+    people: List[Person], laptops: List[Laptop]
+) -> dict[Person, Laptop]:
+    allocation = {}
+    laptops_copy = laptops.copy()
+    for person in people:
+        for laptop in laptops_copy:
+            for index in range(len(person.preferred_operating_system)):
+                if laptop.operating_system == person.preferred_operating_system:
+                    allocation[person] = laptop
+                # elif
+
+    return allocation