94 lines
3.1 KiB
Python
Executable file
94 lines
3.1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import os
|
|
from http.server import BaseHTTPRequestHandler, HTTPServer
|
|
import xml.etree.cElementTree as etree
|
|
|
|
import yaml
|
|
|
|
listen_address = '0.0.0.0'
|
|
listen_port = 80
|
|
|
|
VTUNER_DNS = 'http://radioyamaha.vtuner.com'
|
|
VTUNER_INITURL = '/setupapp/Yamaha/asp/BrowseXML/loginXML.asp'
|
|
XMLHEADER = '<?xml version="1.0" encoding="UTF-8" standalone="yes" ?>'
|
|
YCAST_LOCATION = 'ycast'
|
|
|
|
stations = {}
|
|
|
|
|
|
def get_stations():
|
|
global stations
|
|
ycast_dir = os.path.dirname(os.path.realpath(__file__))
|
|
with open(ycast_dir + '/stations.yml', 'r') as f:
|
|
stations = yaml.load(f)
|
|
|
|
|
|
def text_to_url(text):
|
|
return text.replace(' ', '%20')
|
|
|
|
|
|
def url_to_text(url):
|
|
return url.replace('%20', ' ')
|
|
|
|
|
|
class YCastServer(BaseHTTPRequestHandler):
|
|
def do_GET(self):
|
|
get_stations()
|
|
if self.path == '/' \
|
|
or self.path == '/' + YCAST_LOCATION \
|
|
or self.path == '/' + YCAST_LOCATION + '/'\
|
|
or self.path.startswith(VTUNER_INITURL):
|
|
self.send_response(200)
|
|
self.send_header('Content-type', 'text/html')
|
|
self.end_headers()
|
|
self.wfile.write(bytes(XMLHEADER, 'utf-8'))
|
|
xml = self.create_root()
|
|
for category in sorted(stations, key=str.lower):
|
|
self.add_dir(xml, category,
|
|
VTUNER_DNS + ':' + str(listen_port) + '/' + YCAST_LOCATION + '/' + text_to_url(category))
|
|
self.wfile.write(bytes(etree.tostring(xml).decode('utf-8'), 'utf-8'))
|
|
elif self.path.startswith('/' + YCAST_LOCATION + '/'):
|
|
category = url_to_text(self.path[len(YCAST_LOCATION) + 2:].partition('?')[0])
|
|
if category not in stations:
|
|
self.send_error(404)
|
|
return
|
|
xml = self.create_root()
|
|
for station in sorted(stations[category], key=str.lower):
|
|
self.add_station(xml, station, stations[category][station])
|
|
self.send_response(200)
|
|
self.send_header('Content-type', 'text/html')
|
|
self.end_headers()
|
|
self.wfile.write(bytes(XMLHEADER, 'utf-8'))
|
|
self.wfile.write(bytes(etree.tostring(xml).decode('utf-8'), 'utf-8'))
|
|
else:
|
|
self.send_error(404)
|
|
|
|
def create_root(self):
|
|
return etree.Element('ListOfItems')
|
|
|
|
def add_dir(self, root, name, dest):
|
|
item = etree.SubElement(root, 'Item')
|
|
etree.SubElement(item, 'ItemType').text = 'Dir'
|
|
etree.SubElement(item, 'Title').text = name
|
|
etree.SubElement(item, 'UrlDir').text = dest
|
|
return item
|
|
|
|
def add_station(self, root, name, url):
|
|
item = etree.SubElement(root, 'Item')
|
|
etree.SubElement(item, 'ItemType').text = 'Station'
|
|
etree.SubElement(item, 'StationName').text = name
|
|
etree.SubElement(item, 'StationUrl').text = url
|
|
return item
|
|
|
|
|
|
get_stations()
|
|
server = HTTPServer((listen_address, listen_port), YCastServer)
|
|
print('YCast server listening on %s:%s' % (listen_address, listen_port))
|
|
try:
|
|
server.serve_forever()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
print('YCast server shutting down')
|
|
server.server_close()
|