Merge 15a922c901
into 18408ee938
This commit is contained in:
commit
e9bd7bf1ce
|
@ -28,6 +28,16 @@ Displays usage for the tool.
|
||||||
Generate and provide a URL to the speedtest.net share results image
|
Generate and provide a URL to the speedtest.net share results image
|
||||||
.RE
|
.RE
|
||||||
|
|
||||||
|
\fB\-\-uid USERNAME\fR
|
||||||
|
.RS
|
||||||
|
Supply speedtest.net username so that shared results will be listed in the speedtest.net "My Results" page. Must also supply --pw. Implies --share.
|
||||||
|
.RE
|
||||||
|
|
||||||
|
\fB\-\-pw PASSWORD\fR
|
||||||
|
.RS
|
||||||
|
Supply speedtest.net password
|
||||||
|
.RE
|
||||||
|
|
||||||
\fB\-\-simple\fR
|
\fB\-\-simple\fR
|
||||||
.RS
|
.RS
|
||||||
Suppress verbose output, only show basic information
|
Suppress verbose output, only show basic information
|
||||||
|
|
|
@ -44,9 +44,11 @@ except ImportError:
|
||||||
|
|
||||||
# Begin import game to handle Python 2 and Python 3
|
# Begin import game to handle Python 2 and Python 3
|
||||||
try:
|
try:
|
||||||
from urllib2 import urlopen, Request, HTTPError, URLError
|
from urllib2 import urlopen, Request, HTTPError, URLError, \
|
||||||
|
HTTPErrorProcessor, build_opener
|
||||||
except ImportError:
|
except ImportError:
|
||||||
from urllib.request import urlopen, Request, HTTPError, URLError
|
from urllib.request import urlopen, Request, HTTPError, URLError, \
|
||||||
|
HTTPErrorProcessor, build_opener
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from Queue import Queue
|
from Queue import Queue
|
||||||
|
@ -58,6 +60,11 @@ try:
|
||||||
except ImportError:
|
except ImportError:
|
||||||
from urllib.parse import urlparse
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
|
try:
|
||||||
|
from urllib import urlencode
|
||||||
|
except ImportError:
|
||||||
|
from urllib.parse import urlencode
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from urlparse import parse_qs
|
from urlparse import parse_qs
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
@ -160,6 +167,15 @@ def distance(origin, destination):
|
||||||
return d
|
return d
|
||||||
|
|
||||||
|
|
||||||
|
class NoRedirection(HTTPErrorProcessor):
|
||||||
|
"""httplib class to ignore redirections"""
|
||||||
|
|
||||||
|
def http_response(self, request, response):
|
||||||
|
return response
|
||||||
|
|
||||||
|
https_response = http_response
|
||||||
|
|
||||||
|
|
||||||
class FileGetter(threading.Thread):
|
class FileGetter(threading.Thread):
|
||||||
"""Thread class for retrieving a URL"""
|
"""Thread class for retrieving a URL"""
|
||||||
|
|
||||||
|
@ -428,6 +444,35 @@ def version():
|
||||||
raise SystemExit(__version__)
|
raise SystemExit(__version__)
|
||||||
|
|
||||||
|
|
||||||
|
def speedtest_auth(uid, pw):
|
||||||
|
"""Log in to speedtest.net"""
|
||||||
|
|
||||||
|
authlist = []
|
||||||
|
|
||||||
|
noredir = build_opener(NoRedirection)
|
||||||
|
|
||||||
|
postData = urlencode({'email': uid, 'password': pw, 'action': 'login'})
|
||||||
|
|
||||||
|
req = Request('https://www.speedtest.net/user-login.php', postData)
|
||||||
|
f = noredir.open(req)
|
||||||
|
cookies = f.info().getallmatchingheaders('Set-Cookie')
|
||||||
|
code = f.code
|
||||||
|
f.close()
|
||||||
|
|
||||||
|
if int(code) != 302:
|
||||||
|
print_('Could not log in to speedtest.net')
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
for cookieline in cookies:
|
||||||
|
cookieline = re.sub("(\r|\n)", "", cookieline)
|
||||||
|
cookieline = re.sub("^Set-Cookie:\s*", "", cookieline)
|
||||||
|
cookieline = re.sub(";.*$", "", cookieline)
|
||||||
|
if re.match("(stnetsid|expire)=", cookieline):
|
||||||
|
authlist.append(cookieline)
|
||||||
|
|
||||||
|
return "; ".join(authlist)
|
||||||
|
|
||||||
|
|
||||||
def speedtest():
|
def speedtest():
|
||||||
"""Run the full speedtest.net test"""
|
"""Run the full speedtest.net test"""
|
||||||
|
|
||||||
|
@ -450,6 +495,8 @@ def speedtest():
|
||||||
parser.add_argument = parser.add_option
|
parser.add_argument = parser.add_option
|
||||||
except AttributeError:
|
except AttributeError:
|
||||||
pass
|
pass
|
||||||
|
parser.add_argument('--uid', help='Specify speedtest.net username')
|
||||||
|
parser.add_argument('--pw', help='Specify speedtest.net password')
|
||||||
parser.add_argument('--share', action='store_true',
|
parser.add_argument('--share', action='store_true',
|
||||||
help='Generate and provide a URL to the speedtest.net '
|
help='Generate and provide a URL to the speedtest.net '
|
||||||
'share results image')
|
'share results image')
|
||||||
|
@ -481,6 +528,17 @@ def speedtest():
|
||||||
source = args.source
|
source = args.source
|
||||||
socket.socket = bound_socket
|
socket.socket = bound_socket
|
||||||
|
|
||||||
|
if (args.uid and not args.pw) or (args.pw and not args.uid):
|
||||||
|
print_('Must specify both uid and pw or neither')
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
if args.uid:
|
||||||
|
if not args.share:
|
||||||
|
print_('Must share to post to My Results; auto-enabling')
|
||||||
|
auth = speedtest_auth(args.uid, args.pw)
|
||||||
|
else:
|
||||||
|
auth = None
|
||||||
|
|
||||||
if not args.simple:
|
if not args.simple:
|
||||||
print_('Retrieving speedtest.net configuration...')
|
print_('Retrieving speedtest.net configuration...')
|
||||||
try:
|
try:
|
||||||
|
@ -601,7 +659,7 @@ def speedtest():
|
||||||
if args.share and args.mini:
|
if args.share and args.mini:
|
||||||
print_('Cannot generate a speedtest.net share results image while '
|
print_('Cannot generate a speedtest.net share results image while '
|
||||||
'testing against a Speedtest Mini server')
|
'testing against a Speedtest Mini server')
|
||||||
elif args.share:
|
elif args.share or args.uid:
|
||||||
dlspeedk = int(round((dlspeed / 1000) * 8, 0))
|
dlspeedk = int(round((dlspeed / 1000) * 8, 0))
|
||||||
ping = int(round(best['latency'], 0))
|
ping = int(round(best['latency'], 0))
|
||||||
ulspeedk = int(round((ulspeed / 1000) * 8, 0))
|
ulspeedk = int(round((ulspeed / 1000) * 8, 0))
|
||||||
|
@ -625,6 +683,8 @@ def speedtest():
|
||||||
req = Request('http://www.speedtest.net/api/api.php',
|
req = Request('http://www.speedtest.net/api/api.php',
|
||||||
data='&'.join(apiData).encode())
|
data='&'.join(apiData).encode())
|
||||||
req.add_header('Referer', 'http://c.speedtest.net/flash/speedtest.swf')
|
req.add_header('Referer', 'http://c.speedtest.net/flash/speedtest.swf')
|
||||||
|
if auth:
|
||||||
|
req.add_header('Cookie', auth)
|
||||||
f = urlopen(req)
|
f = urlopen(req)
|
||||||
response = f.read()
|
response = f.read()
|
||||||
code = f.code
|
code = f.code
|
||||||
|
@ -640,7 +700,9 @@ def speedtest():
|
||||||
print_('Could not submit results to speedtest.net')
|
print_('Could not submit results to speedtest.net')
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
print_('Share results: http://www.speedtest.net/result/%s.png' %
|
print_('Share results: http://www.speedtest.net/my-result/%s' %
|
||||||
|
resultid[0])
|
||||||
|
print_('Share results image: http://www.speedtest.net/result/%s.png' %
|
||||||
resultid[0])
|
resultid[0])
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue