forked from zhangjq0908/lunar_China
-
Notifications
You must be signed in to change notification settings - Fork 0
/
holidaywithbirthday.py
151 lines (136 loc) · 5.96 KB
/
holidaywithbirthday.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
from lunardate import LunarDate
from datetime import datetime, timedelta
import hashlib
# 计算复活节的日期
def easter(year):
a = year % 19
b = year // 100
c = year % 100
d = b // 4
e = b % 4
f = (b + 8) // 25
g = (b - f + 1) // 3
h = (19 * a + b - d - g + 15) % 30
i = c // 4
k = c % 4
l = (32 + 2 * e + 2 * i - h - k) % 7
m = (a + 11 * h + 22 * l) // 451
month = (h + l - 7 * m + 114) // 31
day = ((h + l - 7 * m + 114) % 31) + 1
return datetime(year, month, day)
# 辅助函数：计算星期天的日期
def get_nth_weekday(year, month, weekday, nth):
date = datetime(year, month, 1)
count = 0
while count < nth:
if date.weekday() == weekday:
count += 1
if count < nth:
date += timedelta(days=1)
return date
# 辅助函数：生成农历节日的公历日期
def lunar_to_solar(year, month, day):
lunar_date = LunarDate(year, month, day)
return lunar_date.toSolarDate()
# 辅助函数：添加节日到事件列表
def add_event(events, date, name, description=None):
uid = hashlib.md5(f"{name}-{date}".encode()).hexdigest() # 生成唯一 UID
events.append({
"date": datetime.combine(date, datetime.min.time()),
"name": name,
"description": description,
"uid": uid
})
# 生成农历和公历节日的日历
def generate_lunar_calendar(start_year, num_of_years):
events = []
for year in range(start_year, start_year + num_of_years):
# 获取农历节日
mid_autumn = lunar_to_solar(year, 8, 15)
zhongyuan = lunar_to_solar(year, 7, 15)
yuanxiao = lunar_to_solar(year, 1, 15)
qixi = lunar_to_solar(year, 7, 7)
spring_festival = lunar_to_solar(year, 1, 1)
dragon_boat_festival = lunar_to_solar(year, 5, 5)
double_ninth_festival = lunar_to_solar(year, 9, 9)
laba_festival = lunar_to_solar(year, 12, 8)
kitchen_god_day = lunar_to_solar(year, 12, 24)
xiaonian = lunar_to_solar(year, 12, 23)
dongzhi = lunar_to_solar(year, 4, 8)
# 计算除夕
lunar_new_year = lunar_to_solar(year, 1, 1)
chuxi_date = lunar_new_year - timedelta(days=1)
# 公历节日
tomb_sweeping_day = datetime(year, 4, 4)
labor_day = datetime(year, 5, 1)
national_day = datetime(year, 10, 1)
anniversary_baba = datetime(year, 1, 8)
father_day = get_nth_weekday(year, 6, 6, 1) # 父亲节
mother_day = get_nth_weekday(year, 5, 6, 2) # 母亲节
thanksgiving_day = get_nth_weekday(year, 11, 3, 4) # 感恩节
easter_date = easter(year)
valentines_day = datetime(year, 2, 14)
christmas_day = datetime(year, 12, 25)
christmas_eve = datetime(year, 12, 24)
halloween_day = datetime(year, 10, 31)
newyear_day = datetime(year, 1, 1)
youth_day = datetime(year, 5, 4)
children_day = datetime(year, 6, 1)
birthday_zhangsda = datetime(year, 8, 3)
birthday_zhangxda = datetime(year, 8, 9)
birthday_bssheng = datetime(year, 1, 27)
birthday_bsyang = datetime(year, 12, 6)
birthday_mama = datetime(year, 8, 4)
# 添加事件
add_event(events, birthday_zhangsda, "张珊生日")
add_event(events, birthday_mama, "妈妈生日", description="1952年")
add_event(events, birthday_zhangxda, "张晰生日")
add_event(events, birthday_bssheng, "张胜生日")
add_event(events, birthday_bsyang, "杨琳生日")
add_event(events, dongzhi, "佛诞节")
add_event(events, mid_autumn, "中秋节")
add_event(events, zhongyuan, "中元节")
add_event(events, yuanxiao, "元宵节")
add_event(events, qixi, "七夕节")
add_event(events, kitchen_god_day, "灶君上天日")
add_event(events, spring_festival, "春节")
add_event(events, dragon_boat_festival, "端午节")
add_event(events, tomb_sweeping_day, "清明节")
add_event(events, double_ninth_festival, "重阳节(生日)")
add_event(events, laba_festival, "腊八节")
add_event(events, xiaonian, "小年")
add_event(events, chuxi_date, "除夕")
add_event(events, labor_day, "劳动节")
add_event(events, youth_day, "青年节")
add_event(events, children_day, "儿童节")
add_event(events, national_day, "国庆节")
add_event(events, father_day, "父亲节")
add_event(events, mother_day, "母亲节")
add_event(events, thanksgiving_day, "感恩节")
add_event(events, easter_date, "复活节")
add_event(events, valentines_day, "情人节")
add_event(events, christmas_eve, "平安夜")
add_event(events, christmas_day, "圣诞节")
add_event(events, halloween_day, "万圣节")
add_event(events, newyear_day, "元旦节")
add_event(events, anniversary_baba, "爸爸祭日", description="1942年8月26--2012年1月8日")
# 排序
events.sort(key=lambda x: x["date"])
# 创建 iCalendar 数据
ics_data = ""
for event in events:
event_start = event["date"].strftime("%Y%m%d")
event_end = (event["date"] + timedelta(days=1)).strftime("%Y%m%d")
description = f"\nDESCRIPTION:{event['description']}" if event["description"] else ""
event_entry = f"BEGIN:VEVENT\nSUMMARY:{event['name']}{description}\nDTSTART;VALUE=DATE:{event_start}\nDTEND;VALUE=DATE:{event_end}\nUID:{event['uid']}\nEND:VEVENT\n"
ics_data += event_entry
ics_calendar = f"BEGIN:VCALENDAR\nVERSION:2.0\nCALSCALE:GREGORIAN\nPRODID:-//Your Product//Your Calendar//EN\n{ics_data}END:VCALENDAR"
return ics_calendar
# 使用示例
start_year = 2024
num_of_years = 28
ics_result = generate_lunar_calendar(start_year, num_of_years)
# 写入文件
with open("lunar_holidayswithbirthday.ics", "w", encoding="utf-8") as file:
file.write(ics_result)
print("Calendar generated: lunar_holidayswithbirthday.ics")