michael_sayapin/holiday-cn
8
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: use full filename when generate ics

Browse Source
This commit is contained in:
NateScarlet
2021-10-10 02:26:03 +08:00
parent 2cd76c2dd6
commit 83e814eefd
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
generate_ics.py
View File

@@ -79,5 +79,5 @@ def generate_ics(data, filename):
name = "上班(补" + name + ")" name = "上班(补" + name + ")"
cal.add_component(create_event(name, start, end)) cal.add_component(create_event(name, start, end))
with open(f"{filename}.ics", "wb") as ics: with open(filename, "wb", encoding="utf8") as ics:
ics.write(cal.to_ical()) ics.write(cal.to_ical())

5
update.py
View File

@@ -71,7 +71,7 @@ def update_data(year: int) -> str:
cls=CustomJSONEncoder, cls=CustomJSONEncoder,
) )
generate_ics(data, filename=f"{year}") generate_ics(data, filename=f"{year}.ics")
return filename return filename
@@ -86,7 +86,8 @@ def update_holiday_ics(fr_year, to_year):
big_days.extend(data.get("days")) big_days.extend(data.get("days"))
generate_ics( generate_ics(
{"days": sorted(big_days, key=lambda x: x["date"])}, filename="holiday-cn" {"days": sorted(big_days, key=lambda x: x["date"])},
filename="holiday-cn.ics",
) )