networks lab wget

run_tests.py

import os
import re


def stats(url):
    os.system('wget ' + url + ' -o logfile')
    s = open('logfile', 'r').read()

    def speed(s):
        a = re.findall('\(.*/s\)', s)

        def mb2kb(s):
            if s[-4] == 'K':
                return float(s[:-5])
            else:
                return 1024 * float(s[:-5])
        return mb2kb(a[0][1:-1])

    def time(s):
        a = re.findall('=.*s', s)
        return float(a[0][1:-1])
    return speed(s), time(s)

print(stats('localhost:8000/512kb'))