You've already forked RadioPlayer
mirror of
https://github.com/radio95-rnt/RadioPlayer.git
synced 2026-02-26 21:53:54 +01:00
minor internal changes
This commit is contained in:
@@ -118,23 +118,16 @@ def websocket_server_process(shared_data: dict, imc_q: multiprocessing.Queue, ws
|
||||
|
||||
# start server
|
||||
server = await websockets.serve(handler_wrapper, "0.0.0.0", 3001)
|
||||
# background task: broadcast worker
|
||||
broadcaster = asyncio.create_task(broadcast_worker(ws_q, clients))
|
||||
# run forever until server closes
|
||||
await server.wait_closed()
|
||||
# ensure broadcaster stops
|
||||
ws_q.put(None)
|
||||
await broadcaster
|
||||
|
||||
# On SIGINT/SIGTERM, stop gracefully
|
||||
loop = asyncio.new_event_loop()
|
||||
asyncio.set_event_loop(loop)
|
||||
try:
|
||||
loop.run_until_complete(runner())
|
||||
except (KeyboardInterrupt, SystemExit):
|
||||
pass
|
||||
finally:
|
||||
loop.close()
|
||||
try: loop.run_until_complete(runner())
|
||||
except (KeyboardInterrupt, SystemExit): pass
|
||||
finally: loop.close()
|
||||
|
||||
# ---------- Module class (drop-in replacement) ----------
|
||||
|
||||
@@ -143,29 +136,21 @@ class Module(PlayerModule):
|
||||
self.manager = multiprocessing.Manager()
|
||||
self.data = self.manager.dict()
|
||||
self.imc_q = self.manager.Queue()
|
||||
# queue for sending broadcasts to the websocket process
|
||||
self.ws_q = self.manager.Queue()
|
||||
|
||||
# initial state
|
||||
self.data["playlist"] = "[]"
|
||||
self.data["track"] = "{}"
|
||||
self.data["progress"] = "{}"
|
||||
|
||||
# ipc thread: listens for responses from other modules (same as before)
|
||||
self.ipc_thread_running = True
|
||||
self.ipc_thread = threading.Thread(target=self._ipc_worker, daemon=True)
|
||||
self.ipc_thread.start()
|
||||
|
||||
# start websocket server process
|
||||
self.ws_process = multiprocessing.Process(
|
||||
target=websocket_server_process, args=(self.data, self.imc_q, self.ws_q), daemon=False
|
||||
)
|
||||
self.ws_process = multiprocessing.Process(target=websocket_server_process, args=(self.data, self.imc_q, self.ws_q), daemon=False)
|
||||
self.ws_process.start()
|
||||
if os.name == "posix":
|
||||
try:
|
||||
os.setpgid(self.ws_process.pid, self.ws_process.pid)
|
||||
except Exception:
|
||||
pass
|
||||
try: os.setpgid(self.ws_process.pid, self.ws_process.pid)
|
||||
except Exception: pass
|
||||
|
||||
def _ipc_worker(self):
|
||||
"""
|
||||
@@ -176,17 +161,10 @@ class Module(PlayerModule):
|
||||
try:
|
||||
message: dict | None = self.imc_q.get()
|
||||
if message is None: break
|
||||
# send to upper layer (existing player IPC)
|
||||
out = self._imc.send(self, message["name"], message["data"])
|
||||
# if message had a key, store the response for the requester
|
||||
if key := message.get("key", None):
|
||||
# store response into shared dict (accessible to ws process)
|
||||
self.data[key] = out
|
||||
except Exception:
|
||||
# swallow errors to avoid killing the ipc thread
|
||||
pass
|
||||
if key := message.get("key", None): self.data[key] = out
|
||||
except Exception: pass
|
||||
|
||||
# The following functions update the shared_data and also push a broadcast message onto ws_q
|
||||
def on_new_playlist(self, playlist: list[Track]) -> None:
|
||||
api_data = []
|
||||
for track in playlist:
|
||||
@@ -199,7 +177,6 @@ class Module(PlayerModule):
|
||||
"offset": track.offset
|
||||
})
|
||||
self.data["playlist"] = json.dumps(api_data)
|
||||
# broadcast
|
||||
try: self.ws_q.put({"event": "playlist", "data": api_data})
|
||||
except Exception: pass
|
||||
|
||||
@@ -216,18 +193,15 @@ class Module(PlayerModule):
|
||||
track_data = {"path": str(track.path), "fade_out": track.fade_out, "fade_in": track.fade_in, "official": track.official, "args": track.args, "offset": track.offset}
|
||||
payload = {"index": index, "track": track_data, "elapsed": elapsed, "total": total, "real_total": real_total}
|
||||
self.data["progress"] = json.dumps(payload)
|
||||
# For frequent progress updates you might want to rate-limit; this pushes every call
|
||||
try: self.ws_q.put({"event": "progress", "data": payload})
|
||||
except Exception: pass
|
||||
|
||||
def shutdown(self):
|
||||
# stop ipc thread
|
||||
self.ipc_thread_running = False
|
||||
try: self.imc_q.put(None)
|
||||
except Exception: pass
|
||||
self.ipc_thread.join(timeout=2)
|
||||
|
||||
# shutdown websocket process by putting sentinel into ws_q and then terminating if needed
|
||||
try: self.ws_q.put(None)
|
||||
except Exception: pass
|
||||
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
#!/usr/bin/env python3
|
||||
import os, subprocess, importlib.util, importlib.machinery, types
|
||||
import sys, signal, glob, time, traceback, atexit
|
||||
import sys, signal, glob, time, traceback
|
||||
import libcache
|
||||
from modules import *
|
||||
from threading import Lock
|
||||
@@ -117,8 +117,7 @@ class RadioPlayer:
|
||||
self.playlist_advisor: PlaylistAdvisor | None = None
|
||||
self.active_modifier: ActiveModifier | None = None
|
||||
self.exit_pending = False
|
||||
self.exit_status_code = 0
|
||||
self.intr_time = 0
|
||||
self.exit_status_code = self.intr_time = 0
|
||||
self.exit_lock = Lock()
|
||||
self.procman = ProcessManager()
|
||||
self.modules: list[tuple[importlib.machinery.ModuleSpec, types.ModuleType, str]] = []
|
||||
@@ -132,19 +131,16 @@ class RadioPlayer:
|
||||
for module in self.simple_modules:
|
||||
if module:
|
||||
try: module.shutdown()
|
||||
except Exception:
|
||||
traceback.print_last(file=self.logger.output)
|
||||
self.logger.error("Exception while shutting down module.")
|
||||
except Exception: traceback.print_exc(file=self.logger.output)
|
||||
self.logger.output.close()
|
||||
|
||||
def handle_sigint(self, signum: int, frame: types.FrameType | None):
|
||||
with self.exit_lock:
|
||||
self.logger.info("Received CTRL+C (SIGINT)")
|
||||
if (time.monotonic() - self.intr_time) > 5:
|
||||
self.intr_time = time.monotonic()
|
||||
if (now := time.monotonic()) and ((now - self.intr_time) > 5):
|
||||
self.intr_time = now
|
||||
self.logger.info("Will quit on song end.")
|
||||
self.exit_pending = True
|
||||
self.exit_status_code = 130
|
||||
self.exit_pending, self.exit_status_code = True, 130
|
||||
else:
|
||||
self.logger.warning("Force-Quit pending")
|
||||
raise SystemExit(130)
|
||||
@@ -178,8 +174,7 @@ class RadioPlayer:
|
||||
try:
|
||||
start = time.perf_counter()
|
||||
spec.loader.exec_module(module)
|
||||
time_took = time.perf_counter() - start
|
||||
if time_took > 0.15: self.logger.warning(f"{module_name} took {time_took:.1f}s to start")
|
||||
if (time_took := time.perf_counter() - start) > 0.15: self.logger.warning(f"{module_name} took {time_took:.1f}s to start")
|
||||
except Exception as e:
|
||||
traceback.print_exc(file=self.logger.output)
|
||||
self.logger.error(f"Failed loading {module_name} due to {e}, continuing")
|
||||
@@ -235,7 +230,7 @@ class RadioPlayer:
|
||||
playlist = None
|
||||
global_args = {}
|
||||
return_pending = track = False
|
||||
cross_fade = int(global_args.get("crossfade", 5))
|
||||
cross_fade = int(global_args.get("crossfade", 5)) # TODO: get rid of global_args usage in the core and instead just store fades in track (that would require a pretty much rewrite of the active modifier tho...)
|
||||
song_i = i = 0
|
||||
|
||||
def get_track():
|
||||
@@ -287,9 +282,7 @@ class RadioPlayer:
|
||||
while end_time >= time.monotonic() and pr.process.poll() is None:
|
||||
start = time.monotonic()
|
||||
[module.progress(song_i, track, time.monotonic() - pr.started_at, pr.duration, end_time - pr.started_at) for module in self.simple_modules if module]
|
||||
elapsed = time.monotonic() - start
|
||||
remaining_until_end = end_time - time.monotonic()
|
||||
if elapsed < 1 and remaining_until_end > 0: time.sleep(min(1 - elapsed, remaining_until_end))
|
||||
if (elapsed := time.monotonic() - start) < 1 and (remaining_until_end := end_time - time.monotonic()) > 0: time.sleep(min(1 - elapsed, remaining_until_end))
|
||||
|
||||
i += 1
|
||||
if not extend: song_i += 1
|
||||
|
||||
Reference in New Issue
Block a user