from typing import Dict
from .contributions import Contributions
[docs]class ScheduledContributions(Contributions): # pylint: disable=too-few-public-methods
"""Contributions which occur at specific years in the life of the portfolio.
:param scheduled_contributions: contributions by year relative to inception of portfolio
"""
def __init__(self, scheduled_contributions: Dict[int, float]):
self.scheduled_contributions = scheduled_contributions
[docs] def get_contribution_for_year(self, year):
if year in self.scheduled_contributions:
return self.scheduled_contributions[year]
return 0