try to generate matrix in python

Author: masklinn

.github/workflows/pyo3-wheels.yml

@@ -14,55 +14,63 @@ permissions:
   contents: read
 
 jobs:
+  py-wheels-matrix:
+    name: "generate build matrix"
+    runs-on: ubuntu-latest
+    outputs:
+      matrix: ${{ steps.make-matrix.outputs.matrix }}
+    steps:
+      - id: make-matrix
+        shell: python
+        name: generate matrix
+        run: |
+          import itertools
+          import json
+          import os
+          import pprint
+
+          matrix = [
+              d
+              for d in map(dict, itertools.product(
+                  (('python-version', v) for v in ["3.x", "pypy-3.10", "pypyp-3.11", "graalpy-24"]),
+                  (('arch', a) for a in ["x64_64", "aarch64"]),
+                  (('platform', p) for p in ["linux", "musllinux", "windows", "macos"])
+              ))
+              if not (d['platform'] == 'windows' and d['python-version'] == 'graalpy-24')
+          ]
+          for job in matrix:
+              match job['platform']:
+                  case 'linux':
+                    job['manylinux'] = 'auto'
+                    job['args'] = '--release --out dist -m ua-parser-py/Cargo.toml -i python --zig'
+                  case 'mussllinux':
+                    job['manylinux'] = 'musllinux_1_2'
+                    job['args'] = '--release --out dist -m ua-parser-py/Cargo.toml'
+                  case _:
+                    job['args'] = '--release --out dist -m ua-parser-py/Cargo.toml -i python'
+
+              match job['platform']:
+                case 'windows':
+                  job['runs-on'] =  'windows-latest'
+                case 'macos':
+                  job['runs-on'] = 'macos-latest'
+                case _:
+                  job['runs-on'] = 'ubuntu-latest'
+
+              if job['platform'] == 'windows' and job['arch'] == 'aarch64':
+                job['arch'] = 'arm64'
+
+          print(pprint.pprint(matrix))
+          with open(os.environ['GITHUB_OUTPUT'], 'w') as f:
+            print("matrix=" + json.dumps('matrix'), file=f)
+  
   py-release-wheels:
+    needs: [py-wheels-matrix]
     strategy:
       fail-fast: false
-      matrix:
-        python-version:
-          - "3.x"
-          - "pypy-3.10"
-          - "pypy-3.11"
-          - "graalpy-24"
-        aarch:
-          - x86_64
-          - aarch64
-        platform:
-          - linux
-          - musllinux
-          - windows
-          - macos
-
-        exclude:
-          - platform: windows
-            python-version: graalpy-24
-
-        include:
-          - platform: linux
-            manylinux: auto
-          - platform: musllinux
-            manylinux: musllinux_1_2
-
-          - args: --release --out dist -m ua-parser-py/Cargo.toml -i python
-          - platform: linux
-            args: --release --out dist -m ua-parser-py/Cargo.toml -i python --zig
-          - platform: musllinux
-            args: --release --out dist -m ua-parser-py/Cargo.toml
-
-          - runs: ubuntu-latest
-          - platform: windows
-            runs: windows-latest
-          - platform: macos
-            runs: macos-latest
-          
-          - aarch: x64_64
-            arch: x86_64
-          - aarch: aarch64
-            arch: aarch64
-          - platform: windows
-            aarch: aarch64
-            arch: arm64
-
-    runs-on: ${{ matrix.runs || matrix }}
+      matrix: ${{fromJson(needs.py-wheels-matrix.outputs.matrix)}}
+        
+    runs-on: ${{ matrix.runs }}
 
     steps:
       - uses: actions/checkout@v4