Complete Chapter 5 exercises

rchopra · rchopra · commit 3a738a21cad2 · 2023-11-30T19:19:13.000-06:00
diff --git a/Work/pcost.py b/Work/pcost.py
@@ -8,7 +8,7 @@
 def portfolio_cost(filename):
     """Calculate protfolio cost."""
     rows = report.read_portfolio(filename)
-    return sum([row.shares * row.price for row in rows])
+    return sum([row.cost for row in rows])
 
 
 def main(args):
diff --git a/Work/stock.py b/Work/stock.py
@@ -2,6 +2,8 @@
 
 
 class Stock:
+    __slots__ = ("name", "_shares", "price")
+
     def __init__(self, name, shares, price):
         self.name = name
         self.shares = shares
@@ -10,6 +12,17 @@ def __init__(self, name, shares, price):
     def __repr__(self):
         return f"Stock('{self.name}', {self.shares}, {self.price})"
 
+    @property
+    def shares(self):
+        return self._shares
+
+    @shares.setter
+    def shares(self, value):
+        if not isinstance(value, int):
+            raise TypeError("Expected int")
+        self._shares = value
+
+    @property
     def cost(self):
         return self.shares * self.price
 
