This commit is contained in:
Mike Bloy 2023-10-27 16:43:51 -05:00
parent 80976ba906
commit 24d06f2169

View File

@ -1,3 +1,6 @@
import csv
class Stock: class Stock:
def __init__(self, name, shares, price): def __init__(self, name, shares, price):
self.name = name self.name = name
@ -11,3 +14,10 @@ class Stock:
self.shares -= num self.shares -= num
if self.shares < 0: if self.shares < 0:
self.shares = 0 self.shares = 0
def read_portfolio(filename):
with open(filename) as f:
rows = csv.reader(f)
_ = next(rows)
return [Stock(row[0], row[1], row[2]) for row in rows]