# Utility script to port from flatfile to database

import config as cfg
import flatfile as f
import mongo as m
import json

with open(cfg.database) as f:
    data = json.loads(f.read())

for user in data.keys():
    print("\n\nUser:", user)
    print("Portfolio:")

    for stock in data[user]["portfolio"].keys():
        ticker = stock
        amount = data[user]["portfolio"][stock]
        m.add_stock(user, ticker, amount)

    print("Watchlist:")
    for stock in data[user]["watchlist"].keys():
        ticker = stock
        amount = data[user]["watchlist"][stock]
        m.watch(user, ticker, amount)

print("[*] Done!")