generated from dopt-python/py311
add hook to remove source files from wheel
This commit is contained in:
parent
3c8a9a402e
commit
cac37a3946
45
pdm_build.py
Normal file
45
pdm_build.py
Normal file
@ -0,0 +1,45 @@
|
|||||||
|
import re
|
||||||
|
import zipfile
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from Cython.Build import cythonize
|
||||||
|
|
||||||
|
ext_modules = cythonize(["src/dopt_sensor_anomalies/detection.py"])
|
||||||
|
|
||||||
|
|
||||||
|
def pdm_build_initialize(context):
|
||||||
|
context.ensure_build_dir()
|
||||||
|
|
||||||
|
|
||||||
|
def pdm_build_update_setup_kwargs(context, setup_kwargs):
|
||||||
|
setup_kwargs.update(
|
||||||
|
ext_modules=ext_modules,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def pdm_build_finalize(context, artifact):
|
||||||
|
print(">>>>>> Context: ", context)
|
||||||
|
print(">>>>>> Artifact: ", artifact)
|
||||||
|
pth_artifact = Path(artifact)
|
||||||
|
|
||||||
|
if pth_artifact.suffix == ".whl":
|
||||||
|
delete_source_files_from_wheel(pth_artifact)
|
||||||
|
|
||||||
|
|
||||||
|
def delete_source_files_from_wheel(pth_to_whl: Path):
|
||||||
|
assert pth_to_whl.exists(), "wheel file not existing"
|
||||||
|
tmp_dir = pth_to_whl.parent / "tmp"
|
||||||
|
tmp_dir.mkdir()
|
||||||
|
filename = pth_to_whl.name
|
||||||
|
tmp_whl = tmp_dir / filename
|
||||||
|
pattern = re.compile(r".*\.c$|.*detection.py$")
|
||||||
|
|
||||||
|
with zipfile.ZipFile(pth_to_whl, mode="r") as src:
|
||||||
|
with zipfile.ZipFile(tmp_whl, mode="w") as dst:
|
||||||
|
for filename in src.namelist():
|
||||||
|
if pattern.match(filename) is None:
|
||||||
|
data = src.read(filename)
|
||||||
|
dst.writestr(filename, data)
|
||||||
|
|
||||||
|
tmp_whl.replace(pth_to_whl)
|
||||||
|
tmp_dir.rmdir()
|
||||||
Loading…
x
Reference in New Issue
Block a user