diff options
Diffstat (limited to 'mediaplayer.py')
| -rw-r--r-- | mediaplayer.py | 115 |
1 files changed, 16 insertions, 99 deletions
diff --git a/mediaplayer.py b/mediaplayer.py index ea8ea2e..dcad2c2 100644 --- a/mediaplayer.py +++ b/mediaplayer.py @@ -1,118 +1,37 @@ import json from json import JSONDecodeError -from random import randint import os import sys import asyncio from random import randint + from websockets import connect, ConnectionClosedError import concurrent from enum import IntEnum -from os import environ -from argparse import ArgumentParser from aglbaseservice import AGLBaseService -URL = f'ws://{os.environ["AGL_TGT_IP"]}:{os.environ.get("AGL_TGT_PORT","30016")}/api?token=HELLO&uuid=magic' - -# x-AFB-ws-json1 message Type -class AFBT(IntEnum): - REQUEST = 2, - RESPONSE = 3, - ERROR = 4, - EVENT = 5 - -msgq = {} +global DEBUG +DEBUG = True -def addrequest(msgid, msg): - msgq[msgid] = {'request': msg, 'response': None} - -def addresponse(msgid, msg): - if msgid in msgq.keys(): - msgq[msgid]['response'] = msg +class MediaPlayerService(AGLBaseService): + def __await__(self): + return super()._async_init().__await__() -class MediaPlayerService(): def __init__(self, ip, port): super().__init__(api='mediaplayer', ip=ip, port=port) - def __init__(self, url=None, api='', ip='127.0.0.1', port='30000', token='HELLO', uuid='magic'): - self.api = api - self.url = url - self.ip = ip - self.port = port - self.token = token - self.uuid = uuid - - # if url is set, disregard other params; if not - construct url -#TODO: finish implementing constructor with params - - def __await__(self): - return self._async_init().__await__() - - async def __aenter__(self): - return self._async_init() - - async def _async_init(self): - self._conn = connect(uri=URL, subprotocols=['x-afb-ws-json1'], close_timeout=0, ping_interval=None) - self.websocket = await self._conn.__aenter__() - self.url = URL - self.api = 'mediaplayer' - return self - - - async def __aexit__(self, *args, **kwargs): - await self._conn.__aexit__(*args, **kwargs) - - - async def close(self): - await self._conn.__aexit__(*sys.exc_info()) - - async def send(self, message): - await self.websocket.send(message) - - async def receive(self): - return await self.websocket.recv() - - async def listener(self): - try: - while True: - msg = await self.receive() - print(f"received {msg}") - try: - data = json.loads(msg) - if isinstance(data,list): - if data[0] == AFBT.RESPONSE and str.isnumeric(data[1]): - msgid = int(data[1]) - if msgid in msgq: - addresponse(msgid, data) - - except JSONDecodeError: - print("not decoding a non-json message") - - except KeyboardInterrupt: - pass - except asyncio.CancelledError: - print("websocket listener coroutine stopped") - async def playlist(self): verb = 'playlist' msgid = randint(0, 999999) - addrequest(msgid, msg) + await self.send(f'[2,"{msgid}","{self.api}/{verb}",""]') - async def subscribe(self, event='metadata'): # event could also be 'playlist' instead of 'metadata' - verb = 'subscribe' - msgid = randint(0, 999999) - msg = f'[2,"{msgid}","{self.api}/{verb}",{{"value": "{event}"}}]' - addrequest(msgid, msg) - await self.send(msg) + async def subscribe(self, event='metadata'): + await super().subscribe(event=event) async def unsubscribe(self, event='metadata'): - verb = 'unsubscribe' - msgid = randint(0, 999999) - msg = f'[2,"{msgid}","{self.api}/{verb}",{{"value": "{event}"}}]' - addrequest(msgid, msg) - await self.send(msg) + await super().subscribe(event=event) async def control(self, name, value=None): verb = 'controls' @@ -150,30 +69,28 @@ class MediaPlayerService(): elif name == 'loop': assert value in loopstate, f'Tried to set invalid loopstate - {value}, use "off", "playlist" or "track"' msg = f'[2,"{msgid}","{self.api}/{verb}", {{"value": "{name}", "{controls[name]}": {str(value)}}}]' - addrequest(msgid, msg) await self.send(msg) async def main(loop): - addr = os.environ.get('AGL_TGT_IP', 'localhost') + addr = os.environ.get('AGL_TGT_IP', '192.168.234.202') port = os.environ.get('AGL_TGT_PORT', '30016') MPS = await MediaPlayerService(ip=addr, port=port) listener = loop.create_task(MPS.listener()) try: - await MPS.subscribe() + await MPS.subscribe('metadata') + await MPS.control('next') await listener - except ConnectionClosedError as e: - print("Connection timed out or closed abnormally. Trying to reconnect...") - result = await MPS._async_init() - print(result) + except KeyboardInterrupt: pass listener.cancel() - await MPS.unsubscribe() + await MPS.unsubscribe('playlist') + if __name__ == '__main__': loop = asyncio.get_event_loop() |