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

refactor: rename functions

Browse Source
This commit is contained in:
NateScarlet 2021-10-10 02:27:50 +08:00
parent 83e814eefd
commit 85248a1ed1
No known key found for this signature in database
GPG Key ID: 5C242793B070309C
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
generate_ics.py
View File

@ -2,7 +2,7 @@ import datetime
from icalendar import Event, Calendar, Timezone, TimezoneStandard from icalendar import Event, Calendar, Timezone, TimezoneStandard
def create_timezone(): def _create_timezone():
tz = Timezone() tz = Timezone()
tz.add("TZID", "Asia/Shanghai") tz.add("TZID", "Asia/Shanghai")
@ -15,7 +15,7 @@ def create_timezone():
return tz return tz
def create_event(event_name, start, end): def _create_event(event_name, start, end):
# 创建事件/日程 # 创建事件/日程
event = Event() event = Event()
event.add("SUMMARY", event_name) event.add("SUMMARY", event_name)
@ -31,7 +31,7 @@ def create_event(event_name, start, end):
return event return event
def ranger(lst): def _iter_date_ranges(lst):
if len(lst) == 0: if len(lst) == 0:
return [] return []
@ -63,21 +63,21 @@ def generate_ics(data, filename):
cal.add("METHOD", "PUBLISH") cal.add("METHOD", "PUBLISH")
cal.add("CLASS", "PUBLIC") cal.add("CLASS", "PUBLIC")
cal.add_component(create_timezone()) cal.add_component(_create_timezone())
days = data.get("days", []) days = data.get("days", [])
for day in days: for day in days:
if isinstance(day.get("date"), str): if isinstance(day.get("date"), str):
day["date"] = datetime.date(*map(int, day["date"].split("-"))) day["date"] = datetime.date(*map(int, day["date"].split("-")))
for fr, to in ranger(days): for fr, to in _iter_date_ranges(days):
start = fr.get("date") start = fr.get("date")
end = to.get("date") + datetime.timedelta(days=1) end = to.get("date") + datetime.timedelta(days=1)
name = fr.get("name") + "假期" name = fr.get("name") + "假期"
if not fr.get("isOffDay"): if not fr.get("isOffDay"):
name = "上班(补" + name + ")" name = "上班(补" + name + ")"
cal.add_component(create_event(name, start, end)) cal.add_component(_create_event(name, start, end))
with open(filename, "wb", encoding="utf8") as ics: with open(filename, "wb", encoding="utf8") as ics:
ics.write(cal.to_ical()) ics.write(cal.to_ical())