e88e843b65
move EVENT_KEY_FORMAT to events module properly type fetch queue event start/end can be str or datetime
108 lines
3 KiB
Python
108 lines
3 KiB
Python
import logging
|
|
import math
|
|
import time
|
|
from datetime import datetime
|
|
from queue import Queue
|
|
from threading import Thread
|
|
from typing import List, Tuple
|
|
|
|
from ..config import PUSH_INTERVAL, Config
|
|
from .events import AnyEvent, EventKind, Events, event_key_to_datetime
|
|
from .liquidsoap import PypoLiquidsoap
|
|
from .queue import PypoLiqQueue
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def is_stream(media_item):
|
|
return media_item["type"] == "stream_output_start"
|
|
|
|
|
|
def is_file(media_item):
|
|
return media_item["type"] == "file"
|
|
|
|
|
|
class PypoPush(Thread):
|
|
name = "push"
|
|
daemon = True
|
|
|
|
def __init__(
|
|
self,
|
|
push_queue: Queue[Events],
|
|
pypo_liquidsoap: PypoLiquidsoap,
|
|
config: Config,
|
|
):
|
|
Thread.__init__(self)
|
|
self.queue = push_queue
|
|
|
|
self.config = config
|
|
|
|
self.future_scheduled_queue: Queue[Events] = Queue()
|
|
self.pypo_liquidsoap = pypo_liquidsoap
|
|
|
|
self.plq = PypoLiqQueue(self.future_scheduled_queue, self.pypo_liquidsoap)
|
|
self.plq.start()
|
|
|
|
def main(self):
|
|
loops = 0
|
|
heartbeat_period = math.floor(30 / PUSH_INTERVAL)
|
|
|
|
events = None
|
|
|
|
while True:
|
|
try:
|
|
events = self.queue.get(block=True)
|
|
except Exception as exception:
|
|
logger.exception(exception)
|
|
raise exception
|
|
else:
|
|
logger.debug(events)
|
|
# separate media_schedule list into currently_playing and
|
|
# scheduled_for_future lists
|
|
currently_playing, scheduled_for_future = self.separate_present_future(
|
|
events
|
|
)
|
|
|
|
self.pypo_liquidsoap.verify_correct_present_media(currently_playing)
|
|
self.future_scheduled_queue.put(scheduled_for_future)
|
|
|
|
if loops % heartbeat_period == 0:
|
|
logger.info("heartbeat")
|
|
loops = 0
|
|
loops += 1
|
|
|
|
def separate_present_future(self, events: Events) -> Tuple[List[AnyEvent], Events]:
|
|
now = datetime.utcnow()
|
|
|
|
present: List[AnyEvent] = []
|
|
future: Events = {}
|
|
|
|
for key in sorted(events.keys()):
|
|
item = events[key]
|
|
|
|
# Ignore track that already ended
|
|
if (
|
|
item["type"] == EventKind.FILE
|
|
and event_key_to_datetime(item["end"]) < now
|
|
):
|
|
logger.debug("ignoring ended media_item: %s", item)
|
|
continue
|
|
|
|
diff_sec = (now - event_key_to_datetime(item["start"])).total_seconds()
|
|
|
|
if diff_sec >= 0:
|
|
logger.debug("adding media_item to present: %s", item)
|
|
present.append(item)
|
|
else:
|
|
logger.debug("adding media_item to future: %s", item)
|
|
future[key] = item
|
|
|
|
return present, future
|
|
|
|
def run(self):
|
|
while True:
|
|
try:
|
|
self.main()
|
|
except Exception as exception:
|
|
logger.exception(exception)
|
|
time.sleep(5)
|