generated from dopt-python/py311
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
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()
|