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

match provider by prefix

Browse Source
This commit is contained in:
Roy Olav Purser 2021-07-23 17:38:59 +02:00
parent f739418141
commit ab34a21f2c
Signed by: roypur
GPG Key ID: E14D26A036F21656
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/stream.py
View File

@ -139,9 +139,9 @@ class UpstreamHandler():
if self.provider in providers.keys(): if self.provider in providers.keys():
self.valid = True self.valid = True
path = handler.request.path path = handler.request.path
if self.provider == "nextcloud": if self.provider.startswith("nextcloud"):
path = path.removesuffix("/").removesuffix("download").removesuffix("/") path = path.removesuffix("/").removesuffix("download").removesuffix("/")
elif self.provider == "youtube": elif self.provider.startswith("youtube"):
path = path.removeprefix("/") path = path.removeprefix("/")
self.upstream = providers[self.provider] + path self.upstream = providers[self.provider] + path
proxy_list = proxies.get(self.provider) proxy_list = proxies.get(self.provider)
@ -215,7 +215,7 @@ class MainHandler(tornado.web.RequestHandler):
async def handle_render(self, handler): async def handle_render(self, handler):
if template_script is not None and template_html is not None: if template_script is not None and template_html is not None:
provider_data = None provider_data = None
if handler.provider == "nextcloud": if handler.provider.startswith("nextcloud"):
provider_data = await stream_providers.get_nextcloud(handler.upstream, handler.proxy, logger) provider_data = await stream_providers.get_nextcloud(handler.upstream, handler.proxy, logger)
else: else:
provider_data = await stream_providers.get_any(handler.upstream, handler.proxy, logger) provider_data = await stream_providers.get_any(handler.upstream, handler.proxy, logger)