Remove SCHEME global

This commit is contained in:
Matt Martz 2017-05-02 12:29:54 -05:00
parent b33c7533df
commit 2fe369fdf8
1 changed files with 31 additions and 19 deletions

View File

@ -52,7 +52,6 @@ class FakeShutdownEvent(object):
# Some global variables we use
SHUTDOWN_EVENT = FakeShutdownEvent()
SCHEME = 'http'
DEBUG = False
_GLOBAL_DEFAULT_TIMEOUT = object()
@ -554,7 +553,7 @@ def build_user_agent():
return user_agent
def build_request(url, data=None, headers=None, bump=''):
def build_request(url, data=None, headers=None, bump='', secure=False):
"""Build a urllib2 request object
This function automatically adds a User-Agent header to all requests
@ -565,7 +564,9 @@ def build_request(url, data=None, headers=None, bump=''):
headers = {}
if url[0] == ':':
schemed_url = '%s%s' % (SCHEME, url)
scheme = ('http', 'https')[bool(secure)]
print scheme
schemed_url = '%s%s' % (scheme, url)
else:
schemed_url = url
@ -780,7 +781,8 @@ class SpeedtestResults(object):
to get a share results image link.
"""
def __init__(self, download=0, upload=0, ping=0, server=None, opener=None):
def __init__(self, download=0, upload=0, ping=0, server=None, opener=None,
secure=False):
self.download = download
self.upload = upload
self.ping = ping
@ -798,6 +800,8 @@ class SpeedtestResults(object):
else:
self._opener = build_opener()
self._secure = secure
def __repr__(self):
return repr(self.dict())
@ -839,7 +843,7 @@ class SpeedtestResults(object):
headers = {'Referer': 'http://c.speedtest.net/flash/speedtest.swf'}
request = build_request('://www.speedtest.net/api/api.php',
data='&'.join(api_data).encode(),
headers=headers)
headers=headers, secure=self._secure)
f, e = catch_request(request, opener=self._opener)
if e:
raise ShareResultsConnectFailure(e)
@ -904,13 +908,16 @@ class SpeedtestResults(object):
class Speedtest(object):
"""Class for performing standard speedtest.net testing operations"""
def __init__(self, config=None, source_address=None, timeout=10):
def __init__(self, config=None, source_address=None, timeout=10,
secure=False):
self.config = {}
self._source_address = source_address
self._timeout = timeout
self._opener = build_opener(source_address, timeout)
self._secure = secure
self.get_config()
if config is not None:
self.config.update(config)
@ -919,7 +926,7 @@ class Speedtest(object):
self.closest = []
self.best = {}
self.results = SpeedtestResults(opener=self._opener)
self.results = SpeedtestResults(opener=self._opener, secure=secure)
def get_config(self):
"""Download the speedtest.net configuration and return only the data
@ -930,7 +937,7 @@ class Speedtest(object):
if gzip:
headers['Accept-Encoding'] = 'gzip'
request = build_request('://www.speedtest.net/speedtest-config.php',
headers=headers)
headers=headers, secure=self._secure)
uh, e = catch_request(request, opener=self._opener)
if e:
raise ConfigRetrievalError(e)
@ -1044,10 +1051,12 @@ class Speedtest(object):
errors = []
for url in urls:
try:
request = build_request('%s?threads=%s' %
(url,
request = build_request(
'%s?threads=%s' % (url,
self.config['threads']['download']),
headers=headers)
headers=headers,
secure=self._secure
)
uh, e = catch_request(request, opener=self._opener)
if e:
errors.append('%s' % e)
@ -1274,7 +1283,9 @@ class Speedtest(object):
request_count = len(urls)
requests = []
for i, url in enumerate(urls):
requests.append(build_request(url, bump=i))
requests.append(
build_request(url, bump=i, secure=self._secure)
)
def producer(q, requests, request_count):
for i, request in enumerate(requests):
@ -1338,7 +1349,7 @@ class Speedtest(object):
data.pre_allocate()
requests.append(
(
build_request(self.best['url'], data),
build_request(self.best['url'], data, secure=self._secure),
size
)
)
@ -1526,7 +1537,7 @@ def printer(string, quiet=False, debug=False, **kwargs):
def shell():
"""Run the full speedtest.net test"""
global SHUTDOWN_EVENT, SCHEME, DEBUG
global SHUTDOWN_EVENT, DEBUG
SHUTDOWN_EVENT = threading.Event()
signal.signal(signal.SIGINT, ctrl_c)
@ -1549,9 +1560,6 @@ def shell():
validate_optional_args(args)
if args.secure:
SCHEME = 'https'
debug = getattr(args, 'debug', False)
if debug == 'SUPPRESSHELP':
debug = False
@ -1576,7 +1584,11 @@ def shell():
printer('Retrieving speedtest.net configuration...', quiet)
try:
speedtest = Speedtest(source_address=args.source, timeout=args.timeout)
speedtest = Speedtest(
source_address=args.source,
timeout=args.timeout,
secure=args.secure
)
except (ConfigRetrievalError,) + HTTP_ERRORS:
printer('Cannot retrieve speedtest configuration')
raise SpeedtestCLIError(get_exception())