forecast.py

class forecast:
    
  def __init__(self, start_date, number_of_months):
      self.start_date = start_date
      self.number_of_months = number_of_months
      self.months = pd.date_range('2020-01', periods=60, freq='M')
      
      self.contracts = []
      
      self.df = pd.DataFrame(index=self.months)
      self.df['forecast_licences'] = np.zeros(number_of_months)
      self.df['live_contracts'] = np.zeros(number_of_months)
      self.df['total_sales'] = np.zeros(number_of_months)