152 lines
4.6 KiB
Python
152 lines
4.6 KiB
Python
from datetime import datetime, timedelta
|
|
from typing import Dict
|
|
|
|
from dateutil.parser import isoparse
|
|
from libretime_api_client.version2 import AirtimeApiClient as ApiClient
|
|
from libretime_shared.datetime import (
|
|
time_fromisoformat,
|
|
time_in_milliseconds,
|
|
time_in_seconds,
|
|
)
|
|
|
|
from .events import EventKind
|
|
|
|
EVENT_KEY_FORMAT = "%Y-%m-%d-%H-%M-%S"
|
|
|
|
|
|
def datetime_to_event_key(value: datetime) -> str:
|
|
return value.strftime(EVENT_KEY_FORMAT)
|
|
|
|
|
|
def get_schedule(api_client: ApiClient):
|
|
current_time = datetime.utcnow()
|
|
end_time = current_time + timedelta(days=1)
|
|
|
|
current_time_str = current_time.isoformat(timespec="seconds")
|
|
end_time_str = end_time.isoformat(timespec="seconds")
|
|
|
|
schedule = api_client.services.schedule_url(
|
|
params={
|
|
"ends__range": (f"{current_time_str}Z,{end_time_str}Z"),
|
|
"is_valid": True,
|
|
"playout_status__gt": 0,
|
|
}
|
|
)
|
|
|
|
events = {}
|
|
for item in schedule:
|
|
item["starts"] = isoparse(item["starts"])
|
|
item["ends"] = isoparse(item["ends"])
|
|
|
|
show_instance = api_client.services.show_instance_url(id=item["instance_id"])
|
|
show = api_client.services.show_url(id=show_instance["show_id"])
|
|
|
|
if item["file"]:
|
|
file = api_client.services.file_url(id=item["file_id"])
|
|
events.update(generate_file_events(item, file, show))
|
|
|
|
elif item["stream"]:
|
|
webstream = api_client.services.webstream_url(id=item["stream_id"])
|
|
events.update(generate_webstream_events(item, webstream, show))
|
|
|
|
return {"media": events}
|
|
|
|
|
|
def generate_file_events(
|
|
schedule: dict,
|
|
file: dict,
|
|
show: dict,
|
|
) -> Dict[str, dict]:
|
|
"""
|
|
Generate events for a scheduled file.
|
|
"""
|
|
events = {}
|
|
|
|
schedule_start_event_key = datetime_to_event_key(schedule["starts"])
|
|
schedule_end_event_key = datetime_to_event_key(schedule["ends"])
|
|
|
|
events[schedule_start_event_key] = {
|
|
"type": EventKind.FILE,
|
|
"independent_event": False,
|
|
"row_id": schedule["id"],
|
|
"start": schedule_start_event_key,
|
|
"end": schedule_end_event_key,
|
|
"uri": file["url"],
|
|
"id": file["id"],
|
|
# Show data
|
|
"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"],
|
|
}
|
|
|
|
return events
|
|
|
|
|
|
def generate_webstream_events(
|
|
schedule: dict,
|
|
webstream: dict,
|
|
show: dict,
|
|
) -> Dict[str, dict]:
|
|
"""
|
|
Generate events for a scheduled webstream.
|
|
"""
|
|
events = {}
|
|
|
|
schedule_start_event_key = datetime_to_event_key(schedule["starts"])
|
|
schedule_end_event_key = datetime_to_event_key(schedule["ends"])
|
|
|
|
events[schedule_start_event_key] = {
|
|
"type": EventKind.STREAM_BUFFER_START,
|
|
"independent_event": True,
|
|
"row_id": schedule["id"],
|
|
"start": datetime_to_event_key(schedule["starts"] - timedelta(seconds=5)),
|
|
"end": datetime_to_event_key(schedule["starts"] - timedelta(seconds=5)),
|
|
"uri": webstream["url"],
|
|
"id": webstream["id"],
|
|
}
|
|
|
|
events[f"{schedule_start_event_key}_0"] = {
|
|
"type": EventKind.STREAM_OUTPUT_START,
|
|
"independent_event": True,
|
|
"row_id": schedule["id"],
|
|
"start": schedule_start_event_key,
|
|
"end": schedule_end_event_key,
|
|
"uri": webstream["url"],
|
|
"id": webstream["id"],
|
|
# Show data
|
|
"show_name": show["name"],
|
|
}
|
|
|
|
# NOTE: stream_*_end were previously triggerered 1 second before the schedule end.
|
|
events[schedule_end_event_key] = {
|
|
"type": EventKind.STREAM_BUFFER_END,
|
|
"independent_event": True,
|
|
"row_id": schedule["id"],
|
|
"start": schedule_end_event_key,
|
|
"end": schedule_end_event_key,
|
|
"uri": webstream["url"],
|
|
"id": webstream["id"],
|
|
}
|
|
|
|
events[f"{schedule_end_event_key}_0"] = {
|
|
"type": EventKind.STREAM_OUTPUT_END,
|
|
"independent_event": True,
|
|
"row_id": schedule["id"],
|
|
"start": schedule_end_event_key,
|
|
"end": schedule_end_event_key,
|
|
"uri": webstream["url"],
|
|
"id": webstream["id"],
|
|
}
|
|
|
|
return events
|