SuperNovaa41/metro-schedule-parser
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cleans up the python script

Browse Source
This commit is contained in:
SuperNovaa41 2023-06-17 16:53:19 -04:00
parent 7fb9c08770
commit a96b0f6544
1 changed files with 4 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
get_schedule.py
View File

@ -6,10 +6,6 @@ def get_id():
with open("id.txt", "r") as file: with open("id.txt", "r") as file:
return file.read() return file.read()
def cleanse_hour(hour):
split_hour = hour.split(":", 1)
return (split_hour[0] if (len(split_hour[0]) == 2) else "0" + split_hour[0]) + split_hour[1]
def get_shift(schedule_day): def get_shift(schedule_day):
shift = schedule_day["DailyShift"][0] shift = schedule_day["DailyShift"][0]
if (shift == "-"): if (shift == "-"):
@ -41,18 +37,15 @@ with open("calendar.ics", "w") as file:
file.write("VERSION:2.0\n") file.write("VERSION:2.0\n")
file.write("PRODID:SuperNovaa41\n") file.write("PRODID:SuperNovaa41\n")
for i in range(-1, -8, -1): for i in range(-1, -8, -1):
res = get_shift(total_schedule["WorkTime"][i]) res = get_shift(total_schedule["WorkTime"][i])
if (res is None): if (res is None):
continue continue
with open("calendar.ics", "a") as file:
file.write("BEGIN:VEVENT\n") file.write("BEGIN:VEVENT\n")
file.write("DTSTART:" + res[0] + "\n") file.write("DTSTART:" + res[0] + "\n")
file.write("DTEND:" + res[1] + "\n") file.write("DTEND:" + res[1] + "\n")
file.write("SUMMARY: Work (" + res[2] + ")\n") file.write("SUMMARY: Work (" + res[2] + ")\n")
file.write("END:VEVENT\n") file.write("END:VEVENT\n")
with open("calendar.ics", "a") as file:
file.write("END:VCALENDAR\n") file.write("END:VCALENDAR\n")