diff options
Diffstat (limited to 'src/program_parser.py')
| -rwxr-xr-x | src/program_parser.py | 104 |
1 files changed, 104 insertions, 0 deletions
diff --git a/src/program_parser.py b/src/program_parser.py new file mode 100755 index 0000000..59f1159 --- /dev/null +++ b/src/program_parser.py @@ -0,0 +1,104 @@ +#!/run/current-system/sw/bin/python +import os +import json +import datetime +import copy +import dataclasses + +input_file_path = '../../content/program.json' +tmp_file_path = '/tmp/teletext/bh_events_to_show_TMP' +output_file_path = '/tmp/teletext/bh_events_to_show' + +title_max_length = 30 +#current_date = datetime.datetime(year=2024, month=7, day=18, hour=14, minute=59) # DEBUG + +@dataclasses.dataclass +class Event: + title: str = None + room: str = None + day: str = None + start_date: datetime.datetime = None + end_date: datetime.datetime = None + icon: str = '?' + time: str = None + duration: datetime.timedelta = None + pretty: str = None + progress: int = None + +icons = { + 'Startent at Info Desk' : "⭐", + 'Speakers Tent': "🏕", + 'Workshop Room': "🛠️", + 'Bar Meetup Area': "🍺", + 'Bar Area': "🍺", + 'External': "🚗", + 'Pyjam.as Village in Party Area': "🐍" +} + + +def get_program() -> list[Event]: + program: list[Event] = [] + # Parse program into data format (interpret datetime, and such) + with open(input_file_path, "r") as json_file: + program_raw = json.load(json_file) + for event_raw in program_raw: + event = Event() + #print(event_raw["title"]) + for d in ["title", "room"]: # 1 to 1 mapping + setattr(event, d, event_raw[d]) + event.start_date = datetime.datetime.strptime(event_raw["date"], "%Y-%m-%dT%H:%M:%S+00:00") + event.start_date += datetime.timedelta(hours=2) # Hvad er vinter tid? + end_bogus = datetime.datetime.strptime(event_raw["duration"],"%H:%M:%S") + event.duration = datetime.timedelta(hours=end_bogus.hour, minutes=end_bogus.minute, seconds=end_bogus.second) + event.end_date = event.start_date + event.duration + #print('Start date: ', event["start_date"]) + #print("End date: ", event["end_date"]) + #print() + program.append(event) + + program = sorted(program, key=lambda x: x.start_date) + return program + +def get_events_to_show(program: list[Event], programs_to_show_count=4) -> list[Event]: + current_date = datetime.datetime.now() + # Get events to show + events_to_show = [] + for event in program: + if len(events_to_show) >= programs_to_show_count: + break + if event.end_date > current_date: + events_to_show.append(event) + + def format_program(event: Event, cli=False) -> str: + if len(event.title) > title_max_length: + event.title = f'{event.title[:title_max_length - 3]}...' # Truncate title by deleting was 1-3 chars and putting ... + else: + event.title += " " * (title_max_length - len(event.title)) # Pad title + event.day = event.start_date.strftime('%a') + start_time = event.start_date.strftime('%H:%M') + end_time = event.end_date.strftime('%H:%M') + event.icon = icons[event.room] + event.time = f"{start_time}-{end_time}" + return f"{event.title} {event.day} {start_time}-{end_time} {event.icon}" + + for event_to_show in events_to_show: + event_to_show.pretty = format_program(event_to_show) + duration_sec = (event_to_show.end_date - event_to_show.start_date).total_seconds() + progress_sec = (datetime.datetime.now() - event_to_show.end_date).total_seconds() + event_to_show.progress = 100 + (progress_sec / duration_sec) * 100 + + return events_to_show + +program: list[Event] = get_program() +events_to_show: list[Event] = get_events_to_show(program) + +# Convert to the output string +events_to_show_string = '\n'.join([a.pretty for a in events_to_show]) +print(f"{events_to_show_string}\n") + +# Write to output file +with open(tmp_file_path, 'w') as tmp_file: + tmp_file.write(events_to_show_string) +os.rename(tmp_file_path, output_file_path) +print(f'Written to {output_file_path} length: {len(events_to_show_string)}') + |
