01_ch4_challenge.py

Start/Ch 4/challenge.py

@@ -4,17 +4,33 @@
 # Challenge: convert your classes to dataclasses
 # The subclasses are required to override the magic method
 # that makes them sortable
+from abc import ABC, abstractmethod
+from dataclasses import dataclass, field
 
-class Asset():
-    pass
-
+@dataclass
+class Asset(ABC):
+    price: float
 
+    @abstractmethod
+    def __str__(self):
+        pass
+
+@dataclass(order=True)
 class Stock(Asset):
-    pass
+    ticker: str = field(compare=False)
+    company: str = field(compare=False)
 
+    def __str__(self):
+        return f"{self.ticker}: {self.company} -- ${self.price}"
 
+@dataclass(order=True)
 class Bond(Asset):
-    pass
+    description: str = field(compare=False) 
+    duration: int = field(compare=False) 
+    yieldamt: float
+
+    def __str__(self):
+        return f"{self.description}: {self.duration}yr : ${self.price} : {self.yieldamt}%"
 
 # ~~~~~~~~~ TEST CODE ~~~~~~~~~
 stocks = [