Как применить правило к переменным даты и времени?

Question

У меня есть следующие типы дней:

daytypes = {
    'Holidays_1': (
        date(2017, 4, 20),
        date(2017, 7, 10)
    ),
    'Holidays_2': (
        date(2017, 2, 5),
        date(2017, 5, 12),
        date(2017, 12, 14)
    )
}

Я хочу знать, является ли новая дата рабочим днем ​​(понедельник-пятница), который расположен за 1 день до любого праздника, и это не сам праздник.

Например, у меня есть следующие три datetime переменные:

from datetime import datetime
dt1 = datetime.strptime("2017-02-04 11:12:20.0", "%Y-%m-%d %H:%M:%S.%f")
dt2 = datetime.strptime("2017-05-11 20:00:00.0", "%Y-%m-%d %H:%M:%S.%f")
dt3 = datetime.strptime("2017-02-06 12:00:00.0", "%Y-%m-%d %H:%M:%S.%f")

Только dt2 соответствует указанному правилу.

Как я могу это сделать?

Answer 1

Я решил это так:

holidays = []
for d in daytypes:
    holidays.extend(daytypes[d])

if (dt.isoweekday() >= 1 and dt.isoweekday() <= 5 and date(dt.year, dt.month, dt.day + 1) in holidays):
    print "Date is within 1 day of a holiday."
else:
    print "Date is a weekday and not within 1 day of a holiday."

Answer 2

Попробуй это

from datetime import datetime as date
from datetime import timedelta, datetime

daytypes = {
    'Holidays_1': (
        date(2017, 4, 20),
        date(2017, 7, 10)
    ),
    'Holidays_2': (
        date(2017, 2, 5),
        date(2017, 5, 12),
        date(2017, 12, 14)
    )
}

holidays = []
for d in daytypes:
    holidays.extend(daytypes[d])

dt2 = datetime.strptime("2017-05-11 20:00:00.0", "%Y-%m-%d %H:%M:%S.%f")

for h in holidays:
    if h-dt2 < timedelta(1) and h-dt2 > timedelta(0):
        return True

Answer 3

Вы можете использовать weekday() функция, чтобы узнать, какой день недели является вашим объектом даты. Функция возвращает целое число от 0 (понедельник) до 6 (воскресенье).

Например:

>>> from datetime import datetime, date
>>> dt1 = datetime.strptime("2017-02-04 11:12:20.0", "%Y-%m-%d %H:%M:%S.%f")
>>> holiday1 = date(2017, 02, 05)
>>> # Is the holiday one day ahead of the date?
>>> print date(dt1.year, dt1.month, dt1.day + 1) == holiday1 
True
>>> # Is the date a weekday?
>>> print dt1.weekday() in range(0,4) # Days from 0 to 4 are working days
False

Документация находится здесь: https://docs.python.org/2/library/datetime.html

Редактировать: Вы можете использовать цикл for для перебора каждого из ваших праздников:

for holiday_tuple in daytypes.keys():
    for holiday in holiday_tuple:
        # Do whatever you need here
        print date(dtx.year, dtx.month, dtx.day + 1) == holiday

Где dtx - любой из объявленных объектов даты и времени. Вы также можете использовать дополнительный цикл for для перебора списка объектов datetime.

Answer 4

Вот быстрая функция для достижения этой цели. Функция берет ваш словарь праздничных дней и будет сравнивать введенную дату с этими выходными и будет возвращать, является ли введенная дата выходным днем, в течение 1 дня выходного дня или дня недели, а не одного дня выходного. Вы можете изменить функции, чтобы при необходимости возвращать более или менее подробные данные, но это должно соответствовать потребностям, которые вы описываете.

def check_holidays(holidays, input_date):
    if input_date.weekday() >= 5:
        return 'Is a weekend date'
    for holiday, holiday_dates in holidays.items():
        holiday_range = []
        for h_date in holiday_dates:
            holiday_range.append(h_date)
            holiday_range.append(h_date - timedelta(days=1))
        if input_date.date() in holiday_range:
            return 'Date is within 1 day of a holiday'
    return 'Date is a weekday and not within 1 day of a holiday'