enums and inheritance exercises

Author: M-Abdoon

prep-exercises/enums.py

@@ -0,0 +1,41 @@
+from enum import Enum
+from dataclasses import dataclass
+
+
+class OS(Enum):
+    UBUNTU = "Ubuntu"
+    MAC = "Mac"
+    WINDOWS = "Windows"
+
+
+@dataclass
+class Laptop:
+    model: str
+    os: OS
+
+
+@dataclass
+class Person:
+    name: str
+    age: int
+    preferred_os: OS
+
+
+laptops = [
+    Laptop("Dell", OS.UBUNTU),
+    Laptop("MacBook", OS.MAC),
+    Laptop("HP", OS.WINDOWS),
+]
+
+person = Person("Ali", 22, OS.UBUNTU)
+
+
+def match(laptops, person):
+    result = []
+    for l in laptops:
+        if l.os == person.preferred_os:
+            result.append(l)
+    return result
+
+
+print(match(laptops, person))

prep-exercises/inheritance.py

@@ -0,0 +1,23 @@
+class Animal:
+    def __init__(self, name):
+        self.name = name
+
+    def speak(self):
+        return "some sound"
+
+
+class Dog(Animal):
+    def speak(self):
+        return "Woof!"
+
+
+class Cat(Animal):
+    def speak(self):
+        return "Meow!"
+
+
+dog = Dog("Rex")
+cat = Cat("Milo")
+
+print(dog.name, dog.speak())
+print(cat.name, cat.speak())