Changed display_portfolio into a table

ljonkergouw · Nov 25, 2022 · 9d20fb8 · 9d20fb8
1 parent b15e19b
commit 9d20fb8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/PythonTrainingGroupB/Portfolio.py b/PythonTrainingGroupB/Portfolio.py
@@ -36,8 +36,12 @@ def display_portfolio(user_name, portfolios):
     personal_portfolio = get_portfolio(user_name, portfolios)
 
     print(f"Portfolio displayed for user {user_name}:")
-
-    print(f"Your Portfolio contains {len(personal_portfolio['portfolio'])} stocks and your balance is €{personal_portfolio['portfolio']['balance']}!")
+    print(f"Your Portfolio contains {len(personal_portfolio['portfolio']['stocks'])} stocks and your balance is €{personal_portfolio['portfolio']['balance']}!")
+    if len(personal_portfolio['portfolio']['stocks']) > 0:
+        print('Stock \t\t Price \t\t Quantiy')
+        print('----------------------------------')
+        for stock in personal_portfolio['portfolio']['stocks'].keys():
+            print(f"{stock} \t\t {personal_portfolio['portfolio']['stocks'][stock]['price']} \t {personal_portfolio['portfolio']['stocks'][stock]['quantity']}")
 
 def add_money(user_name, portfolios, amount):
     """