feat(playout): replace schedule event dicts with objects

This commit is contained in:
jo 2023-03-04 21:50:12 +01:00 committed by Kyle Robbertze
parent 684e7a6f24
commit a1db2a157a
11 changed files with 646 additions and 667 deletions

View file

@ -2,7 +2,6 @@ from datetime import datetime, time, timedelta
from operator import itemgetter
from typing import Dict
from dateutil.parser import isoparse
from libretime_api_client.v2 import ApiClient
from libretime_shared.datetime import time_in_milliseconds, time_in_seconds
@ -15,6 +14,8 @@ from .events import (
FileEvent,
WebStreamEvent,
datetime_to_event_key,
event_isoparse,
parse_any_event,
)
@ -54,15 +55,15 @@ def get_schedule(api_client: ApiClient) -> Events:
events: Dict[str, AnyEvent] = {}
for item in sorted(schedule, key=itemgetter("starts_at")):
item["starts_at"] = isoparse(item["starts_at"])
item["ends_at"] = isoparse(item["ends_at"])
item["starts_at"] = event_isoparse(item["starts_at"])
item["ends_at"] = event_isoparse(item["ends_at"])
show_instance = api_client.get_show_instance(item["instance"]).json()
show = api_client.get_show(show_instance["show"]).json()
if show["live_enabled"]:
show_instance["starts_at"] = isoparse(show_instance["starts_at"])
show_instance["ends_at"] = isoparse(show_instance["ends_at"])
show_instance["starts_at"] = event_isoparse(show_instance["starts_at"])
show_instance["ends_at"] = event_isoparse(show_instance["ends_at"])
generate_live_events(
events,
show_instance,
@ -87,26 +88,28 @@ def generate_live_events(
):
transition = timedelta(seconds=input_fade_transition)
switch_off_event_key = datetime_to_event_key(show_instance["ends_at"] - transition)
kick_out_event_key = datetime_to_event_key(show_instance["ends_at"])
switch_off = show_instance["ends_at"] - transition
kick_out = show_instance["ends_at"]
switch_off_event_key = datetime_to_event_key(switch_off)
kick_out_event_key = datetime_to_event_key(kick_out)
# If enabled, fade the input source out
if switch_off_event_key != kick_out_event_key:
switch_off_event: ActionEvent = {
"type": EventKind.ACTION,
"event_type": "switch_off",
"start": switch_off_event_key,
"end": switch_off_event_key,
}
if switch_off != kick_out:
switch_off_event = ActionEvent(
type=EventKind.ACTION,
event_type="switch_off",
start=switch_off,
end=switch_off,
)
insert_event(events, switch_off_event_key, switch_off_event)
# Then kick the source out
kick_out_event: ActionEvent = {
"type": EventKind.ACTION,
"event_type": "kick_out",
"start": kick_out_event_key,
"end": kick_out_event_key,
}
kick_out_event = ActionEvent(
type=EventKind.ACTION,
event_type="kick_out",
start=kick_out,
end=kick_out,
)
insert_event(events, kick_out_event_key, kick_out_event)
@ -119,32 +122,28 @@ def generate_file_events(
"""
Generate events for a scheduled file.
"""
schedule_start_event_key = datetime_to_event_key(schedule["starts_at"])
schedule_end_event_key = datetime_to_event_key(schedule["ends_at"])
event: FileEvent = {
"type": EventKind.FILE,
"row_id": schedule["id"],
"start": schedule_start_event_key,
"end": schedule_end_event_key,
"uri": file["url"],
"id": file["id"],
event = FileEvent(
type=EventKind.FILE,
row_id=schedule["id"],
start=schedule["starts_at"],
end=schedule["ends_at"],
uri=file["url"],
id=file["id"],
# Show data
"show_name": show["name"],
show_name=show["name"],
# Extra data
"fade_in": time_in_milliseconds(time.fromisoformat(schedule["fade_in"])),
"fade_out": time_in_milliseconds(time.fromisoformat(schedule["fade_out"])),
"cue_in": time_in_seconds(time.fromisoformat(schedule["cue_in"])),
"cue_out": time_in_seconds(time.fromisoformat(schedule["cue_out"])),
"metadata": {
"track_title": file["track_title"],
"artist_name": file["artist_name"],
"mime": file["mime"],
},
"replay_gain": file["replay_gain"],
"filesize": file["size"],
}
insert_event(events, schedule_start_event_key, event)
fade_in=time_in_milliseconds(time.fromisoformat(schedule["fade_in"])),
fade_out=time_in_milliseconds(time.fromisoformat(schedule["fade_out"])),
cue_in=time_in_seconds(time.fromisoformat(schedule["cue_in"])),
cue_out=time_in_seconds(time.fromisoformat(schedule["cue_out"])),
# File data
track_title=file.get("track_title"),
artist_name=file.get("artist_name"),
mime=file["mime"],
replay_gain=file["replay_gain"],
filesize=file["size"],
)
insert_event(events, event.start_key, event)
def generate_webstream_events(
@ -159,46 +158,61 @@ def generate_webstream_events(
schedule_start_event_key = datetime_to_event_key(schedule["starts_at"])
schedule_end_event_key = datetime_to_event_key(schedule["ends_at"])
stream_buffer_start_event: WebStreamEvent = {
"type": EventKind.WEB_STREAM_BUFFER_START,
"row_id": schedule["id"],
"start": datetime_to_event_key(schedule["starts_at"] - timedelta(seconds=5)),
"end": datetime_to_event_key(schedule["starts_at"] - timedelta(seconds=5)),
"uri": webstream["url"],
"id": webstream["id"],
}
stream_buffer_start_event = WebStreamEvent(
type=EventKind.WEB_STREAM_BUFFER_START,
row_id=schedule["id"],
start=schedule["starts_at"] - timedelta(seconds=5),
end=schedule["starts_at"] - timedelta(seconds=5),
uri=webstream["url"],
id=webstream["id"],
# Show data
show_name=show["name"],
)
insert_event(events, schedule_start_event_key, stream_buffer_start_event)
stream_output_start_event: WebStreamEvent = {
"type": EventKind.WEB_STREAM_OUTPUT_START,
"row_id": schedule["id"],
"start": schedule_start_event_key,
"end": schedule_end_event_key,
"uri": webstream["url"],
"id": webstream["id"],
stream_output_start_event = WebStreamEvent(
type=EventKind.WEB_STREAM_OUTPUT_START,
row_id=schedule["id"],
start=schedule["starts_at"],
end=schedule["ends_at"],
uri=webstream["url"],
id=webstream["id"],
# Show data
"show_name": show["name"],
}
show_name=show["name"],
)
insert_event(events, schedule_start_event_key, stream_output_start_event)
# NOTE: stream_*_end were previously triggered 1 second before
# the schedule end.
stream_buffer_end_event: WebStreamEvent = {
"type": EventKind.WEB_STREAM_BUFFER_END,
"row_id": schedule["id"],
"start": schedule_end_event_key,
"end": schedule_end_event_key,
"uri": webstream["url"],
"id": webstream["id"],
}
stream_buffer_end_event = WebStreamEvent(
type=EventKind.WEB_STREAM_BUFFER_END,
row_id=schedule["id"],
start=schedule["ends_at"],
end=schedule["ends_at"],
uri=webstream["url"],
id=webstream["id"],
# Show data
show_name=show["name"],
)
insert_event(events, schedule_end_event_key, stream_buffer_end_event)
stream_output_end_event: WebStreamEvent = {
"type": EventKind.WEB_STREAM_OUTPUT_END,
"row_id": schedule["id"],
"start": schedule_end_event_key,
"end": schedule_end_event_key,
"uri": webstream["url"],
"id": webstream["id"],
}
stream_output_end_event = WebStreamEvent(
type=EventKind.WEB_STREAM_OUTPUT_END,
row_id=schedule["id"],
start=schedule["ends_at"],
end=schedule["ends_at"],
uri=webstream["url"],
id=webstream["id"],
# Show data
show_name=show["name"],
)
insert_event(events, schedule_end_event_key, stream_output_end_event)
def receive_schedule(schedule: Dict[str, dict]) -> Events:
events: Dict[str, AnyEvent] = {}
for event_key, event in schedule.items():
events[event_key] = parse_any_event(event)
return events