123 lines
4.5 KiB
Python
123 lines
4.5 KiB
Python
from datetime import date
|
|
import warnings
|
|
|
|
from ..core import ChineseNewYearCalendar, NewYearsDayMixin
|
|
from ..registry_tools import iso_register
|
|
from ..exceptions import CalendarError
|
|
|
|
holidays = {
|
|
2018:
|
|
{
|
|
'Ching Ming Festival': [(4, 5), (4, 6), (4, 7)],
|
|
'Labour Day Holiday': [(4, 29), (4, 30), (5, 1)],
|
|
'Dragon Boat Festival': [(6, 18)],
|
|
'Mid-Autumn Festival': [(9, 24)],
|
|
'New year': [(12, 30), (12, 31)]
|
|
},
|
|
2019:
|
|
{
|
|
'Ching Ming Festival': [(4, 5)],
|
|
'Labour Day Holiday': [(5, 1), (5, 2), (5, 3)],
|
|
'Dragon Boat Festival': [(6, 7)],
|
|
'Mid-Autumn Festival': [(9, 13)]
|
|
},
|
|
2020:
|
|
{
|
|
'Ching Ming Festival': [(4, 4), (4, 5), (4, 6)],
|
|
'Labour Day Holiday': [(5, 1), (5, 2), (5, 3), (5, 4), (5, 5)],
|
|
'Dragon Boat Festival': [(6, 25), (6, 26), (6, 27)],
|
|
'National Day': [(10, 8)]
|
|
},
|
|
}
|
|
|
|
workdays = {
|
|
2018:
|
|
{
|
|
'Spring Festival Shift': [(2, 11), (2, 24)],
|
|
'Ching Ming Festival Shift': [(4, 8)],
|
|
'Labour Day Holiday Shift': [(4, 28)],
|
|
'National Day Shift': [(9, 29), (9, 30)],
|
|
'New year Shift': [(12, 29)]
|
|
},
|
|
2019:
|
|
{
|
|
'Spring Festival Shift': [(2, 2), (2, 3)],
|
|
'Labour Day Holiday Shift': [(4, 28), (5, 5)],
|
|
'National Day Shift': [(9, 29), (10, 12)],
|
|
},
|
|
2020:
|
|
{
|
|
'Spring Festival Shift': [(1, 19), (2, 1)],
|
|
'Labour Day Holiday Shift': [(4, 26), (5, 9)],
|
|
'Dragon Boat Festival Shift': [(6, 28)],
|
|
'National Day Shift': [(9, 27), (10, 10)]
|
|
},
|
|
}
|
|
|
|
|
|
@iso_register('CN')
|
|
class China(NewYearsDayMixin, ChineseNewYearCalendar):
|
|
"China"
|
|
# WARNING: Support 2018, 2019 currently, need update every year.
|
|
# National Days, 10.1 - 10.7
|
|
national_days = [(10, i, "National Day") for i in range(1, 8)]
|
|
FIXED_HOLIDAYS = NewYearsDayMixin.FIXED_HOLIDAYS + tuple(national_days)
|
|
|
|
include_chinese_new_year_eve = True
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.extra_working_days = []
|
|
for year, data in workdays.items():
|
|
for holiday_name, day_list in data.items():
|
|
for v in day_list:
|
|
self.extra_working_days.append(date(year, v[0], v[1]))
|
|
|
|
def get_calendar_holidays(self, year):
|
|
year_min, year_max = min(holidays.keys()), max(holidays.keys())
|
|
warnings.warn(
|
|
"Support years {}-{} currently, need update every year.".format(
|
|
year_min,
|
|
year_max,
|
|
)
|
|
)
|
|
if year not in holidays.keys():
|
|
msg = "Need configure {} for China.".format(year)
|
|
raise CalendarError(msg)
|
|
return super().get_calendar_holidays(year)
|
|
|
|
def get_variable_days(self, year):
|
|
days = super().get_variable_days(year)
|
|
# Spring Festival, eve, 1.1, and 1.2 - 1.6 in lunar day
|
|
for i in range(2, 7):
|
|
days.append((ChineseNewYearCalendar.lunar(year, 1, i),
|
|
"Spring Festival"))
|
|
# other holidays
|
|
for holiday_name, day_list in holidays[year].items():
|
|
for v in day_list:
|
|
days.append((date(year, v[0], v[1]), holiday_name))
|
|
return days
|
|
|
|
def is_working_day(self, day,
|
|
extra_working_days=None, extra_holidays=None):
|
|
extra_working_days = extra_working_days or self.extra_working_days
|
|
return super().is_working_day(day, extra_working_days, extra_holidays)
|
|
|
|
def add_working_days(self, day, delta,
|
|
extra_working_days=None, extra_holidays=None,
|
|
keep_datetime=False):
|
|
extra_working_days = extra_working_days or self.extra_working_days
|
|
return super().add_working_days(day, delta,
|
|
extra_working_days,
|
|
extra_holidays,
|
|
keep_datetime)
|
|
|
|
def sub_working_days(self, day, delta,
|
|
extra_working_days=None, extra_holidays=None,
|
|
keep_datetime=False):
|
|
extra_working_days = extra_working_days or self.extra_working_days
|
|
return super().sub_working_days(day, delta,
|
|
extra_working_days,
|
|
extra_holidays,
|
|
keep_datetime)
|