from beancount.prices.source import SourcePrice
from beancount.prices.sources import yahoo


def div_100(sp: SourcePrice) -> SourcePrice:
    return SourcePrice(sp.price / 100, sp.time, sp.quote_currency)


class Source(yahoo.Source):
    def get_latest_price(self, ticker):
        source_price = super().get_latest_price(ticker)
        return div_100(source_price)

    def get_historical_price(self, ticker, time):
        source_price = super().get_historical_price(ticker, time)
        return div_100(source_price)

__copyright__ = "Copyright (C) 2023-2023  iLeoDo"
__license__ = "GNU GPLv2"

from typing import Dict
import collections
from decimal import Decimal
from beancount.core.data import Commodity, Currency, Price
import beancount.core.amount

__plugins__ = ("price_multiplier",)

import logging
logger = logging.getLogger("price_multiplier")
OverrideError = collections.namedtuple("CommodityError", "source message entry")


def price_multiplier(entries, unused_options_map, config_str=None):

    multiplier_map :Dict[Currency, Decimal] = {}
    for entry in entries:
        if isinstance(entry, Commodity):
            multiplier = entry.meta.get("multiplier", None)
            if multiplier:
                multiplier_map[entry.currency] = Decimal(multiplier)

    def transform(entry):
        if isinstance(entry, Price):
            if entry.currency in multiplier_map:
                override_entry = Price(entry.meta, entry.date, entry.currency, beancount.core.amount.mul(entry.amount, multiplier_map[entry.currency]))
                return override_entry
        return entry
        
    ret_entries = [
        transform(entry)
        for entry in entries
    ]
    errors = []

    return ret_entries, errors