164 lines
5 KiB
Python
Executable file
164 lines
5 KiB
Python
Executable file
#!/usr/bin/python3
|
|
import argparse
|
|
import datetime
|
|
import os
|
|
import random
|
|
import socket
|
|
import string
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
from glob import glob
|
|
from queue import Queue
|
|
from socketserver import UDPServer, ThreadingMixIn, BaseRequestHandler
|
|
from threading import Thread
|
|
|
|
from .utils import mpv_log, update_playlist, trigger_lights
|
|
|
|
import logging
|
|
logger = logging.getLogger('cdosea')
|
|
|
|
DEFAULT_PORT = 2680
|
|
DEBUG = False
|
|
|
|
|
|
class ThreadingUDPServer(ThreadingMixIn, UDPServer):
|
|
pass
|
|
|
|
def q_binding(self, *args):
|
|
self.player.running = False
|
|
self.shutdown()
|
|
|
|
class Handler(BaseRequestHandler):
|
|
|
|
def handle(self):
|
|
data = self.request[0].strip()
|
|
logger.debug("%s said: %s", self.client_address[0], data)
|
|
queue = False
|
|
if data == b'POS':
|
|
response = ('%d' % self.server.player.position).encode()
|
|
else:
|
|
if self.server.player.playing:
|
|
response = b'PLAYING'
|
|
else:
|
|
response = b'WAITING'
|
|
queue = True
|
|
socket = self.request[1]
|
|
socket.sendto(response, self.client_address)
|
|
if queue:
|
|
self.server.player.queue.put('')
|
|
|
|
class Player(Thread):
|
|
first = True
|
|
running = True
|
|
playing = False
|
|
current = ''
|
|
playlist_items = 260
|
|
|
|
def __init__(self, peer, player, playlist, prefix, lights=False):
|
|
self.peer = peer
|
|
self.playlist = playlist
|
|
self.prefix = prefix
|
|
self.player = player
|
|
self.lights = lights
|
|
self.queue = Queue()
|
|
Thread.__init__(self)
|
|
self.daemon = True
|
|
self.start()
|
|
|
|
def run(self):
|
|
logger.debug("update playlist")
|
|
self.position, self.playlist_items = update_playlist(self.playlist, self.prefix, position=self.get_position())
|
|
logger.debug("load playlist")
|
|
self.player.loadlist(self.playlist)
|
|
self.player.pause = True
|
|
while self.running:
|
|
logger.debug("play")
|
|
self.play()
|
|
self.position = (self.position + 1) % self.playlist_items
|
|
logger.debug("ping")
|
|
self.ping()
|
|
logger.debug("wait")
|
|
self.queue.get()
|
|
|
|
def play(self):
|
|
self.playing = True
|
|
self.player.pause = False
|
|
if self.lights:
|
|
trigger_lights(self.player.path.decode())
|
|
self.player.wait_for_playback()
|
|
self.player.pause = True
|
|
self.playing = False
|
|
|
|
def ping(self):
|
|
data = 'NEXT'
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
logger.debug("ping: %s %s", self.peer, data)
|
|
sock.sendto(bytes(data + "\n", "utf-8"), self.peer)
|
|
received = str(sock.recv(1024), "utf-8")
|
|
if received == 'WAITING' and not self.queue.qsize():
|
|
if self.first:
|
|
self.queue.put('')
|
|
self.first = False
|
|
logger.debug("Sent: %s", data)
|
|
logger.debug("Received: %s", received)
|
|
|
|
def get_position(self):
|
|
try:
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
sock.settimeout(1)
|
|
data = 'POS'
|
|
sock.sendto(bytes(data + "\n", "utf-8"), self.peer)
|
|
received = str(sock.recv(1024), "utf-8")
|
|
pos = int(received)
|
|
except socket.timeout:
|
|
logger.debug('no response', exc_info=1)
|
|
pos = None
|
|
return pos
|
|
|
|
|
|
def main():
|
|
playlist = os.path.expanduser('~/Videos/cdosea.m3u')
|
|
prefix = os.path.expanduser('~/Videos/CDOSEA')
|
|
|
|
parser = argparse.ArgumentParser(description='play 2 screens in sync')
|
|
parser.add_argument('--peer', help='ip[:port] of peer', required=True)
|
|
parser.add_argument('--port', type=int, help='local port', default=DEFAULT_PORT)
|
|
parser.add_argument('--playlist', help='play.m3u', default=playlist)
|
|
parser.add_argument('--prefix', help='video location', default=prefix)
|
|
parser.add_argument('--window', action='store_true', help='run in window', default=False)
|
|
parser.add_argument('--debug', action='store_true', help='debug', default=False)
|
|
parser.add_argument('--lights', action='store_true', help='lights', default=False)
|
|
args = parser.parse_args()
|
|
|
|
if ':' in args.peer:
|
|
peer = args.peer.split(':')
|
|
peer = (peer[0], int(peer[1]))
|
|
else:
|
|
peer = (args.peer, DEFAULT_PORT)
|
|
|
|
DEBUG = args.debug
|
|
if DEBUG:
|
|
log_format = '%(asctime)s:%(levelname)s:%(name)s:%(message)s'
|
|
logging.basicConfig(level=logging.DEBUG, format=log_format)
|
|
base = os.path.dirname(os.path.abspath(__file__))
|
|
os.chdir(base)
|
|
if args.lights:
|
|
time.sleep(5)
|
|
|
|
player = get_player(fullscreen=not args.window)
|
|
|
|
server = ThreadingUDPServer(('0.0.0.0', args.port), Handler)
|
|
server.player = Player(peer, player, args.playlist, args.prefix, args.lights)
|
|
player.register_key_binding('q', server.q_binding)
|
|
|
|
logger.debug("listen on %s...", args.port)
|
|
try:
|
|
server.serve_forever()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
del player
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|