76 add type annotations in the python bindings (#86)

* Add python stub generation in CI.
* Switch to a pure rust binding with a single module to avoid build weird issue.

Author: pierre-rouanet

.github/workflows/python-upload.yml

@@ -3,9 +3,17 @@
 #
 #    maturin generate-ci github
 #
-name: python-bindings-build-ci
+name: python-bindings
 
-on: [pull_request]
+on:
+  push:
+    branches:
+      - main
+      - master
+    tags:
+      - '*'
+  pull_request:
+  workflow_dispatch:
 
 permissions:
   contents: read
@@ -29,13 +37,21 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: 3.x
+      - name: Generate pyo3 stub
+        run: |
+          cargo run --bin stub_gen --features python 
       - name: Build wheels
         uses: PyO3/maturin-action@v1
         with:
           target: ${{ matrix.platform.target }}
-          args: --release --features python --out dist --find-interpreter
+          args: --release --features python --features pyo3/extension-module --out dist --find-interpreter
           sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
           manylinux: auto
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-linux-${{ matrix.platform.target }}
+          path: dist
 
   musllinux:
     runs-on: ${{ matrix.platform.runner }}
@@ -55,13 +71,21 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: 3.x
+      - name: Generate pyo3 stub
+        run: |
+          cargo run --bin stub_gen --features python 
       - name: Build wheels
         uses: PyO3/maturin-action@v1
         with:
           target: ${{ matrix.platform.target }}
-          args: --release --features python --out dist --find-interpreter
+          args: --release --features python --features pyo3/extension-module --out dist --find-interpreter
           sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
           manylinux: musllinux_1_2
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-musllinux-${{ matrix.platform.target }}
+          path: dist
 
   windows:
     runs-on: ${{ matrix.platform.runner }}
@@ -82,8 +106,13 @@ jobs:
         uses: PyO3/maturin-action@v1
         with:
           target: ${{ matrix.platform.target }}
-          args: --release --features python --out dist --find-interpreter
+          args: --release --features python --features pyo3/extension-module --out dist --find-interpreter
           sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-windows-${{ matrix.platform.target }}
+          path: dist
 
   macos:
     runs-on: ${{ matrix.platform.runner }}
@@ -99,19 +128,62 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: 3.x
+      - name: Generate pyo3 stub
+        run: |
+          cargo run --bin stub_gen --features python 
       - name: Build wheels
         uses: PyO3/maturin-action@v1
         with:
           target: ${{ matrix.platform.target }}
-          args: --release --features python --out dist --find-interpreter
+          args: --release --features python --features pyo3/extension-module --out dist --find-interpreter
           sccache: ${{ !startsWith(github.ref, 'refs/tags/') }}
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-macos-${{ matrix.platform.target }}
+          path: dist
 
   sdist:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
+      - name: Generate pyo3 stub
+        run: |
+          cargo run --bin stub_gen --features python 
       - name: Build sdist
         uses: PyO3/maturin-action@v1
         with:
           command: sdist
           args: --out dist
+      - name: Upload sdist
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-sdist
+          path: dist
+
+  release:
+    name: Release
+    runs-on: ubuntu-latest
+    if: ${{ startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' }}
+    needs: [linux, musllinux, windows, macos, sdist]
+    permissions:
+      # Use to sign the release artifacts
+      id-token: write
+      # Used to upload release artifacts
+      contents: write
+      # Used to generate artifact attestation
+      attestations: write
+    steps:
+      - uses: actions/download-artifact@v4
+      - name: Generate artifact attestation
+        uses: actions/attest-build-provenance@v1
+        with:
+          subject-path: 'wheels-*/*'
+      - name: Publish to PyPI
+        if: ${{ startsWith(github.ref, 'refs/tags/') }}
+        uses: PyO3/maturin-action@v1
+        env:
+          MATURIN_PYPI_TOKEN: ${{ secrets.PYPI_API_TOKEN }}
+        with:
+          command: upload
+          args: --non-interactive --skip-existing wheels-*/*

.github/workflows/python.yml

@@ -16,7 +16,7 @@ crate-type = ["lib", "cdylib"]
 
 [features]
 default = []
-python = ["dep:pyo3", "dep:pyo3-log"]
+python = ["dep:pyo3", "dep:pyo3-log", "dep:pyo3-stub-gen"]
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
@@ -28,8 +28,11 @@ clap = { version = "4.0.32", features = ["derive"] }
 proc-macro2 = { version = "1.0", features=["default", "proc-macro"] }
 signal-hook = "0.3.4"
 num_enum = "0.7.3"
-pyo3 = { version = "0.24.1", optional = true, features = ["multiple-pymethods", "extension-module"] }
+pyo3 = { version = "0.24.1", optional = true, features = ["multiple-pymethods"] }
 pyo3-log = { version = "0.12.3", optional = true }
-
-[dev-dependencies]
+pyo3-stub-gen = { version = "0.9.0", optional = true }
 env_logger = "0.10.0"
+
+[[bin]]
+name = "stub_gen"
+required-features = ["python"]