import os
from subprocess import PIPE, Popen
import ctypes
import ssl
import urllib2
import datetime
import time
import shutil
import zipfile

class disable_file_system_redirection:
    _disable = ctypes.windll.kernel32.Wow64DisableWow64FsRedirection
    _revert = ctypes.windll.kernel32.Wow64RevertWow64FsRedirection
    def __enter__(self):
        self.old_value = ctypes.c_long()
        self.success = self._disable(ctypes.byref(self.old_value))
    def __exit__(self, type, value, traceback):
        if self.success:
            self._revert(self.old_value)

url = r"https://script-downloads.itarian.com/ookla-speedtest/ookla-speedtest-1.2.0-win64.zip"
    
Down_path=os.environ['TEMP'] + r"\speedtest_files"

if not os.path.exists(Down_path):
    os.makedirs(Down_path)
    
fileName = url.split('/')[-1]
DownTo = os.path.join(Down_path, fileName)

def ecmd(command):   
    with disable_file_system_redirection():
        obj = Popen(command, shell = True, stdout = PIPE, stderr = PIPE)
    out, err = obj.communicate()
    ret=obj.returncode
    if ret==0:
        if out:
            return(out.strip())
        else:
            return("couldn't get the speedtest output. returncode: %s"%(ret))
    else:
        if err:
            return(err.strip())
        else:
            return("something went wrong while testing internet speed. returncode: %s"%(ret))

def downloadFile(DownTo, fromURL):
    headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/74.0.3729.169 Safari/537.36'}
    context = ssl._create_unverified_context()
    request = urllib2.Request(fromURL, headers=headers)
    req = urllib2.urlopen(request,context=context)
    try:
        with open(DownTo, 'wb') as f:
            while True:
                chunk = req.read(100*1000*1000)
                if chunk:
                    f.write(chunk)
                else:
                    break
        if os.path.isfile(DownTo):
            return '{} - {}KB'.format(DownTo, os.path.getsize(DownTo)/1024)
		
    except:
        return 'Please Check URL or Download Path!'
        


def run():
    with disable_file_system_redirection():
        print(downloadFile(DownTo, url))
        with zipfile.ZipFile(DownTo, "r") as z:
            z.extractall(Down_path)
        print(ecmd('"%s\speedtest.exe" --accept-license --accept-gdpr --progress=no'%(Down_path)))
        shutil.rmtree(Down_path)

run()