Source code for tradeexecutor.strategy.chart.standard.interest

"""Interest/vault profit calculations."""
import pandas as pd

from tradeexecutor.analysis.credit import calculate_yield_metrics, YieldType
from tradeexecutor.strategy.chart.definition import ChartInput


def lending_pool_interest_accrued(
    input: ChartInput,
) -> pd.DataFrame:
    """How much our strategy accrued in interest.

    :return: Table with statistics
    """
    state = input.state
    interest_df = calculate_yield_metrics(
        state,
        yield_type=YieldType.credit,
        execution_mode=input.execution_context.mode
    )
    return interest_df


def vault_statistics(
    input: ChartInput,
) -> pd.DataFrame:
    """Yield vault statistics table.

    :return: Table with statistics
    """
    state = input.state
    interest_df = calculate_yield_metrics(
        state,
        yield_type=YieldType.vault,
        execution_mode=input.execution_context.mode
    )
    return interest_df