Как применить правило к переменным даты и времени?
У меня есть следующие типы дней:
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
соответствует указанному правилу.
Как я могу это сделать?
4 ответа
Я решил это так:
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."
Попробуй это
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
Вы можете использовать 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.
Вот быстрая функция для достижения этой цели. Функция берет ваш словарь праздничных дней и будет сравнивать введенную дату с этими выходными и будет возвращать, является ли введенная дата выходным днем, в течение 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'