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

clean up duplicate header code

Browse source
This commit is contained in:
milaq 2019-01-21 02:59:21 +01:00
parent a1e6fdb8f8
commit 5e2733f841
1 changed files with 9 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
ycast.py
View file

@ -9,7 +9,6 @@ import xml.etree.cElementTree as etree
import yaml
VTUNER_INITURL = '/setupapp/Yamaha/asp/BrowseXML/loginXML.asp'
XMLHEADER = '<?xml version="1.0" encoding="UTF-8" standalone="yes" ?>'
YCAST_LOCATION = 'ycast'
stations = {}
@ -39,19 +38,13 @@ class YCastServer(BaseHTTPRequestHandler):
get_stations()
self.address = 'http://' + self.headers['Host']
if self.path.startswith(VTUNER_INITURL + '?token='):
self.send_response(200)
self.send_header('Content-type', 'text/html')
self.end_headers()
self.wfile.write(bytes(XMLHEADER, 'utf-8'))
self.send_xml_header()
self.wfile.write(bytes('<EncryptedToken>stub</EncryptedToken>', 'utf-8'))
elif 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'))
self.send_xml_header()
xml = self.create_root()
for category in sorted(stations, key=str.lower):
self.add_dir(xml, category,
@ -65,14 +58,17 @@ class YCastServer(BaseHTTPRequestHandler):
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.send_xml_header()
self.wfile.write(bytes(etree.tostring(xml).decode('utf-8'), 'utf-8'))
else:
self.send_error(404)
def send_xml_header(self):
self.send_response(200)
self.send_header('Content-type', 'text/html')
self.end_headers()
self.wfile.write(bytes('<?xml version="1.0" encoding="UTF-8" standalone="yes" ?>', 'utf-8'))
def create_root(self):
return etree.Element('ListOfItems')