fmplapla

#!/usr/bin/env python3
# coding: utf-8
# vim:fenc=utf-8 ff=unix ft=python ts=4 sw=4 sts=4 si et :

"""
pre-flight:
    pip install requests websocket-client

usage:
    fmplapla --station fmhanabi --time 1800 | ffplay -nodisp -hide_banner -autoexit -i pipe:
    fmplapla --station fmhanabi             | mpv -
    env DEBUG=1 fmplapla -s fmhanabi # debug print

thanks to:
    fmplapla.py - Pastebin.com https://pastebin.com/gQ0Ay2aN
"""

import os
import sys
import time
import argparse
import requests
import websocket
import pprint
import http


class fmplapla:
    def __init__(self, station_id, duration=0):
        self.station_id = station_id
        self.duration = duration
        self.start_time = time.time()
        if 'DEBUG' in os.environ:
            websocket.enableTrace(True)
            http.client.HTTPConnection.debuglevel=1

        # token, location
        headers = {
                #"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:98.0) Gecko/20100101 Firefox/98.0",
                "Origin": "https://fmplapla.com"
                }
        url = 'https://fmplapla.com/api/select_stream?station={st}&burst=5'.format(st=self.station_id)
        res = requests.post(url, headers=headers)
        json = res.json()
        if 'DEBUG' in os.environ:
            print(url)
            pprint.pprint(json)
        self.token = json['token']
        self.location = json['location']

        self.ws = websocket.WebSocketApp(
            self.location,
            subprotocols=['listener.fmplapla.com'],
            on_open=self._on_open, on_message=self._on_message)
        try:
            self.ws.run_forever()
        except (Exception, KeyboardInterrupt, SystemExit) as e:
            self.ws.close()

    def _on_message(self, data, message):
        if data:
            sys.stdout.buffer.write(message)
        if self.duration > 0:
            if self.duration < (time.time() - self.start_time):
                raise KeyboardInterrupt

    def _on_open(self, data):
        self.ws.send(self.token)


def main():
    parser = argparse.ArgumentParser(
        description='example: python fmplapla.py -s fmrara768 -t 1800 | ffplay -i pipe:')
    parser.add_argument('-s', '--station', required=True,
                        help='station id. example: fmrara768')
    parser.add_argument('-t', '--time', type=int, default=0,
                        help='stop writing the output after its seconds reaches duration. it defaults to 0, meaning that loop forever.')
    args = parser.parse_args()
    radio = fmplapla(args.station, args.time)


if __name__ == '__main__':
    main()