You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
2.1 KiB
37 lines
2.1 KiB
import datetime |
|
from chinese_calendar import (is_holiday, is_workday) |
|
|
|
|
|
def get_data_datetime(show_date): |
|
show_date_obj = datetime.datetime.strptime(show_date, '%Y-%m-%d') |
|
target_start_datetime_obj = show_date_obj + datetime.timedelta(hours=6) |
|
target_end_datetime_obj = show_date_obj + datetime.timedelta(hours=29, minutes=59, seconds=59) |
|
data_start_datetime_obj = show_date_obj + datetime.timedelta(days=-7) + datetime.timedelta(hours=6) |
|
data_end_datetime_obj = show_date_obj + datetime.timedelta(hours=5, minutes=59, seconds=59) |
|
history_start = datetime.datetime.strftime(data_start_datetime_obj, '%Y-%m-%d %H:%M:%S') |
|
history_end = datetime.datetime.strftime(data_end_datetime_obj, '%Y-%m-%d %H:%M:%S') |
|
target_start = datetime.datetime.strftime(target_start_datetime_obj, '%Y-%m-%d %H:%M:%S') |
|
target_end = datetime.datetime.strftime(target_end_datetime_obj, '%Y-%m-%d %H:%M:%S') |
|
return history_start, history_end, target_start, target_end |
|
|
|
|
|
def get_date_type(show_date, template_date): |
|
show_date_obj = datetime.datetime.strptime(show_date, '%Y-%m-%d') |
|
template_date_obj = datetime.datetime.strptime(template_date, '%Y-%m-%d') |
|
result = dict() |
|
# 处理参考数据日期 |
|
history_date_list = [(show_date_obj - datetime.timedelta(days=i)) for i in range(7, 0, -1)] |
|
history_date = '、'.join( |
|
[f'{datetime.date.strftime(d, "%Y-%m-%d")}（{"节假日" if is_holiday(d) else "工作日"}）' for d in |
|
history_date_list]) |
|
# 处理目标日期 |
|
show_date = f'{datetime.date.strftime(show_date_obj, "%Y-%m-%d")}（{"节假日" if is_holiday(show_date_obj) else "工作日"}）' |
|
show_date_is_holiday = is_holiday(show_date_obj) |
|
# 处理模板日期 |
|
template_date = f'{datetime.date.strftime(template_date_obj, "%Y-%m-%d")}（{"节假日" if is_holiday(template_date_obj) else "工作日"}）' |
|
template_date_is_holiday = is_holiday(template_date_obj) |
|
return history_date, show_date, show_date_is_holiday, template_date, template_date_is_holiday |
|
|
|
|
|
if __name__ == '__main__': |
|
get_date_type("2026-06-13", "2026-06-01")
|
|
|