inheritance.py

Author: Xinxi

Sprint5/enums.py

@@ -0,0 +1,65 @@
+import sys
+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 Laptop:
+    id: int
+    operating_system: OperatingSystem
+
+@dataclass(frozen=True)
+class Person:
+    name: str
+    age: int
+    preferred_operating_system: OperatingSystem
+
+laptops = [
+    Laptop(id=1, operating_system=OperatingSystem.ARCH),
+    Laptop(id=2, operating_system=OperatingSystem.UBUNTU),
+    Laptop(id=3, operating_system=OperatingSystem.UBUNTU),
+    Laptop(id=4, operating_system=OperatingSystem.MACOS),
+    Laptop(id=5, operating_system=OperatingSystem.UBUNTU),
+]
+
+def get_user_person() -> Person:
+    name = input("Enter your name: ")
+    
+    try:
+        age = int(input("Enter your age: "))
+    except ValueError:
+        print("Error: Age must be a number.", file=sys.stderr)
+        sys.exit(1)
+
+    os_input = input("Enter preferred OS (macOS, Arch Linux, Ubuntu): ")
+    try:
+        pref_os = OperatingSystem(os_input)
+    except ValueError:
+        print(f"Error: '{os_input}' is not a valid operating system.", file=sys.stderr)
+        sys.exit(1)
+    return Person(name=name, age=age, preferred_operating_system=pref_os)
+
+def main():
+    user = get_user_person()
+    counts = {os: 0 for os in OperatingSystem}
+    for laptop in laptops:
+        counts[laptop.operating_system] += 1
+
+    user_pref = user.preferred_operating_system
+    pref_count = counts[user_pref]
+    print(f"There are {pref_count} laptops available with {user_pref.value}.")
+
+    best_os = max(counts, key=counts.get)
+    max_count = counts[best_os]
+
+    if max_count > pref_count:
+        print(f"Note: If you are willing to use {best_os.value}, we have {max_count} available. "
+              "You're more likely to get one!")
+
+if __name__ == "__main__":
+    main()

Sprint5/generics.py

@@ -0,0 +1,23 @@
+from dataclasses import dataclass
+from typing import List
+
+@dataclass(frozen=True)
+class Person:
+    name: str
+    age: int  # Added the missing field
+    children: List["Person"]
+
+# Inventing ages for the family members
+fatma = Person(name="Fatma", age=8, children=[])
+aisha = Person(name="Aisha", age=10, children=[])
+
+# Imran needs an age as well, and now includes his children
+imran = Person(name="Imran", age=40, children=[fatma, aisha])
+
+def print_family_tree(person: Person) -> None:
+    print(person.name)
+    for child in person.children:
+        # This line now works because 'age' exists in the Person class
+        print(f"- {child.name} ({child.age})")
+
+print_family_tree(imran)

Sprint5/inheritance.py

@@ -0,0 +1,40 @@
+class Parent:
+    def __init__(self, first_name: str, last_name: str):
+        self.first_name = first_name
+        self.last_name = last_name
+
+    def get_name(self) -> str:
+        return f"{self.first_name} {self.last_name}"
+
+
+class Child(Parent):
+    def __init__(self, first_name: str, last_name: str):
+        super().__init__(first_name, last_name)
+        self.previous_last_names = []
+
+    def change_last_name(self, last_name) -> None:
+        self.previous_last_names.append(self.last_name)
+        self.last_name = last_name
+
+    def get_full_name(self) -> str:
+        suffix = ""
+        if len(self.previous_last_names) > 0:
+            suffix = f" (née {self.previous_last_names[0]})"
+        return f"{self.first_name} {self.last_name}{suffix}"
+    
+person1 = Child("Elizaveta", "Alekseeva")
+print(person1.get_name())
+print(person1.get_full_name())
+person1.change_last_name("Tyurina")
+print(person1.get_name())
+print(person1.get_full_name())
+
+person2 = Parent("Elizaveta", "Alekseeva")
+print(person2.get_name())
+print(person2.get_full_name())
+person2.change_last_name("Tyurina")
+print(person2.get_name())
+print(person2.get_full_name())
+
+# The Child can access everything the Parent has because it inherits from it.
+# The parent has no attribute of the Child.

Sprint5/type-guided refactorings.py

@@ -0,0 +1,42 @@
+from dataclasses import dataclass
+from typing import List
+
+@dataclass(frozen=True)
+class Person:
+    name: str
+    age: int
+    preferred_operating_systems: List[str]
+
+
+@dataclass(frozen=True)
+class Laptop:
+    id: int
+    manufacturer: str
+    model: str
+    screen_size_in_inches: float
+    operating_system: str
+
+
+def find_possible_laptops(laptops: List[Laptop], person: Person) -> List[Laptop]:
+    possible_laptops = []
+    for laptop in laptops:
+        if laptop.operating_system in person.preferred_operating_systems:
+            possible_laptops.append(laptop)
+    return possible_laptops
+
+
+people = [
+    Person(name="Imran", age=22, preferred_operating_systems=["Ubuntu", "Arch Linux"]),
+    Person(name="Eliza", age=34, preferred_operating_systems=["Arch Linux"]),
+]
+
+laptops = [
+    Laptop(id=1, manufacturer="Dell", model="XPS", screen_size_in_inches=13, operating_system="Arch Linux"),
+    Laptop(id=2, manufacturer="Dell", model="XPS", screen_size_in_inches=15, operating_system="Ubuntu"),
+    Laptop(id=3, manufacturer="Dell", model="XPS", screen_size_in_inches=15, operating_system="ubuntu"),
+    Laptop(id=4, manufacturer="Apple", model="macBook", screen_size_in_inches=13, operating_system="macOS"),
+]
+
+for person in people:
+    possible_laptops = find_possible_laptops(laptops, person)
+    print(f"Possible laptops for {person.name}: {possible_laptops}")