#!/usr/bin/python3

"""
Module to measure and report Internet speed
Method: get one or more files, and measure how long it takes
Reports in MB/s (so mega BYTES per seconds), not to be confused with Mbps
"""

import time
import logging
import urllib.request

SizeUrlList = [
    [5, "https://sabnzbd.org/tests/internetspeed/5MB.bin"],
    [10, "https://sabnzbd.org/tests/internetspeed/10MB.bin"],
    [20, "https://sabnzbd.org/tests/internetspeed/20MB.bin"],
]


def measurespeed(url):
    """ Download the specified url, and report back MB/s (as a float) """
    logging.debug("URL is %s" % url)
    start = time.time()
    downloadedbytes = 0  # default
    try:
        req = urllib.request.Request(url, data=None, headers={"User-Agent": "Mozilla/5.0 (Macintosh)"})
        downloadedbytes = len(urllib.request.urlopen(req, timeout=4).read())
    except:
        # No connection at all?
        pass

    duration = time.time() - start

    logging.debug("Downloaded bytes: %d" % downloadedbytes)
    if downloadedbytes == 0:
        return 0

    logging.debug("Duration in seconds: %f" % duration)
    MBps = (downloadedbytes / 1000111) / duration  # Bytes
    return MBps


def BytestoBits(MBps):
    return 8.05 * MBps  # bits


def internetspeed():
    """ Report Internet speed in MB/s as a float """
    # Do basic test with a small download
    logging.debug("Basic measurement, with small download:")
    urlbasic = SizeUrlList[0][1]  # get first URL, which is smallest download
    baseMBps = measurespeed(urlbasic)
    logging.debug("Speed in MB/s: %.2f" % baseMBps)
    if baseMBps == 0:
        # no Internet connection, or other problem
        return baseMBps

    """
    Based on this first, small download, do a bigger download; the biggest download that still fits in 10 seconds
    Rationale: a bigger download could yield higher MB/s because the 'starting delay' is relatively less
    Calculation as example: 
    If the 5MB download took 0.3 seconds, you can do a 30 times bigger download, so about 150 MB, will round to 100 MB
    """

    # Calculate:
    maxtime = 4  # seconds
    URLtoDO = None
    for size, sizeurl in SizeUrlList:
        expectedtime = size / baseMBps
        if expectedtime < maxtime:
            # ok, this one is feasible, so keep it in mind
            URLtoDO = sizeurl

    maxMBps = baseMBps
    # Execute it twice, and get the best result
    for i in range(2):
        if URLtoDO:
            logging.debug(URLtoDO)
            MBps = measurespeed(URLtoDO)
            logging.debug("Speed in MB/s: %.2f" % MBps)
            maxMBps = max(maxMBps, MBps)

    return maxMBps


############################################

############### MAIN #######################

if __name__ == "__main__":
    logging.basicConfig(level=logging.DEBUG)
    logging.debug("Log level is DEBUG")
    print("Starting speed test:")
    maxMBps = internetspeed()
    print("Speed in MB/s: %.2f" % maxMBps)
    print("Speed in Mbps: %.2f" % BytestoBits(maxMBps))
