SeasonalityCalibSite.py

import logging
import calendar
from abc import ABCMeta

from calibtool.CalibSite import CalibSite

from ChannelByMultiYearSeasonCohortInsetAnalyzer import ChannelByMultiYearSeasonCohortInsetAnalyzer
from Helpers import get_cases

logger = logging.getLogger(__name__)


class SeasonalityCalibSite(CalibSite):

    __metaclass__ = ABCMeta

    def __init__(self, hfca=None, **kwargs):
        self.metadata = {
        'HFCA': hfca
         }
        if 'throwaway' in kwargs:
            self.throwaway = kwargs['throwaway']
        else :
            self.throwaway = 0
        self.name = hfca

        super(SeasonalityCalibSite, self).__init__(hfca)

    def get_reference_data(self, reference_type):
        super(SeasonalityCalibSite, self).get_reference_data(reference_type)

        # Load the case CSV
        reference_data = get_cases(self.metadata['HFCA'])

        return reference_data

    def get_analyzers(self):

        return [
            ChannelByMultiYearSeasonCohortInsetAnalyzer(site=self, throwaway=self.throwaway)]