write metadata tests; start source tests
This commit is contained in:
bkfox
@ -1,9 +1,16 @@
|
||||
# TODO: for the moment, update in station and program names do not update the
|
||||
# related fields.
|
||||
|
||||
from .base import Request
|
||||
from .metadata import Metadata, Request
|
||||
from .streamer import Streamer
|
||||
from .sources import Source, PlaylistSource, QueueSource
|
||||
|
||||
|
||||
__all__ = ("Request", "Streamer", "Source", "PlaylistSource", "QueueSource")
|
||||
__all__ = (
|
||||
"Metadata",
|
||||
"Request",
|
||||
"Streamer",
|
||||
"Source",
|
||||
"PlaylistSource",
|
||||
"QueueSource",
|
||||
)
|
||||
|
||||
@ -6,7 +6,7 @@ from django.utils.translation import gettext_lazy as _
|
||||
|
||||
|
||||
__all__ = (
|
||||
"BaseMetadata",
|
||||
"Metadata",
|
||||
"Request",
|
||||
)
|
||||
|
||||
@ -19,7 +19,7 @@ logger = logging.getLogger("aircox")
|
||||
# correctly.
|
||||
|
||||
|
||||
class BaseMetadata:
|
||||
class Metadata:
|
||||
"""Base class for handling request metadata."""
|
||||
|
||||
controller = None
|
||||
@ -43,29 +43,34 @@ class BaseMetadata:
|
||||
|
||||
@property
|
||||
def is_playing(self):
|
||||
"""True if the source is playing."""
|
||||
# FIXME: validate on controller's current source?
|
||||
return self.status == "playing"
|
||||
|
||||
@property
|
||||
def status_verbose(self):
|
||||
return self.validate_status(self.status, True)
|
||||
"""Verbose version of self's status (translated string)."""
|
||||
status = self.validate_status(self.status)
|
||||
return _(status) if status else ""
|
||||
|
||||
def fetch(self):
|
||||
data = self.controller.send("request.metadata ", self.rid, parse=True)
|
||||
if data:
|
||||
self.validate(data)
|
||||
|
||||
def validate_status(self, status, i18n=False):
|
||||
def validate_status(self, status):
|
||||
"""Return correct status for this metadata based on provided one and
|
||||
controller.
|
||||
|
||||
:returns: status string
|
||||
"""
|
||||
on_air = self.controller.source
|
||||
if (
|
||||
on_air
|
||||
and status == "playing"
|
||||
and (on_air == self or on_air.rid == self.rid)
|
||||
):
|
||||
return _("playing") if i18n else "playing"
|
||||
elif status == "playing":
|
||||
return _("paused") if i18n else "paused"
|
||||
if "playing" and on_air and (on_air == self or on_air.rid == self.rid):
|
||||
return "playing"
|
||||
elif status in ("paused", "playing"):
|
||||
return "paused"
|
||||
else:
|
||||
return _("stopped") if i18n else "stopped"
|
||||
return "stopped"
|
||||
|
||||
def validate_air_time(self, air_time):
|
||||
if air_time:
|
||||
@ -85,6 +90,6 @@ class BaseMetadata:
|
||||
self.request_status = data.get("status")
|
||||
|
||||
|
||||
class Request(BaseMetadata):
|
||||
class Request(Metadata):
|
||||
title = None
|
||||
artist = None
|
||||
@ -1,20 +1,9 @@
|
||||
import atexit
|
||||
import logging
|
||||
import os
|
||||
import re
|
||||
import signal
|
||||
import subprocess
|
||||
|
||||
import psutil
|
||||
import tzlocal
|
||||
from django.template.loader import render_to_string
|
||||
from django.utils import timezone as tz
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
|
||||
from aircox.conf import settings
|
||||
from aircox.utils import to_seconds
|
||||
|
||||
from .connector import Connector
|
||||
from .metadata import Metadata, Request
|
||||
|
||||
|
||||
__all__ = (
|
||||
@ -23,254 +12,10 @@ __all__ = (
|
||||
"QueueSource",
|
||||
)
|
||||
|
||||
|
||||
local_tz = tzlocal.get_localzone()
|
||||
logger = logging.getLogger("aircox")
|
||||
|
||||
|
||||
class BaseMetadata:
|
||||
"""Base class for handling request metadata."""
|
||||
|
||||
controller = None
|
||||
"""Controller."""
|
||||
rid = None
|
||||
"""Request id."""
|
||||
uri = None
|
||||
"""Request uri."""
|
||||
status = None
|
||||
"""Current playing status."""
|
||||
request_status = None
|
||||
"""Requests' status."""
|
||||
air_time = None
|
||||
"""Launch datetime."""
|
||||
|
||||
def __init__(self, controller=None, rid=None, data=None):
|
||||
self.controller = controller
|
||||
self.rid = rid
|
||||
if data is not None:
|
||||
self.validate(data)
|
||||
|
||||
@property
|
||||
def is_playing(self):
|
||||
return self.status == "playing"
|
||||
|
||||
@property
|
||||
def status_verbose(self):
|
||||
return self.validate_status(self.status, True)
|
||||
|
||||
def fetch(self):
|
||||
data = self.controller.send("request.metadata ", self.rid, parse=True)
|
||||
if data:
|
||||
self.validate(data)
|
||||
|
||||
def validate_status(self, status, i18n=False):
|
||||
on_air = self.controller.source
|
||||
if (
|
||||
on_air
|
||||
and status == "playing"
|
||||
and (on_air == self or on_air.rid == self.rid)
|
||||
):
|
||||
return _("playing") if i18n else "playing"
|
||||
elif status == "playing":
|
||||
return _("paused") if i18n else "paused"
|
||||
else:
|
||||
return _("stopped") if i18n else "stopped"
|
||||
|
||||
def validate_air_time(self, air_time):
|
||||
if air_time:
|
||||
air_time = tz.datetime.strptime(air_time, "%Y/%m/%d %H:%M:%S")
|
||||
return local_tz.localize(air_time)
|
||||
|
||||
def validate(self, data):
|
||||
"""Validate provided data and set as attribute (must already be
|
||||
declared)"""
|
||||
for key, value in data.items():
|
||||
if hasattr(self, key) and not callable(getattr(self, key)):
|
||||
setattr(self, key, value)
|
||||
self.uri = data.get("initial_uri")
|
||||
|
||||
self.air_time = self.validate_air_time(data.get("on_air"))
|
||||
self.status = self.validate_status(data.get("status"))
|
||||
self.request_status = data.get("status")
|
||||
|
||||
|
||||
class Request(BaseMetadata):
|
||||
title = None
|
||||
artist = None
|
||||
|
||||
|
||||
class Streamer:
|
||||
connector = None
|
||||
process = None
|
||||
|
||||
station = None
|
||||
template_name = "aircox_streamer/scripts/station.liq"
|
||||
path = None
|
||||
"""Config path."""
|
||||
sources = None
|
||||
"""List of all monitored sources."""
|
||||
source = None
|
||||
"""Current source being played on air."""
|
||||
# note: we disable on_air rids since we don't have use of it for the
|
||||
# moment
|
||||
# on_air = None
|
||||
# """ On-air request ids (rid) """
|
||||
inputs = None
|
||||
"""Queryset to input ports."""
|
||||
outputs = None
|
||||
"""Queryset to output ports."""
|
||||
|
||||
def __init__(self, station, connector=None):
|
||||
self.station = station
|
||||
self.inputs = self.station.port_set.active().input()
|
||||
self.outputs = self.station.port_set.active().output()
|
||||
|
||||
self.id = self.station.slug.replace("-", "_")
|
||||
self.path = os.path.join(station.path, "station.liq")
|
||||
self.connector = Connector(os.path.join(station.path, "station.sock"))
|
||||
self.init_sources()
|
||||
|
||||
@property
|
||||
def socket_path(self):
|
||||
"""Path to Unix socket file."""
|
||||
return self.connector.address
|
||||
|
||||
@property
|
||||
def is_ready(self):
|
||||
"""If external program is ready to use, returns True."""
|
||||
return self.send("list") != ""
|
||||
|
||||
@property
|
||||
def is_running(self):
|
||||
"""True if holds a running process."""
|
||||
if self.process is None:
|
||||
return False
|
||||
|
||||
returncode = self.process.poll()
|
||||
if returncode is None:
|
||||
return True
|
||||
|
||||
self.process = None
|
||||
logger.debug("process died with return code %s" % returncode)
|
||||
return False
|
||||
|
||||
@property
|
||||
def playlists(self):
|
||||
return (s for s in self.sources if isinstance(s, PlaylistSource))
|
||||
|
||||
@property
|
||||
def queues(self):
|
||||
return (s for s in self.sources if isinstance(s, QueueSource))
|
||||
|
||||
# Sources and config ###############################################
|
||||
def send(self, *args, **kwargs):
|
||||
return self.connector.send(*args, **kwargs) or ""
|
||||
|
||||
def init_sources(self):
|
||||
streams = self.station.program_set.filter(stream__isnull=False)
|
||||
self.dealer = QueueSource(self, "dealer")
|
||||
self.sources = [self.dealer] + [
|
||||
PlaylistSource(self, program=program) for program in streams
|
||||
]
|
||||
|
||||
def make_config(self):
|
||||
"""Make configuration files and directory (and sync sources)"""
|
||||
data = render_to_string(
|
||||
self.template_name,
|
||||
{
|
||||
"station": self.station,
|
||||
"streamer": self,
|
||||
"settings": settings,
|
||||
},
|
||||
)
|
||||
data = re.sub("[\t ]+\n", "\n", data)
|
||||
data = re.sub("\n{3,}", "\n\n", data)
|
||||
|
||||
os.makedirs(os.path.dirname(self.path), exist_ok=True)
|
||||
with open(self.path, "w+") as file:
|
||||
file.write(data)
|
||||
|
||||
self.sync()
|
||||
|
||||
def sync(self):
|
||||
"""Sync all sources."""
|
||||
for source in self.sources:
|
||||
source.sync()
|
||||
|
||||
def fetch(self):
|
||||
"""Fetch data from liquidsoap."""
|
||||
for source in self.sources:
|
||||
source.fetch()
|
||||
|
||||
# request.on_air is not ordered: we need to do it manually
|
||||
self.source = next(
|
||||
iter(
|
||||
sorted(
|
||||
(
|
||||
source
|
||||
for source in self.sources
|
||||
if source.request_status == "playing"
|
||||
and source.air_time
|
||||
),
|
||||
key=lambda o: o.air_time,
|
||||
reverse=True,
|
||||
)
|
||||
),
|
||||
None,
|
||||
)
|
||||
|
||||
# Process ##########################################################
|
||||
def get_process_args(self):
|
||||
return ["liquidsoap", "-v", self.path]
|
||||
|
||||
def check_zombie_process(self):
|
||||
if not os.path.exists(self.socket_path):
|
||||
return
|
||||
|
||||
conns = [
|
||||
conn
|
||||
for conn in psutil.net_connections(kind="unix")
|
||||
if conn.laddr == self.socket_path
|
||||
]
|
||||
for conn in conns:
|
||||
if conn.pid is not None:
|
||||
os.kill(conn.pid, signal.SIGKILL)
|
||||
|
||||
def run_process(self):
|
||||
"""Execute the external application with corresponding informations.
|
||||
|
||||
This function must make sure that all needed files have been
|
||||
generated.
|
||||
"""
|
||||
if self.process:
|
||||
return
|
||||
|
||||
args = self.get_process_args()
|
||||
if not args:
|
||||
return
|
||||
|
||||
self.check_zombie_process()
|
||||
self.process = subprocess.Popen(args, stderr=subprocess.STDOUT)
|
||||
atexit.register(lambda: self.kill_process())
|
||||
|
||||
def kill_process(self):
|
||||
if self.process:
|
||||
logger.debug(
|
||||
"kill process %s: %s",
|
||||
self.process.pid,
|
||||
" ".join(self.get_process_args()),
|
||||
)
|
||||
self.process.kill()
|
||||
self.process = None
|
||||
|
||||
def wait_process(self):
|
||||
"""Wait for the process to terminate if there is a process."""
|
||||
if self.process:
|
||||
self.process.wait()
|
||||
self.process = None
|
||||
|
||||
|
||||
class Source(BaseMetadata):
|
||||
class Source(Metadata):
|
||||
controller = None
|
||||
"""Parent controller."""
|
||||
id = None
|
||||
|
||||
@ -6,7 +6,6 @@ import signal
|
||||
import subprocess
|
||||
|
||||
import psutil
|
||||
import tzlocal
|
||||
from django.template.loader import render_to_string
|
||||
|
||||
from aircox.conf import settings
|
||||
@ -17,7 +16,6 @@ from .sources import PlaylistSource, QueueSource
|
||||
|
||||
__all__ = ("Streamer",)
|
||||
|
||||
local_tz = tzlocal.get_localzone()
|
||||
logger = logging.getLogger("aircox")
|
||||
|
||||
|
||||
@ -49,7 +47,9 @@ class Streamer:
|
||||
|
||||
self.id = self.station.slug.replace("-", "_")
|
||||
self.path = os.path.join(station.path, "station.liq")
|
||||
self.connector = Connector(os.path.join(station.path, "station.sock"))
|
||||
self.connector = connector or Connector(
|
||||
os.path.join(station.path, "station.sock")
|
||||
)
|
||||
self.init_sources()
|
||||
|
||||
@property
|
||||
|
||||
Reference in New Issue
Block a user