Merge wheels

.github/workflows/build.yml

@@ -40,3 +40,29 @@ jobs:
        name: ${{ matrix.name }}
        path: dist
        if-no-files-found: error
+  merge-wheels:
+    runs-on: ubuntu-22.04
+    needs: [build]
+    steps:
+    - uses: actions/checkout@v4
+    - uses: actions/download-artifact@v4
+      with:
+        name: linux
+        path: wheels/linux
+    - uses: actions/download-artifact@v4
+      with:
+        name: darwin
+        path: wheels/darwin
+    - uses: actions/download-artifact@v4
+      with:
+        name: windows
+        path: wheels/windows
+    - uses: actions/setup-python@v5
+      with:
+        python-version: '3.10'
+    - run: python merge_wheels.py
+    - uses: actions/upload-artifact@v4
+      with:
+       name: merged
+       path: wheels/merged
+       if-no-files-found: error

merge_wheels.py

@@ -5,14 +5,14 @@
 from shutil import make_archive, unpack_archive
 
 root = Path(__file__).parent
-merge = root / 'merge'
-merged = root / 'merged'
+temp = root / 'wheels' / 'temp'
+merged = root / 'wheels' / 'merged'
 
 
-if merge.exists():
-    shutil.rmtree(merge)
+if temp.exists():
+    shutil.rmtree(temp)
 
-os.makedirs(merge)
+os.makedirs(temp)
 
 if merged.exists():
     shutil.rmtree(merged)
@@ -21,7 +21,7 @@
 
 wheels = []
 
-for dirpath, dirnames, filenames in os.walk(root):
+for dirpath, _, filenames in os.walk(root / 'wheels'):
     for filename in filenames:
         if filename.endswith('.whl'):
             wheels.append(root / dirpath / filename)
@@ -35,15 +35,15 @@
                 record = info.filename
                 break
         records.update(archive.read(record).split(b'\n'))
-        archive.extractall(merge)
+        archive.extractall(temp)
 
-    unpack_archive(wheel, merge, 'zip')
+    unpack_archive(wheel, temp, 'zip')
 
-with open(merge / record, 'w') as record:
+with open(temp / record, 'w') as record:
     lines = [line.decode('utf-8') for line in records]
     lines.sort()
     lines = list(filter(lambda line: line != "", lines))
     record.writelines('\n'.join(lines))
 
-make_archive(str(merged / wheels[0].stem), 'zip', merge)
+make_archive(str(merged / wheels[0].stem), 'zip', temp)
 os.rename(merged / (wheels[0].stem + '.zip'), merged / wheels[0].name)