from datetime import datetime


class PeriodHelper:
    """Period helper for integration tests."""

    def __init__(self):
        """Initialize PeriodHelper class."""
        self.frst_date = "Jan 1999"

    def period_date(self, period_name):
        return datetime.strptime(period_name, '%b %Y').date()

    def period_number(self, period_date):
        return period_date.year * 12 + period_date.month

    def name_to_period_id(self, period_name):
        return self.period_number(
            self.period_date(period_name)
        ) - self.period_number(
            self.period_date(
                self.frst_date)
        ) + 1