Allow silent mode using 'verbose' parameter

If a verbose=False parameter is passed to the speedtest function, the
test doesn't print any message, but returns the results list.
This commit is contained in:
Ale Almuna 2015-12-20 20:43:08 -03:00
parent 8730705515
commit a2d3d72858
1 changed files with 16 additions and 5 deletions

View File

@ -614,10 +614,19 @@ def speedtest(**kwargs):
opts_dict['secure'] = True if 'secure' in kwargs else False
opts_dict['version'] = True if 'version' in kwargs else False
opts_dict['simple'] = kwargs['simple'] if 'simple' in kwargs else False
opts_dict['verbose'] = kwargs[
'verbose'] if 'verbose' in kwargs else True if opts_dict[
'simple'] else False
if not opts_dict['verbose']:
opts_dict['simple'] = True
# Object to access dictionary values by attribute lookup
args = type('TestOptions', (object,), opts_dict)
try:
args.verbose
except:
args.verbose = True
# Print the version and exit
if args.version:
version()
@ -723,7 +732,7 @@ def speedtest(**kwargs):
if not args.simple:
print_(('Hosted by %(sponsor)s (%(name)s) [%(d)0.2f km]: '
'%(latency)s ms' % best).encode('utf-8', 'ignore'))
else:
elif args.verbose:
print_('Ping: %(latency)s ms' % best)
sizes = [350, 500, 750, 1000, 1500, 2000, 2500, 3000, 3500, 4000]
@ -737,6 +746,7 @@ def speedtest(**kwargs):
dlspeed = downloadSpeed(urls, args.simple)
if not args.simple:
print_()
if args.verbose:
print_('Download: %0.2f M%s/s' %
((dlspeed / 1000 / 1000) * args.units[1], args.units[0]))
@ -750,6 +760,7 @@ def speedtest(**kwargs):
ulspeed = uploadSpeed(best['url'], sizes, args.simple)
if not args.simple:
print_()
if args.verbose:
print_('Upload: %0.2f M%s/s' %
((ulspeed / 1000 / 1000) * args.units[1], args.units[0]))