beenull/YCast
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

use host address from headers for redirects

Browse source 
allow compatibility with other addresses (e.g. AVRs from other manufacturers)
This commit is contained in:
milaq 2019-01-21 02:38:56 +01:00
parent 891384350c
commit c085974137
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
ycast.py
View file

@ -8,7 +8,6 @@ import xml.etree.cElementTree as etree
import yaml
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'
@ -38,6 +37,7 @@ def url_to_text(url):
class YCastServer(BaseHTTPRequestHandler):
def do_GET(self):
get_stations()
self.address = 'http://' + self.headers['Host']
if self.path == '/' \
or self.path == '/' + YCAST_LOCATION \
or self.path == '/' + YCAST_LOCATION + '/'\
@ -49,7 +49,7 @@ class YCastServer(BaseHTTPRequestHandler):
xml = self.create_root()
for category in sorted(stations, key=str.lower):
self.add_dir(xml, category,
VTUNER_DNS + '/' + YCAST_LOCATION + '/' + text_to_url(category))
self.address + '/' + 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])