RadioStats/parser/ara.py

import json

from websocket import create_connection, WebSocket

from parser import BaseFetcher
from utils import *



class ArabellaFetcher(BaseFetcher):
    URL = "wss://www.arabella.at/api/_socket/"
    def get(self, channel):
        ws: WebSocket = create_connection(
            self.URL,
            suppress_origin=True,
            header=["User-Agent: Mozilla/5.0 (X11; Linux x86_64; rv:98.0) Gecko/20100101 Firefox/98.0"]
        )
        init = ws.recv()

        ws.send(json.dumps({"type": "select_channels", "channelIds": [1]}))

        result = ws.recv()
        ws.close()
        data = json.loads(result)
        tracks = [data["currentTrack"]]
        if "previousTracks" in data:
            tracks.extend(data["previousTracks"])
        if "futureTracks" in data:
            tracks.extend(data["futureTracks"])
        for track in tracks:
            artist = track["artist"]
            title = track["title"]
            time = time_to_date(string_to_time(track["time"], seconds=False))
            yield time, artist, title