plugin.video.torrenter3/resources/lib/overrrides/session.py

86 lines
3.8 KiB
Python
Raw Normal View History

2019-05-04 22:38:06 +03:00
import urlquick # @UnresolvedImport
from ..settings import option
import socks
class SocksiPyConnection(urlquick.HTTPConnection):
def __init__(self, proxytype, proxyaddr, proxyport=None, host='127.0.0.1', port=9050, rdns=True, username=None, password=None, **kwargs):
self.proxyargs = (proxytype, proxyaddr, proxyport, rdns, username, password)
urlquick.HTTPConnection.__init__(self, host, port=port, **kwargs)
def connect(self):
self.sock = socks.socksocket()
self.sock.setproxy(*self.proxyargs)
if type(self.timeout) in (int, float):
self.sock.settimeout(self.timeout)
self.sock.connect((self.host, self.port))
class SocksiPyConnectionS(urlquick.HTTPSConnection):
def __init__(self, proxytype, proxyaddr, proxyport=None, host='127.0.0.1', port=9050, rdns=True, username=None, password=None, **kwargs):
self.proxyargs = (proxytype, proxyaddr, proxyport, rdns, username, password)
urlquick.HTTPSConnection.__init__(self, host, port=port, **kwargs)
def connect(self):
sock = socks.socksocket()
sock.setproxy(*self.proxyargs)
if type(self.timeout) in (int, float):
sock.settimeout(self.timeout)
sock.connect((self.host, self.port))
self.sock = urlquick.ssl.wrap_socket(sock, self.key_file, self.cert_file)
class Session(urlquick.Session):
def connect(self, req, timeout, verify):
# Fetch connection from pool and attempt to reuse if available
pool = self.request_handler[req.type]
if req.host in pool:
try:
# noinspection PyTypeChecker
return self.send_request(pool[req.host], req)
except Exception as e:
# Remove the connection from the pool as it's unusable
pool[req.host].close()
del pool[req.host]
# Raise the exception if it's not a subclass of UrlError
if not isinstance(e, urlquick.UrlError):
raise
host_port = req.host.split(':')
host = host_port[0]
port = int(host_port[2]) if len(host_port) > 1 else 443 if req.type == 'https' else 80
# Create a new connection
if not option.get_boolean('use_socks'): # @UndefinedVariable
if req.type == "https":
context = urlquick.ssl._create_unverified_context() if verify is False else None
conn = urlquick.HTTPSConnection(host, port, timeout=timeout, context=context)
else:
conn = urlquick.HTTPConnection(host, port, timeout=timeout)
else:
if req.type == "https":
context = urlquick.ssl._create_unverified_context() if verify is False else None
conn = SocksiPyConnectionS(socks.PROXY_TYPE_SOCKS5, option['socks_ip'], proxyport=int(option['socks_port']),
host=host, port=port, timeout=timeout, context=context)
else:
conn = SocksiPyConnection(socks.PROXY_TYPE_SOCKS5, option['socks_ip'], proxyport=int(option['socks_port']),
host=host, port=port, timeout=timeout)
# Make first connection to server
response = self.send_request(conn, req)
# Add connection to the pool if the response is not set to close
if not response.will_close:
pool[req.host] = conn
return response
def torrent_file_fetch(url, referer, cookies):
sess = Session()
headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/37.0.2062.124 YaBrowser/14.10.2062.12061 Safari/537.36',
'Referer': referer
}
resp = sess.get(url, cookies=cookies, headers=headers)
if resp.ok:
return resp.content
else:
return None