diff options
Diffstat (limited to 'generate_icalendar.py')
| -rw-r--r-- | generate_icalendar.py | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/generate_icalendar.py b/generate_icalendar.py index a842f3b..ab18d41 100644 --- a/generate_icalendar.py +++ b/generate_icalendar.py @@ -1,17 +1,21 @@ import json from datetime import datetime import icalendar +import sys -with open("./hololive_birthdays.json", "r", encoding="utf-8") as file: +profile = sys.argv[1] if len(sys.argv) > 1 else "hololive" + +with open(f"./{profile}_birthdays.json", "r", encoding="utf-8") as file: birthdays = json.load(file) calendar = icalendar.Calendar() +title = "hololive & HOLOSTARS" if profile == "hololive" else "NIJISANJI Project" -calendar.add("prodid", "-//Fuwn//hololive & HOLOSTARS Birthday Calendar//EN") +calendar.add("prodid", f"-//Fuwn//{title} Birthday Calendar//EN") calendar.add("version", "2.0") calendar.add("calscale", "GREGORIAN") calendar.add("method", "PUBLISH") -calendar.add("x-wr-calname", "hololive & HOLOSTARS Birthdays") +calendar.add("x-wr-calname", f"{title} Birthdays") calendar.add("x-wr-timezone", "Asia/Tokyo") calendar.add("x-wr-caldesc", "https://github.com/Fuwn/hololist-to-json-and-ical") @@ -34,11 +38,11 @@ for birthday in birthdays: event.add("rrule", {"freq": "yearly"}) event.add( "uid", - f"{birthday['name'].replace(' ', '_').lower()}_{birthday['month']}{birthday['day']}@hololivepro.com", + f"{birthday['name'].replace(' ', '_').lower()}_{birthday['month']}{birthday['day']}@{'hololivepro.com' if profile == 'hololive' else 'nijisanji.jp'}", ) calendar.add_component(event) except ValueError as e: print(e) -with open("hololive_birthdays.ics", "wb") as ics_file: +with open(f"{profile}_birthdays.ics", "wb") as ics_file: ics_file.write(calendar.to_ical()) |