Recompose windows wheels: adapt to 'wheel/0.46.1' version

Author: howetuft

build-helpers/wheels/recompose_wheel_windows.py

@@ -2,17 +2,52 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-import os
+"""This script makes a final rearrangement to a Windows wheel.
+
+2 actions to be done:
+- Rename and move oidnDenoise.exe
+- Rename and move OpenImageDenoise_device_cpu.dll
+"""
+
 from pathlib import Path
 import argparse
 import tempfile
 import shutil
+import subprocess
 
 from wheel.wheelfile import WheelFile
-from wheel.cli.unpack import unpack
-from wheel.cli.pack import pack
+
+
+def unpack(path, dest):
+    """Unpack a wheel."""
+    args = [
+        "python",
+        "-m",
+        "wheel",
+        "unpack",
+        "-d",
+        str(dest),
+        str(path),
+    ]
+    print(subprocess.check_output(args, text=True))
+
+
+def pack(directory, dest_dir):
+    """(Re)pack a wheel."""
+    args = [
+        "python",
+        "-m",
+        "wheel",
+        "pack",
+        "-d",
+        str(dest_dir),
+        str(directory),
+    ]
+    print(subprocess.check_output(args, text=True))
+
 
 def main():
+    """Main entry point."""
     # Parse arguments
     parser = argparse.ArgumentParser()
     parser.add_argument("wheelpath", type=Path)
@@ -26,8 +61,8 @@ def main():
     with tempfile.TemporaryDirectory() as tmpdir:  # Working space
         # Unpack wheel
         unpack(path=args.wheelpath, dest=tmpdir)
-        with WheelFile(args.wheelpath) as wf:
-            namever = wf.parsed_filename.group("namever")
+        with WheelFile(args.wheelpath) as wheelfile:
+            namever = wheelfile.parsed_filename.group("namever")
             unpacked_wheel_path = Path(tmpdir) / namever
 
         # Rename and move oidnDenoise
@@ -38,17 +73,23 @@ def main():
         )
 
         # Rename and move OpenImageDenoise_device_cpu
-        print("Rename OpenImageDenoise_device_cpu.pyd into OpenImageDenoise_device_cpu.dll")
+        print(
+            "Rename OpenImageDenoise_device_cpu.pyd "
+            "into OpenImageDenoise_device_cpu.dll"
+        )
         shutil.move(
-            unpacked_wheel_path / "pyluxcore.libs" / "OpenImageDenoise_device_cpu.pyd",
-            unpacked_wheel_path / "pyluxcore.libs" / "OpenImageDenoise_device_cpu.dll",
+            unpacked_wheel_path
+            / "pyluxcore.libs"
+            / "OpenImageDenoise_device_cpu.pyd",
+            unpacked_wheel_path
+            / "pyluxcore.libs"
+            / "OpenImageDenoise_device_cpu.dll",
         )
 
         # Repack wheel
         pack(
             directory=unpacked_wheel_path,
             dest_dir=wheel_folder,
-            build_number=None
         )