from tabulate import tabulate
from decimal import Decimal
import yfi as y


def generate_table(stonks):
    table_data = []
    total = 0
    for ticker in stonks.keys():
        price = y.get_current_price(ticker)
        shares_count = stonks[ticker]
        table_data.append(
            [ticker, shares_count, round(Decimal(price) * Decimal(stonks[ticker]), 2)]
        )
        total += Decimal(price) * Decimal(stonks[ticker])

    table_data.append(["Total", "", round(total, 2)])

    return str(
        tabulate(table_data, ["Ticker", "Shares", "Value ($)"], tablefmt="pretty")
    )


def watchlist_table(watchlist):

    table_data = []
    header = ["Ticker", "Current Price ($)", "Estimated Price ($)"]

    for stonk in watchlist.keys():
        table_data.append(
            [
                stonk,
                y.get_current_price(stonk),
                watchlist[stonk],
            ]
        )

    return str(tabulate(table_data, header, tablefmt="pretty"))