install460

import socket
import os.path
import os
import zipfile
import urllib.request
import requests
import queue
import math
from winreg import *
queue.Queue()
class Download:
        def download_file(self,url,basepath):
                procenta = 0
                procentastare = 0
                local_filename = os.path.abspath(os.path.join(basepath, "instalace460TEMP.zip"))
                r = requests.get(url, stream=True)
                with open(local_filename, 'wb') as f:
                        for chunk in r.iter_content(chunk_size=1024):
                                if chunk:
                                        f.write(chunk)
                                        if math.floor(procenta) != procentastare:
                                                procentastare = math.floor(procenta)
                                                print("Staženo cca. "+str(math.floor(procenta))+"%")
                                        procenta = procenta + 0.00066453226345023
                return local_filename
class Main:
        def __init__(self):
                try:
                        root_key=OpenKey(HKEY_CURRENT_USER, 'Software\Valve\Steam', 0, KEY_READ)
                        [Pathname,regtype]=(QueryValueEx(root_key,"SourceModInstallPath"))
                        CloseKey(root_key)
                except:
                        input("Steam nebyl nalezen na vašem počítači! Kliknutím na libovolnou klávesu ukončíte aplikaci!")
                else:
                        StAppsPath = Pathname[:Pathname.find("steamapps")+10]
                        RWpath = os.path.join(StAppsPath,"common","RailWorks")
                        if os.path.isdir(os.path.abspath(RWpath)):
                                print("Navazuji spojení se serverem...")
                                download = Download()
                                try:
                                        print("Spojení navázáno. Stahuji!")
                                        download.download_file("http://jachyhm.cz/460autoaktualizace.rwp",RWpath)
                                except:
                                        print("Chyba při stahování! Nelze navázat spojení se serverem!")
                                else:
                                        print("Úspěšně staženo, rozbaluji...")
                                        with zipfile.ZipFile(os.path.abspath(os.path.join(RWpath, "instalace460TEMP.zip")),"r") as zip_ref:
                                                zip_ref.extractall(os.path.abspath(RWpath))
                                        os.remove(os.path.abspath(os.path.join(RWpath,"instalace460TEMP.zip")))
                                        input("ČD460 nainstalována, klinutím na libovolnou klávesu ukončíte aplikaci!")
                        else:
                                print("Hra RailWorks nebyla v platformě Steam nalezena! Nejdříve si hru zakupte!")
                                input("Kliknutím na libovolnou klávesu ukončíte aplikaci!")
print("Instalace addonu CD460")
m=Main()