dataclasses and laptopAllocation exercises

Author: M-Abdoon

prep-exercises/dataclasses.py

@@ -0,0 +1,25 @@
+from dataclasses import dataclass
+from datetime import date
+
+
+@dataclass
+class Person:
+    name: str
+    date_of_birth: date
+    preferred_operating_system: str
+
+    def is_adult(self):
+        today = date.today()
+
+        age = today.year - self.date_of_birth.year
+
+        if (today.month, today.day) < (self.date_of_birth.month, self.date_of_birth.day):
+            age -= 1
+
+        return age >= 18
+
+
+imran = Person("Imran", date(2002, 5, 10), "Ubuntu")
+
+print(imran)
+print(imran.is_adult())

prep-exercises/laptopAllocated.py

@@ -0,0 +1,43 @@
+from dataclasses import dataclass
+from typing import List
+from enum import Enum
+
+
+class OperatingSystem(Enum):
+    UBUNTU = "Ubuntu"
+    MAC = "Mac"
+    WINDOWS = "Windows"
+
+
+@dataclass
+class Person:
+    name: str
+    age: int
+    preferred_os: OperatingSystem
+
+
+@dataclass
+class Laptop:
+    model: str
+    os: OperatingSystem
+
+
+def get_matching_laptops(laptops: List[Laptop], person: Person) -> List[Laptop]:
+    matches = []
+
+    for laptop in laptops:
+        if laptop.os == person.preferred_os:
+            matches.append(laptop)
+
+    return matches
+
+
+laptops = [
+    Laptop("Dell", OperatingSystem.UBUNTU),
+    Laptop("MacBook", OperatingSystem.MAC),
+    Laptop("HP", OperatingSystem.WINDOWS),
+]
+
+person = Person("Ali", 22, OperatingSystem.UBUNTU)
+
+print(get_matching_laptops(laptops, person))