chore: move api client get_schedule to playout
This commit is contained in:
parent
3ce60d4881
commit
1faf5a724b
7 changed files with 407 additions and 388 deletions
|
@ -7,25 +7,11 @@
|
|||
# schedule a playlist one minute from the current time.
|
||||
###############################################################################
|
||||
import logging
|
||||
from datetime import datetime, timedelta
|
||||
from typing import Dict
|
||||
|
||||
from dateutil.parser import isoparse
|
||||
from libretime_shared.datetime import (
|
||||
fromisoformat,
|
||||
time_in_milliseconds,
|
||||
time_in_seconds,
|
||||
)
|
||||
|
||||
from ._config import Config
|
||||
from .utils import RequestProvider
|
||||
|
||||
LIBRETIME_API_VERSION = "2.0"
|
||||
EVENT_KEY_FORMAT = "%Y-%m-%d-%H-%M-%S"
|
||||
|
||||
|
||||
def datetime_to_event_key(value: datetime) -> str:
|
||||
return value.strftime(EVENT_KEY_FORMAT)
|
||||
|
||||
|
||||
api_endpoints = {}
|
||||
|
@ -55,135 +41,7 @@ class AirtimeApiClient:
|
|||
endpoints=api_endpoints,
|
||||
)
|
||||
|
||||
def get_schedule(self):
|
||||
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 = self.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 = self.services.show_instance_url(id=item["instance_id"])
|
||||
show = self.services.show_url(id=show_instance["show_id"])
|
||||
|
||||
if item["file"]:
|
||||
file = self.services.file_url(id=item["file_id"])
|
||||
events.update(generate_file_events(item, file, show))
|
||||
|
||||
elif item["stream"]:
|
||||
webstream = self.services.webstream_url(id=item["stream_id"])
|
||||
events.update(generate_webstream_events(item, webstream, show))
|
||||
|
||||
return {"media": events}
|
||||
|
||||
def update_file(self, file_id, payload):
|
||||
data = self.services.file_url(id=file_id)
|
||||
data.update(payload)
|
||||
return self.services.file_url(id=file_id, _put_data=data)
|
||||
|
||||
|
||||
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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue