add proxy support

This commit is contained in:
Roy Olav Purser 2021-04-30 13:55:10 +02:00
parent bbca0d2fd7
commit 6650cd1818
No known key found for this signature in database
GPG Key ID: 0BA77797F072BC52

View File

@ -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"):