Fix wheel mod

robotpy · Feb 4, 2025 · 581fc35 · 581fc35
1 parent b5ccbd8
commit 581fc35
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
@@ -76,7 +76,7 @@ jobs:
       name: Modify ${{ matrix.package }} wheel
       shell: bash
       run: |
-        /build/venv/bin/cross-python whl_mod.py ${{ steps.build.outputs.wheel }}
+        /build/venv/bin/cross-python whl_mod.py ${{ matrix.package }} ${{ steps.build.outputs.wheel }}
         ls dist/*
 
     # Some packages are stupid

diff --git a/whl_mod.py b/whl_mod.py
@@ -103,6 +103,7 @@ def add_requirements_to_wheel(
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
+    parser.add_argument("project")
     parser.add_argument("wheel")
     parser.add_argument("--config", default="packages.toml")
 
@@ -111,13 +112,13 @@ def add_requirements_to_wheel(
     with open(args.config, "rb") as fp:
         cfg = tomllib.load(fp)
 
-    project, _ = os.path.basename(args.wheel).split("-", 1)
-
     try:
-        pkgdata = cfg["packages"][project]
+        pkgdata = cfg["packages"][args.project]
         version = pkgdata["version"]
     except KeyError:
-        parser.error(f"{project} not found in {args.config}")
+        parser.error(f"{args.project} not found in {args.config}")
+
+    project = pkgdata.get("name", args.project)
 
     out_version = pkgdata.get("mod_version", None)
     if out_version is None: