8d9a757ef5
- wheel_pipeline.py: CLI entry point for repack/rebuild flow - pipeline/detect.py: detect pure vs native extension wheels - pipeline/repack.py: repack pure-Python wheels with harmonyos_arm64 tag - pipeline/rebuild.py: native wheel rebuild entry (stub for OHOS NDK) - pipeline/cli.py: argument parsing and orchestration - Makefile: convenience targets (repack, rebuild, clean) - README.md: documentation with usage and limitations - examples/gen_test_wheels.py: test wheel generator - output/: output directory for generated wheels
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
import zipfile
|
|
import os
|
|
import json
|
|
|
|
EXAMPLES_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
|
|
def make_pure_wheel(name, version, out_dir):
|
|
dist_info = f"{name}-{version}.dist-info"
|
|
os.makedirs(f"/tmp/{dist_info}", exist_ok=True)
|
|
|
|
metadata = f"""Metadata-Version: 2.1
|
|
Name: {name}
|
|
Version: {version}
|
|
Summary: Test pure-Python wheel
|
|
"""
|
|
|
|
with open(f"/tmp/{dist_info}/METADATA", "w") as f:
|
|
f.write(metadata)
|
|
with open(f"/tmp/{dist_info}/WHEEL", "w") as f:
|
|
f.write("Wheel-Version: 1.0\nGenerator: test\nRoot-Is-Purelib: true\nTag: py3-none-any\n")
|
|
with open(f"/tmp/{dist_info}/RECORD", "w") as f:
|
|
f.write("")
|
|
|
|
wheel_name = f"{name}-{version}-py3-none-any.whl"
|
|
wheel_path = os.path.join(out_dir, wheel_name)
|
|
|
|
with zipfile.ZipFile(wheel_path, "w", zipfile.ZIP_DEFLATED) as zf:
|
|
zf.write(f"/tmp/{dist_info}/METADATA", f"{dist_info}/METADATA")
|
|
zf.write(f"/tmp/{dist_info}/WHEEL", f"{dist_info}/WHEEL")
|
|
zf.write(f"/tmp/{dist_info}/RECORD", f"{dist_info}/RECORD")
|
|
pkg_dir = f"{name}"
|
|
zf.writestr(f"{pkg_dir}/__init__.py", "def hello(): return 'hello'\n")
|
|
|
|
for f in os.listdir(f"/tmp/{dist_info}"):
|
|
os.remove(f"/tmp/{dist_info}/{f}")
|
|
os.rmdir(f"/tmp/{dist_info}")
|
|
|
|
return wheel_path
|
|
|
|
|
|
def make_native_wheel(name, version, out_dir):
|
|
dist_info = f"{name}-{version}.dist-info"
|
|
os.makedirs(f"/tmp/{dist_info}", exist_ok=True)
|
|
|
|
metadata = f"""Metadata-Version: 2.1
|
|
Name: {name}
|
|
Version: {version}
|
|
Summary: Test native wheel
|
|
"""
|
|
|
|
with open(f"/tmp/{dist_info}/METADATA", "w") as f:
|
|
f.write(metadata)
|
|
with open(f"/tmp/{dist_info}/WHEEL", "w") as f:
|
|
f.write("Wheel-Version: 1.0\nGenerator: test\nRoot-Is-Purelib: false\nTag: cp38-cp38-manylinux_2_17_x86_64\n")
|
|
with open(f"/tmp/{dist_info}/RECORD", "w") as f:
|
|
f.write("")
|
|
|
|
wheel_name = f"{name}-{version}-cp38-cp38-manylinux_2_17_x86_64.whl"
|
|
wheel_path = os.path.join(out_dir, wheel_name)
|
|
|
|
with zipfile.ZipFile(wheel_path, "w", zipfile.ZIP_DEFLATED) as zf:
|
|
zf.write(f"/tmp/{dist_info}/METADATA", f"{dist_info}/METADATA")
|
|
zf.write(f"/tmp/{dist_info}/WHEEL", f"{dist_info}/WHEEL")
|
|
zf.write(f"/tmp/{dist_info}/RECORD", f"{dist_info}/RECORD")
|
|
zf.writestr(f"{name}/__init__.py", "def hello(): return 'hello'\n")
|
|
zf.writestr(f"{name}/_native.cpython-38-x86_64-linux-gnu.so", b"\x7fELF...fake")
|
|
|
|
for f in os.listdir(f"/tmp/{dist_info}"):
|
|
os.remove(f"/tmp/{dist_info}/{f}")
|
|
os.rmdir(f"/tmp/{dist_info}")
|
|
|
|
return wheel_path
|
|
|
|
|
|
if __name__ == "__main__":
|
|
out = EXAMPLES_DIR
|
|
pure = make_pure_wheel("ecoark_demo", "0.1.0", out)
|
|
native = make_native_wheel("ecoark_native_demo", "0.1.0", out)
|
|
print(json.dumps({"pure": pure, "native": native}, indent=2))
|