add proxy support
This commit is contained in:
parent
bbca0d2fd7
commit
6650cd1818
16
stream.py
16
stream.py
@ -3,6 +3,7 @@ import streamlink
|
|||||||
import tornado.web
|
import tornado.web
|
||||||
import tornado.routing
|
import tornado.routing
|
||||||
import requests
|
import requests
|
||||||
|
import os
|
||||||
|
|
||||||
providers = {}
|
providers = {}
|
||||||
providers["nrk"] = "https://tv.nrk.no"
|
providers["nrk"] = "https://tv.nrk.no"
|
||||||
@ -10,11 +11,24 @@ providers["svt"] = "https://svtplay.se"
|
|||||||
providers["youtube"] = "https://youtu.be"
|
providers["youtube"] = "https://youtu.be"
|
||||||
providers["twitch"] = "https://twitch.tv"
|
providers["twitch"] = "https://twitch.tv"
|
||||||
|
|
||||||
|
proxies = {}
|
||||||
|
proxies["nrk"] = streamlink.Streamlink()
|
||||||
|
proxies["svt"] = streamlink.Streamlink()
|
||||||
|
proxies["youtube"] = streamlink.Streamlink()
|
||||||
|
proxies["twitch"] = streamlink.Streamlink()
|
||||||
|
|
||||||
|
for key in proxies:
|
||||||
|
proxy = os.environ.get(key.upper() + "_PROXY")
|
||||||
|
proxies[key].set_option("http-timeout", 2.0)
|
||||||
|
if proxy is not None:
|
||||||
|
proxies[key].set_option("https-proxy", proxy)
|
||||||
|
|
||||||
class MainHandler(tornado.web.RequestHandler):
|
class MainHandler(tornado.web.RequestHandler):
|
||||||
def get(self):
|
def get(self):
|
||||||
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():
|
||||||
|
proxy = proxies.get(provider)
|
||||||
src = providers[provider] + self.request.uri
|
src = providers[provider] + self.request.uri
|
||||||
try:
|
try:
|
||||||
resp = requests.get(src)
|
resp = requests.get(src)
|
||||||
@ -25,7 +39,7 @@ class MainHandler(tornado.web.RequestHandler):
|
|||||||
return
|
return
|
||||||
else:
|
else:
|
||||||
try:
|
try:
|
||||||
streams = streamlink.streams(src)
|
streams = proxy.streams(src)
|
||||||
for key in reversed(streams):
|
for key in reversed(streams):
|
||||||
stream = streams.get(key)
|
stream = streams.get(key)
|
||||||
if hasattr(stream, "url"):
|
if hasattr(stream, "url"):
|
||||||
|
Loading…
x
Reference in New Issue
Block a user