CF + override

This commit is contained in:
marco
2020-01-10 19:15:20 +01:00
parent 6ab5e3db79
commit 763c3e67da
2 changed files with 15 additions and 21 deletions
+8 -4
View File
@@ -251,19 +251,23 @@ def downloadpage(url, **opt):
""" """
load_cookies() load_cookies()
if urlparse.urlparse(url).netloc in ['www.guardaserie.media', 'casacinema.space']: domain = urlparse.urlparse(url).netloc
CF = False
if domain in ['www.guardaserie.media', 'casacinema.space']:
from lib import cloudscraper from lib import cloudscraper
session = cloudscraper.create_scraper() session = cloudscraper.create_scraper()
CF = True
elif opt.get('session', False): elif opt.get('session', False):
session = opt['session'] # same session to speed up search session = opt['session'] # same session to speed up search
logger.info('same session') logger.info('same session')
elif config.get_setting('resolver_dns') and not opt.get('use_requests', False):
from specials import resolverdns
session = resolverdns.session()
else: else:
from lib import requests from lib import requests
session = requests.session() session = requests.session()
if config.get_setting('resolver_dns') and not opt.get('use_requests', False):
from specials import resolverdns
session.mount('https://', resolverdns.CipherSuiteAdapter(domain, CF))
req_headers = default_headers.copy() req_headers = default_headers.copy()
verify = opt.get('verify', True) verify = opt.get('verify', True)
+7 -17
View File
@@ -37,10 +37,12 @@ class CustomContext(ssl.SSLContext):
class CipherSuiteAdapter(host_header_ssl.HostHeaderSSLAdapter): class CipherSuiteAdapter(host_header_ssl.HostHeaderSSLAdapter):
def __init__(self, domain, *args, **kwargs): def __init__(self, domain, CF=False, *args, **kwargs):
self.conn = sql.connect(db) self.conn = sql.connect(db)
self.cur = self.conn.cursor() self.cur = self.conn.cursor()
self.ssl_context = CustomContext(ssl.PROTOCOL_TLS, domain) self.ssl_context = CustomContext(ssl.PROTOCOL_TLS, domain)
self.CF = CF # if cloudscrape is in action
self.cipherSuite = kwargs.pop('cipherSuite', ssl._DEFAULT_CIPHERS)
super(CipherSuiteAdapter, self).__init__(**kwargs) super(CipherSuiteAdapter, self).__init__(**kwargs)
@@ -97,9 +99,10 @@ class CipherSuiteAdapter(host_header_ssl.HostHeaderSSLAdapter):
domain = parse.netloc domain = parse.netloc
else: else:
raise requests.exceptions.URLRequired raise requests.exceptions.URLRequired
print domain
self.ssl_context = CustomContext(ssl.PROTOCOL_TLS, domain) self.ssl_context = CustomContext(ssl.PROTOCOL_TLS, domain)
self.ssl_context.set_ciphers(ssl._DEFAULT_CIPHERS) if self.CF:
self.ssl_context.options |= (ssl.OP_NO_SSLv2 | ssl.OP_NO_SSLv3 | ssl.OP_NO_TLSv1 | ssl.OP_NO_TLSv1_1)
self.ssl_context.set_ciphers(self.cipherSuite)
self.init_poolmanager(self._pool_connections, self._pool_maxsize, block=self._pool_block) self.init_poolmanager(self._pool_connections, self._pool_maxsize, block=self._pool_block)
ip = self.getIp(domain) ip = self.getIp(domain)
@@ -121,21 +124,8 @@ class CipherSuiteAdapter(host_header_ssl.HostHeaderSSLAdapter):
logger.info('Request for ' + domain + ' with ip ' + ip + ' failed') logger.info('Request for ' + domain + ' with ip ' + ip + ' failed')
logger.info(e) logger.info(e)
tryFlush = True tryFlush = True
if (tryFlush or not ret) and not flushedDns: # re-request ips and update cache if tryFlush and not flushedDns: # re-request ips and update cache
logger.info('Flushing dns cache for ' + domain) logger.info('Flushing dns cache for ' + domain)
return self.flushDns(request, domain, **kwargs) return self.flushDns(request, domain, **kwargs)
ret.url = realUrl ret.url = realUrl
return ret return ret
class session(requests.Session):
def request(self, method, url, headers=None, flushedDns=False, **kwargs):
try:
parse = urlparse.urlparse(url)
except:
raise requests.exceptions.InvalidURL
if parse.netloc:
domain = parse.netloc
else:
raise requests.exceptions.URLRequired
self.mount('https://', CipherSuiteAdapter(domain))
return super(session, self).request(method, url, **kwargs)