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

proxy selection logic

Browse Source
This commit is contained in:
Roy Olav Purser 2021-05-11 16:16:33 +02:00
parent 02f8786414
commit 8a113fbb41
Signed by: roypur
GPG Key ID: E14D26A036F21656
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
stream.py
View File

@ -166,11 +166,19 @@ class MainHandler(tornado.web.RequestHandler):
path = path.strip("/") path = path.strip("/")
src = providers[provider] + self.request.path src = providers[provider] + self.request.path
proxy_list = None proxy_list = None
new_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):
proxy_list = proxy_list_orig.copy() proxy_list = proxy_list_orig.copy()
new_proxy_list = proxy_list_orig.copy()
if len(new_proxy_list) > 1:
first_proxy = new_proxy_list.pop()
new_proxy_list = [first_proxy] + new_proxy_list
if isinstance(proxy_list, list): if isinstance(proxy_list, list):
for proxy_temp in proxy_list: for proxy_temp in proxy_list:
if len(proxy_list) > 1:
first_proxy = new_proxy_list.pop(0)
new_proxy_list = new_proxy_list + [first_proxy]
try: try:
resp = requests.head(src, allow_redirects=True, proxies=proxy_temp.req, timeout=(0.3, 2.0)) resp = requests.head(src, allow_redirects=True, proxies=proxy_temp.req, timeout=(0.3, 2.0))
if resp is not None: if resp is not None:
@ -178,8 +186,7 @@ class MainHandler(tornado.web.RequestHandler):
except Exception as e: except Exception as e:
print(e) print(e)
else: else:
proxy_list.remove(proxy_temp) proxies[provider] = new_proxy_list
proxies[provider] = [proxy_temp] + proxy_list
proxy = proxy_temp proxy = proxy_temp
break break
if proxy is not None: if proxy is not None: