Added to workflow

Author: evertlammerts

.github/workflows/cleanup_pypi.yml

@@ -0,0 +1,31 @@
+name: cleanup_pypi.yml
+on:
+  workflow_dispatch:
+    inputs:
+      dry-run:
+        description: List packages that would be deleted but don't delete them
+        type: boolean
+        default: false
+  workflow_call:
+jobs:
+  cleanup_pypi:
+    name: Remove Nightlies from PyPI
+    runs-on: ubuntu-latest
+    env:
+      PYPI_CLEANUP_PASSWORD: ${{secrets.PYPI_CLEANUP_PASSWORD}}
+      PYPI_CLEANUP_OTP: ${{secrets.PYPI_CLEANUP_OTP}}
+    steps:
+      - uses: actions/checkout@v4
+      - name: Install Astral UV
+        uses: astral-sh/setup-uv@v6
+        with:
+          version: "0.7.14"
+          python-version: 3.11
+
+      - name: Run Cleanup
+        run: |
+          uv sync --only-group pypi --no-install-project
+          uv run --no-sync -s scripts/pypi_cleanup.py ${{ inputs.dry-run && '--dry' || '' }} \
+            --index-hostname "${{ vars.PYPI_HOST }}" \
+            --username "${{ vars.PYPI_CLEANUP_USERNAME }}" \
+            --max-nightlies ${{ vars.PYPI_MAX_NIGHTLIES }}

pyproject.toml

@@ -250,7 +250,6 @@ test = [ # dependencies used for running tests
     "numpy<2; python_version < '3.12'",
     "numpy>=2; python_version >= '3.12'",
 ]
-
 scripts = [ # dependencies used for running scripts
     "cxxheaderparser",
     "ipython",
@@ -260,9 +259,12 @@ scripts = [ # dependencies used for running scripts
     "pcpp",
     "polars",
     "pyarrow",
-    "pyotp>=2.9.0",
     "pytz"
 ]
+pypi = [ # dependencies used by the pypi cleanup script
+    "pyotp>=2.9.0",
+    "requests>=2.32.4",
+]
 build = [
     "cmake>=3.29.0",
     "ninja>=1.10",

scripts/pypi_cleanup.py

@@ -41,9 +41,9 @@ def non_empty_string(value):
     epilog="Environment variables required (unless --dry): PYPI_CLEANUP_PASSWORD, PYPI_CLEANUP_OTP"
 )
 parser.add_argument("--dry", action="store_true", help="Show what would be deleted but don't actually do it")
-parser.add_argument("--index-hostname", type=valid_hostname, required=True, help="Index hostname (required)")
-parser.add_argument("--max-nightlies", type=int, default=2, help="Max number of nightlies of unreleased versions (default=2)")
-parser.add_argument("--username", type=non_empty_string, help="Username (required unless --dry)")
+parser.add_argument("-i", "--index-hostname", type=valid_hostname, required=True, help="Index hostname (required)")
+parser.add_argument("-m", "--max-nightlies", type=int, default=2, help="Max number of nightlies of unreleased versions (default=2)")
+parser.add_argument("-u", "--username", type=non_empty_string, help="Username (required unless --dry)")
 args = parser.parse_args()
 
 # Handle secrets from environment variables