diff --git a/stock.py b/stock.py index 3ed2ee0..1039fa1 100644 --- a/stock.py +++ b/stock.py @@ -21,3 +21,12 @@ def read_portfolio(filename): rows = csv.reader(f) _ = next(rows) return [Stock(row[0], row[1], row[2]) for row in rows] + + +def print_portfolio(portfolio): + headers = ("name", "shares", "price") + fmtstr = "{0: >10} {1: >10} {2: >10}" + print(fmtstr.format(*headers)) + print("{0:->10} {1:->10} {2:->10}".format("", "", "")) + for stock in portfolio: + print(fmtstr.format(stock.name, stock.shares, stock.price))