# -*- coding: utf-8 -*-

#  python-holidays
#  ---------------
#  A fast, efficient Python library for generating country, province and state
#  specific sets of holidays on the fly. It aims to make determining whether a
#  specific date is a holiday as fast and flexible as possible.
#
#  Author:  ryanss <ryanssdev@icloud.com> (c) 2014-2017
#           dr-prodigy <maurizio.montel@gmail.com> (c) 2017-2020
#  Website: https://github.com/dr-prodigy/python-holidays
#  License: MIT (see LICENSE file)

from datetime import date

from dateutil.easter import easter
from dateutil.relativedelta import relativedelta as rd, TH, FR, MO
from holidays.constants import JAN, FEB, MAR, APR, MAY, JUN, JUL, AUG, SEP,\
    OCT, NOV, DEC
from holidays.constants import SUN
from holidays.holiday_base import HolidayBase


class Spain(HolidayBase):
    PROVINCES = ['AN', 'AR', 'AS', 'CB', 'CM', 'CL', 'CT', 'VC',
                 'EX', 'GA', 'IB', 'CN', 'MD', 'MC', 'ML', 'NC', 'PV', 'RI']

    def __init__(self, **kwargs):
        self.country = 'ES'
        self.prov = kwargs.pop('prov', kwargs.pop('state', ''))
        HolidayBase.__init__(self, **kwargs)

    def _is_observed(self, date_holiday, name_holiday):
        if self.observed and date_holiday.weekday() == SUN:
            self[date_holiday + rd(days=+1)] = name_holiday + " (Trasladado)"
        else:
            self[date_holiday] = name_holiday

    def _populate(self, year):
        self._is_observed(date(year, JAN, 1), "Año nuevo")
        self._is_observed(date(year, JAN, 6), "Epifanía del Señor")

        if year < 2015 and self.prov and self.prov in \
                ['AR', 'CL', 'CM', 'EX', 'GA', 'MD', 'ML', 'MC', 'NC',
                 'PV', 'VC']:
            self._is_observed(date(year, MAR, 19), "San José")
        elif year == 2015 and self.prov and self.prov in \
                ['CM', 'MD', 'ML', 'MC', 'NC', 'PV', 'VC']:
            self._is_observed(date(year, MAR, 19), "San José")
        elif year == 2016 and self.prov and self.prov in \
                ['ML', 'MC', 'PV', 'VC']:
            self._is_observed(date(year, MAR, 19), "San José")
        elif year == 2017 and self.prov and self.prov in ['PV']:
            self._is_observed(date(year, MAR, 19), "San José")
        elif 2018 <= year <= 2019 and self.prov and self.prov in \
                ['GA', 'MC', 'NC', 'PV', 'VC']:
            self._is_observed(date(year, MAR, 19), "San José")
        elif 2020 <= year <= 2025 and self.prov and self.prov in \
                ['CM', 'GA', 'MC', 'NC', 'PV', 'VC']:
            self._is_observed(date(year, MAR, 19), "San José")
        if self.prov and self.prov not in ['CT', 'VC']:
            self[easter(year) + rd(weeks=-1, weekday=TH)] = "Jueves Santo"
            self[easter(year) + rd(weeks=-1, weekday=FR)] = "Viernes Santo"
        if self.prov and self.prov in \
                ['CT', 'PV', 'NC', 'VC', 'IB', 'CM']:
            self[easter(year) + rd(weekday=MO)] = "Lunes de Pascua"
        self._is_observed(date(year, MAY, 1), "Día del Trabajador")
        if self.prov and self.prov in ['CT', 'GA', 'VC']:
            self._is_observed(date(year, JUN, 24), "San Juan")
        self._is_observed(date(year, AUG, 15), "Asunción de la Virgen")
        self._is_observed(date(year, OCT, 12), "Día de la Hispanidad")
        self._is_observed(date(year, NOV, 1), "Todos los Santos")
        self._is_observed(date(year, DEC, 6), "Día de la Constitución "
                                              "Española")
        self._is_observed(date(year, DEC, 8), "La Inmaculada Concepción")
        self._is_observed(date(year, DEC, 25), "Navidad")
        if self.prov and self.prov in ['CT', 'IB']:
            self._is_observed(date(year, DEC, 26), "San Esteban")
        # Provinces festive day
        if self.prov:
            if self.prov == 'AN':
                self._is_observed(date(year, FEB, 28), "Día de Andalucia")
            elif self.prov == 'AR':
                self._is_observed(date(year, APR, 23), "Día de San Jorge")
            elif self.prov == 'AS':
                self._is_observed(date(year, SEP, 8), "Día de Asturias")
            elif self.prov == 'CB':
                self._is_observed(date(year, JUL, 28), "Día de las Instituci"
                                                       "ones de Cantabria")
            elif self.prov == 'CM':
                self._is_observed(date(year, MAY, 31), "Día de Castilla "
                                                       "La Mancha")
            elif self.prov == 'CL':
                self._is_observed(date(year, APR, 23), "Día de Castilla y "
                                                       "Leon")
            elif self.prov == 'CT':
                self._is_observed(date(year, SEP, 11), "Día Nacional de "
                                                       "Catalunya")
            elif self.prov == 'VC':
                self._is_observed(date(year, OCT, 9), "Día de la Comunidad "
                                                      "Valenciana")
            elif self.prov == 'EX':
                self._is_observed(date(year, SEP, 8), "Día de Extremadura")
            elif self.prov == 'GA':
                self._is_observed(date(year, JUL, 25), "Día Nacional de "
                                                       "Galicia")
            elif self.prov == 'IB':
                self._is_observed(date(year, MAR, 1), "Día de las Islas "
                                                      "Baleares")
            elif self.prov == 'CN':
                self._is_observed(date(year, MAY, 30), "Día de Canarias")
            elif self.prov == 'MD':
                self._is_observed(date(year, MAY, 2), "Día de Comunidad de "
                                                      "Madrid")
            elif self.prov == 'MC':
                self._is_observed(date(year, JUN, 9), "Día de la Región de "
                                                      "Murcia")
            elif self.prov == 'NC':
                self._is_observed(date(year, SEP, 27), "Día de Navarra")
            elif self.prov == 'PV':
                self._is_observed(date(year, OCT, 25), "Día del Páis Vasco")
            elif self.prov == 'RI':
                self._is_observed(date(year, JUN, 9), "Día de La Rioja")


class ES(Spain):
    pass


class ESP(Spain):
    pass