0.1.6s

updater/config.json

@@ -0,0 +1,5 @@
+{
+    "host":"https://git.nekono.su",
+    "user":"Nyako",
+    "repo":"Local-Booru"
+}

updater/run.py

@@ -0,0 +1,59 @@
+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!")