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
1 changed files with 9 additions and 2 deletions
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("/")
src = providers[provider] + self.request.path
proxy_list = None
new_proxy_list = None
proxy_list_orig = proxies.get(provider)
if isinstance(proxy_list_orig, list):
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):
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:
resp = requests.head(src, allow_redirects=True, proxies=proxy_temp.req, timeout=(0.3, 2.0))
if resp is not None:
@@ -178,8 +186,7 @@ class MainHandler(tornado.web.RequestHandler):
except Exception as e:
print(e)
else:
proxy_list.remove(proxy_temp)
proxies[provider] = [proxy_temp] + proxy_list
proxies[provider] = new_proxy_list
proxy = proxy_temp
break
if proxy is not None: