roypur/stream-api
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add error message

Browse Source
This commit is contained in:
Roy Olav Purser 2021-05-06 15:54:18 +02:00
parent 8f2744ced7
commit 3fc31165e5
Signed by: roypur
GPG Key ID: E14D26A036F21656
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
stream.py
View File

@ -27,6 +27,7 @@ for key in proxies:
playlist = None playlist = None
icecast_server = os.environ.get("ICECAST_SERVER") icecast_server = os.environ.get("ICECAST_SERVER")
stream_server = os.environ.get("STREAM_SERVER") stream_server = os.environ.get("STREAM_SERVER")
if icecast_server is not None and stream_server is not None: if icecast_server is not None and stream_server is not None:
with open("/app/sources.json", "r") as f: with open("/app/sources.json", "r") as f:
data = json.loads(f.read()) data = json.loads(f.read())
@ -43,7 +44,7 @@ if icecast_server is not None and stream_server is not None:
playlist += stream_server + key + "\n" playlist += stream_server + key + "\n"
class MainHandler(tornado.web.RequestHandler): class MainHandler(tornado.web.RequestHandler):
def handle_any(self): def handle_any(self, write):
provider = self.get_query_argument("provider", None) provider = self.get_query_argument("provider", None)
endpoint = None endpoint = None
if provider is not None and provider in providers.keys(): if provider is not None and provider in providers.keys():
@ -67,13 +68,15 @@ class MainHandler(tornado.web.RequestHandler):
print(str(e)) print(str(e))
endpoint = None endpoint = None
if endpoint is None: if endpoint is None:
self.set_status(404, reason="stream not found") self.set_status(404)
if write:
self.write("stream not found")
else: else:
self.redirect(endpoint, status=303) self.redirect(endpoint, status=303)
def get(self): def get(self):
self.handle_any() self.handle_any(True)
def head(self): def head(self):
self.handle_any() self.handle_any(False)
class FileHandler(tornado.web.RequestHandler): class FileHandler(tornado.web.RequestHandler):
def get(self): def get(self):