ykcheungab
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
"""Assemble and build the Nuitka wheel with setuptools."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from paths import sdk_dir
|
|
|
|
|
|
def write_wheel_setup(
|
|
wheel_dir: Path,
|
|
*,
|
|
version: str,
|
|
package_data: dict[str, list[str]],
|
|
install_requires: list[str],
|
|
extras_require: dict[str, list[str]],
|
|
) -> Path:
|
|
setup_py_path = wheel_dir / "setup.py"
|
|
setup_content = f"""
|
|
from setuptools import setup, Distribution
|
|
|
|
class BinaryDistribution(Distribution):
|
|
def has_ext_modules(self):
|
|
return True
|
|
|
|
setup(
|
|
name="orisys",
|
|
version="{version}",
|
|
description="Orisys - Optical Tactile Sensor SDK (Nuitka Compiled)",
|
|
author="Orisys",
|
|
packages=['orisys', 'orisys.examples', 'orisys_blend'],
|
|
package_dir={{'': 'wheel_root'}},
|
|
package_data={package_data},
|
|
include_package_data=True,
|
|
install_requires={install_requires!r},
|
|
extras_require={extras_require!r},
|
|
distclass=BinaryDistribution,
|
|
zip_safe=False,
|
|
)
|
|
"""
|
|
setup_py_path.write_text(setup_content, encoding="utf-8")
|
|
return setup_py_path
|
|
|
|
|
|
def run_bdist_wheel(wheel_dir: Path) -> Path | None:
|
|
print("\n=== Building Wheel ===")
|
|
subprocess.check_call([sys.executable, "setup.py", "bdist_wheel"], cwd=wheel_dir)
|
|
wheels = sorted((wheel_dir / "dist").glob("*.whl"))
|
|
return wheels[0] if wheels else None
|
|
|
|
|
|
def publish_wheel_to_sdk(wheel_path: Path, version: str) -> Path:
|
|
sdk_version_dir = sdk_dir(version)
|
|
sdk_version_dir.mkdir(parents=True, exist_ok=True)
|
|
dest_wheel = sdk_version_dir / wheel_path.name
|
|
if dest_wheel.exists():
|
|
dest_wheel.unlink()
|
|
shutil.move(str(wheel_path), str(dest_wheel))
|
|
return dest_wheel
|