Hell13Cat
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
import os
|
|
import json
|
|
import shutil
|
|
import tempfile
|
|
import zipfile
|
|
import urllib.request
|
|
|
|
CONFIG_PATH = os.path.join(os.path.dirname(__file__), "config.json")
|
|
|
|
EXCLUDE_PATHS = {
|
|
"data",
|
|
"config.json",
|
|
"logs.log",
|
|
"updater"
|
|
}
|
|
|
|
def load_config():
|
|
with open(CONFIG_PATH, "r", encoding="utf-8") as f:
|
|
return json.load(f)
|
|
|
|
def get_latest_release(cfg):
|
|
url = f"{cfg['host']}/api/v1/repos/{cfg['user']}/{cfg['repo']}/releases/latest"
|
|
with urllib.request.urlopen(url) as r:
|
|
return json.loads(r.read().decode("utf-8"))
|
|
|
|
def download(url, path):
|
|
urllib.request.urlretrieve(url, path)
|
|
|
|
def should_exclude(relative_path: str) -> bool:
|
|
parts = set(relative_path.split(os.sep))
|
|
return any(ex in parts for ex in EXCLUDE_PATHS)
|
|
|
|
def apply_update(src_root, target_root):
|
|
for root, dirs, files in os.walk(src_root):
|
|
rel_dir = os.path.relpath(root, src_root)
|
|
if rel_dir == ".":
|
|
rel_dir = ""
|
|
for file in files:
|
|
rel_path = os.path.normpath(os.path.join(rel_dir, file))
|
|
if should_exclude(rel_path):
|
|
continue
|
|
src_file = os.path.join(root, file)
|
|
dst_file = os.path.join(target_root, rel_path)
|
|
os.makedirs(os.path.dirname(dst_file), exist_ok=True)
|
|
shutil.copy2(src_file, dst_file)
|
|
|
|
cfg = load_config()
|
|
release = get_latest_release(cfg)
|
|
zip_url = release["zipball_url"]
|
|
tmp = tempfile.mkdtemp()
|
|
zip_path = os.path.join(tmp, "release.zip")
|
|
print("Downloading:", release["tag_name"])
|
|
download(zip_url, zip_path)
|
|
with zipfile.ZipFile(zip_path, "r") as z:
|
|
z.extractall(tmp)
|
|
extracted_root = os.path.join(tmp, os.listdir(tmp)[0])
|
|
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
|
|
apply_update(extracted_root, project_root)
|
|
print("Update success!")
|