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

fix proxy

Browse Source
This commit is contained in:
Roy Olav Purser 2021-05-11 14:59:43 +02:00
parent 69e4448db7
commit 482877e5c6
Signed by: roypur
GPG Key ID: E14D26A036F21656
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
stream.py
View File

@ -85,8 +85,7 @@ def get_proxy_url(proxy, current, path):
presp = requests.post(proxy_server, json=data) presp = requests.post(proxy_server, json=data)
return presp.text return presp.text
def rewrite(current, provider): def rewrite(current, provider, proxy):
proxy = proxies.get(provider)
resp = requests.head(current, proxies=proxy.req) resp = requests.head(current, proxies=proxy.req)
ctype = resp.headers.get("Content-Type") ctype = resp.headers.get("Content-Type")
if ctype is None: if ctype is None:
@ -168,7 +167,6 @@ class MainHandler(tornado.web.RequestHandler):
if provider == "youtube": if provider == "youtube":
path = path.strip("/") path = path.strip("/")
src = providers[provider] + self.request.path src = providers[provider] + self.request.path
proxy = None
proxy_list = None proxy_list = None
proxy_list_orig = proxies.get(provider) proxy_list_orig = proxies.get(provider)
if isinstance(proxy_list_orig, list): if isinstance(proxy_list_orig, list):
@ -200,7 +198,7 @@ class MainHandler(tornado.web.RequestHandler):
if write: if write:
self.write("Stream not found.") self.write("Stream not found.")
else: else:
data = rewrite(upstream, provider) data = rewrite(upstream, provider, proxy)
if data is None: if data is None:
self.redirect(upstream, status=303) self.redirect(upstream, status=303)
else: else: