Added experimental MYM application code, bumped to Python 3.12

commands/archive/theme.py

@@ -1,4 +1,58 @@
 # "commands/archive/theme.py" from WiiPy by NinjaCheetah
 # https://github.com/NinjaCheetah/WiiPy
 
+import configparser
+import pathlib
+import shutil
+import tempfile
+import zipfile
+import libWiiPy
+import time
 
+
+def handle_apply_mym(args):
+    mym_path = pathlib.Path(args.mym)
+    base_path = pathlib.Path(args.base)
+    output_path = pathlib.Path(args.output)
+
+    if not mym_path.exists():
+        raise FileNotFoundError(mym_path)
+    if not base_path.exists():
+        raise FileNotFoundError(base_path)
+    if output_path.suffix != ".csm":
+        output_path = output_path.with_suffix(".csm")
+
+    with tempfile.TemporaryDirectory() as tmp_dir:
+        tmp_path = pathlib.Path(tmp_dir)
+        # Extract the MYM file into the temp directory.
+        # MYM files are just ZIP files, so if zipfile doesn't accept it then it can't be valid.
+        try:
+            with zipfile.ZipFile(mym_path) as mym:
+                mym.extractall(tmp_path.joinpath("mym_out"))
+        except zipfile.BadZipfile:
+            print("Error: The provided MYM theme is not valid!")
+            exit(1)
+        mym_tmp_path = pathlib.Path(tmp_path.joinpath("mym_out"))
+        # Extract the asset archive into the temp directory.
+        try:
+            libWiiPy.archive.extract_u8(base_path.read_bytes(), str(tmp_path.joinpath("base_out")))
+        except ValueError:
+            print("Error: The provided base assets are not valid!")
+            exit(1)
+        base_temp_path = pathlib.Path(tmp_path.joinpath("base_out"))
+        # Parse the mym.ini file in the root of the extracted MYM file.
+        mym_ini = configparser.ConfigParser()
+        mym_ini.read(mym_tmp_path.joinpath("mym.ini"))
+        # Iterate over every key in the ini file and apply the theme based the source and target of each key.
+        for section in mym_ini.sections():
+            source_file = mym_tmp_path
+            for piece in mym_ini[section]["source"].replace("\\", "/").split("/"):
+                source_file = source_file.joinpath(piece)
+            target_file = base_temp_path
+            for piece in mym_ini[section]["file"].replace("\\", "/").split("/"):
+                target_file = target_file.joinpath(piece)
+            shutil.move(source_file, target_file)
+        # Repack the now-themed asset archive and write it out.
+        output_path.write_bytes(libWiiPy.archive.pack_u8(base_temp_path))
+
+    print(f"Applied theme \"{mym_path.name}\" to \"{output_path.name}\"!")