chore: move api client get_schedule to playout

playout/libretime_playout/pypofetch.py

@@ -18,6 +18,7 @@ from loguru import logger
 
 from . import pure
 from .config import CACHE_DIR, POLL_INTERVAL, Config
+from .schedule import get_schedule
 from .timeout import ls_timeout
 
 
@@ -458,7 +459,7 @@ class PypoFetch(Thread):
 
     def manual_schedule_fetch(self):
         try:
-            self.schedule_data = self.api_client.get_schedule()
+            self.schedule_data = get_schedule(self.api_client)
             logger.debug(f"Received event from API client: {self.schedule_data}")
             self.process_schedule(self.schedule_data)
             return True

playout/libretime_playout/schedule.py

@@ -0,0 +1,145 @@
+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 (
+    fromisoformat,
+    time_in_milliseconds,
+    time_in_seconds,
+)
+
+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": "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(fromisoformat(schedule["fade_in"])),
+        "fade_out": time_in_milliseconds(fromisoformat(schedule["fade_out"])),
+        "cue_in": time_in_seconds(fromisoformat(schedule["cue_in"])),
+        "cue_out": time_in_seconds(fromisoformat(schedule["cue_out"])),
+        "metadata": file,
+        "replay_gain": file["replay_gain"],
+        "filesize": file["filesize"],
+    }
+
+    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": "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": "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": "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": "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