diff --git a/flat2mongo.py b/flat2mongo.py index 125b7f9..0c9b2db 100644 --- a/flat2mongo.py +++ b/flat2mongo.py @@ -10,17 +10,19 @@ with open(cfg.database) as f: for user in data.keys(): print("\n\nUser:", user) - print("Portfolio:") + print("[1] Portfolio") for stock in data[user]["portfolio"].keys(): ticker = stock amount = data[user]["portfolio"][stock] + print(" ", ticker, amount) m.add_stock(user, ticker, amount) - print("Watchlist:") + print("[2] Watchlist") for stock in data[user]["watchlist"].keys(): ticker = stock amount = data[user]["watchlist"][stock] + print(" ", ticker, amount) m.watch(user, ticker, amount) print("[*] Done!")