cyrmax/sku-updater
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Changes to fix bugs

Browse Source
This commit is contained in:
Kirill Belousov 2023-02-23 13:15:17 +03:00
parent 48c4b0f5fd
commit 2d6625c52a
2 changed files with 128 additions and 128 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
build-requirements-nuitka.txt
View File

@ -1,3 +1,3 @@
nuitka==1.2.4 nuitka==1.4.8
ordered-set==4.1.0 ordered-set==4.1.0
zstandard==0.19.0 zstandard==0.20.0

250
sku-updater.py
View File

@ -1,125 +1,125 @@
import os import os
import pathlib import pathlib
import re import re
import sys import sys
import winreg import winreg
import zipfile import zipfile
from bs4 import BeautifulSoup as Soup from bs4 import BeautifulSoup as Soup
import requests import requests
SKU_URL = "https://duugu.github.io/Sku" SKU_URL = "https://duugu.github.io/Sku"
def confirmed_exit(code: int): def confirmed_exit(code: int):
print("Press enter to exit program") print("Press enter to exit program")
input() input()
sys.exit(code) sys.exit(code)
def find_wowc() -> str: def find_wowc() -> str:
key = winreg.OpenKey( key = winreg.OpenKey(
winreg.HKEY_LOCAL_MACHINE, winreg.HKEY_LOCAL_MACHINE,
r"SOFTWARE\WOW6432Node\Blizzard Entertainment\World of Warcraft", r"SOFTWARE\WOW6432Node\Blizzard Entertainment\World of Warcraft",
) )
value = winreg.QueryValueEx(key, "InstallPath") value = winreg.QueryValueEx(key, "InstallPath")
return value[0] return value[0]
def get_sku_version(sku_path: pathlib.Path) -> float: def get_sku_version(sku_path: pathlib.Path) -> float:
changelog_path = sku_path / "CHANGELOG.md" changelog_path = sku_path / "CHANGELOG.md"
with changelog_path.open("r") as f: with changelog_path.open("r") as f:
txt = f.read() txt = f.read()
version_match = re.search(r"^\# Sku \((\d+\.\d+)\)", txt) version_match = re.search(r"^\# Sku \(((\d+\.\d+)|(\d+))\)", txt)
if not version_match: if not version_match:
print("Unable to determine Sku version") print("Unable to determine Sku version")
confirmed_exit(1) confirmed_exit(1)
try: try:
version = float(version_match.group(1)) version = float(version_match.group(1))
except ValueError: except ValueError:
print("Unable to determine Sku version") print("Unable to determine Sku version")
confirmed_exit(1) confirmed_exit(1)
return version return version
def fetch_sku_version() -> tuple[float, str]: def fetch_sku_version() -> tuple[float, str]:
rre = re.compile( rre = re.compile(
r"^https://github.com/Duugu/Sku/releases/download/r\d+\.\d+/Sku-r\d+\.\d+-.+\.zip$", r"^https://github.com/Duugu/Sku/releases/download/r(\d+\.\d+)|(\d+)/Sku-r(\d+\.\d+)|(\d+)-.+\.zip$",
re.I, re.I,
) )
r = requests.get(SKU_URL) r = requests.get(SKU_URL)
page = Soup(r.text, features="html.parser") page = Soup(r.text, features="html.parser")
links = page.findAll("a", attrs={"href": rre}) links = page.findAll("a", attrs={"href": rre})
if len(links) < 1: if len(links) < 1:
print("Unable to fetch latest Sku version info") print("Unable to fetch latest Sku version info")
confirmed_exit(1) confirmed_exit(1)
href = links[0].get("href") href = links[0].get("href")
version_match = re.search( version_match = re.search(
r"^https://github.com/Duugu/Sku/releases/download/r\d+\.\d+/Sku-r(\d+\.\d+)-.+\.zip$", r"^https://github.com/Duugu/Sku/releases/download/r(\d+\.\d+)|(\d+)/Sku-r((\d+\.\d+)|(\d+))-.+\.zip$",
href, href,
) )
if not version_match: if not version_match:
print("Unable to fetch latest Sku version") print("Unable to fetch latest Sku version")
confirmed_exit(1) confirmed_exit(1)
version = float(version_match.group(1)) version = float(version_match.group(1))
return (version, href) return (version, href)
def update_sku(sku_info: tuple[float, str], sku_path: pathlib.Path): def update_sku(sku_info: tuple[float, str], sku_path: pathlib.Path):
url = sku_info[1] url = sku_info[1]
local_filename = url.split("/")[-1] local_filename = url.split("/")[-1]
print("Downloading... ") print("Downloading... ")
with requests.get(url, stream=True) as r: with requests.get(url, stream=True) as r:
r.raise_for_status() r.raise_for_status()
with open(local_filename, "wb") as f: with open(local_filename, "wb") as f:
for chunk in r.iter_content(4096): for chunk in r.iter_content(4096):
f.write(chunk) f.write(chunk)
print("Installing...") print("Installing...")
with zipfile.ZipFile(local_filename, "r") as zf: with zipfile.ZipFile(local_filename, "r") as zf:
zf.extractall(str(sku_path.parent.resolve())) zf.extractall(str(sku_path.parent.resolve()))
print("Cleaning...") print("Cleaning...")
os.remove(local_filename) os.remove(local_filename)
def main(): def main():
print("Searching World of Warcraft Classic installation...") print("Searching World of Warcraft Classic installation...")
try: try:
wowc_path = find_wowc() wowc_path = find_wowc()
except FileNotFoundError: except FileNotFoundError:
print("Unable to find World of Warcraft Classic installation.") print("Unable to find World of Warcraft Classic installation.")
confirmed_exit(1) confirmed_exit(1)
print(f"Found WoW Classic at path {wowc_path}") print(f"Found WoW Classic at path {wowc_path}")
sku_path = pathlib.Path(wowc_path) / "Interface" / "AddOns" / "Sku" sku_path = pathlib.Path(wowc_path) / "Interface" / "AddOns" / "Sku"
if not sku_path.exists(): if not sku_path.exists():
print("Couldn't find Sku folder. Check your installation") print("Couldn't find Sku folder. Check your installation")
confirmed_exit(1) confirmed_exit(1)
print(f"Found Sku at path {str(sku_path)}") print(f"Found Sku at path {str(sku_path)}")
sku_version = get_sku_version(sku_path) sku_version = get_sku_version(sku_path)
print(f"Current Sku version is {sku_version}") print(f"Current Sku version is {sku_version}")
print("Checking for updates...") print("Checking for updates...")
info = fetch_sku_version() info = fetch_sku_version()
print(f"Latest available Sku version is {info[0]}") print(f"Latest available Sku version is {info[0]}")
if info[0] <= sku_version: if info[0] <= sku_version:
print("Your version of Sku is equal or newer than latest available. Exiting...") print("Your version of Sku is equal or newer than latest available. Exiting...")
confirmed_exit(0) confirmed_exit(0)
answer = input( answer = input(
"Do you want to update to the latest version? y - yes, n or other letter - no: " "Do you want to update to the latest version? y - yes, n or other letter - no: "
) )
if answer != "y": if answer != "y":
print("Ok, not updating.") print("Ok, not updating.")
confirmed_exit(0) confirmed_exit(0)
print("Updating...") print("Updating...")
update_sku(info, sku_path) update_sku(info, sku_path)
print("Verifying update...") print("Verifying update...")
new_version = get_sku_version(sku_path) new_version = get_sku_version(sku_path)
if sku_version == new_version: if sku_version == new_version:
print("Verification failed!") print("Verification failed!")
confirmed_exit(1) confirmed_exit(1)
print("Update complete!") print("Update complete!")
print(f"Sku updated from {sku_version} to {new_version}") print(f"Sku updated from {sku_version} to {new_version}")
confirmed_exit(0) confirmed_exit(0)
if __name__ == "__main__": if __name__ == "__main__":
main() main()