Generate whole WHEEL file

openturns · Mar 8, 2024 · db757fc · db757fc
1 parent 3edece9
commit db757fc
Show file tree

Hide file tree

Showing 5 changed files with 36 additions and 38 deletions.
diff --git a/build-wheels-linux.sh b/build-wheels-linux.sh
@@ -45,9 +45,7 @@ mkdir -p openturns.libs
 cp -v ../../../etc/openturns/openturns.conf openturns.libs
 
 # write metadata
-python ${SCRIPTPATH}/write_RECORD.py openturns ${VERSION}
-sed -i "/Tag:/d" openturns-${VERSION}.dist-info/WHEEL
-echo "Tag: ${TAG}" >> openturns-${VERSION}.dist-info/WHEEL
+python ${SCRIPTPATH}/write_distinfo.py openturns ${VERSION} ${TAG}
 
 # create archive
 zip -r openturns-${VERSION}-${TAG}.whl openturns openturns.libs openturns-${VERSION}.dist-info
@@ -91,7 +89,7 @@ do
   rm -rf ${pkgname}/__pycache__
 
   # write metadata
-  python ${SCRIPTPATH}/write_RECORD.py ${pkgname} ${pkgver}
+  python ${SCRIPTPATH}/write_distinfo.py ${pkgname} ${pkgver} ${TAG}
 
   # copy libs
   mkdir ${pkgname}.libs

diff --git a/build-wheels-macos.sh b/build-wheels-macos.sh
@@ -63,9 +63,7 @@ mkdir -p openturns.libs
 cp -v ../../../etc/openturns/openturns.conf openturns.libs
 
 # write metadata
-python${PYVER} ${SCRIPTPATH}/write_RECORD.py openturns ${VERSION}
-sed -i "" "/Tag:/d" openturns-${VERSION}.dist-info/WHEEL
-echo "Tag: ${TAG}" >> openturns-${VERSION}.dist-info/WHEEL
+python${PYVER} ${SCRIPTPATH}/write_distinfo.py openturns ${VERSION} ${TAG}
 
 # create archive
 zip -r openturns-${VERSION}-${TAG}.whl openturns openturns.libs openturns-${VERSION}.dist-info

diff --git a/build-wheels-mingw.sh b/build-wheels-mingw.sh
@@ -42,9 +42,7 @@ rm ${PREFIX}/Lib/site-packages/openturns/{libboost,python,libgraphblas}*.dll
 cd ${PREFIX}/Lib/site-packages/
 
 # write metadata
-python /io/write_RECORD.py openturns ${VERSION}
-sed -i "/Tag:/d" openturns-${VERSION}.dist-info/WHEEL
-echo "Tag: ${TAG}" >> openturns-${VERSION}.dist-info/WHEEL
+python /io/write_distinfo.py openturns ${VERSION} ${TAG}
 
 # create archive
 zip -r openturns-${VERSION}-${TAG}.whl openturns openturns-${VERSION}.dist-info

diff --git a/write_RECORD.py b/write_RECORD.py
diff --git a/write_distinfo.py b/write_distinfo.py
@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+
+import os
+import hashlib
+import base64
+
+import sys
+if len(sys.argv) != 4:
+    raise ValueError("no name/version/tag")
+name, version, tag = sys.argv[1:]
+
+path = os.path.join(f"{name}-{version}.dist-info", "RECORD")
+with open(path, "w") as record:
+    for subdir in [name, f"{name}-{version}.dist-info"]:
+        for f in os.listdir(subdir):
+            fpath = os.path.join(subdir, f)
+            if os.path.isfile(fpath):
+                if not "RECORD" in fpath:
+                    data = open(fpath, "rb").read()
+                    digest = hashlib.sha256(data).digest()
+                    checksum = base64.urlsafe_b64encode(digest).decode()
+                    size = len(data)
+                    record.write(f"{fpath},sha256={checksum},{size}\n")
+                else:
+                    record.write(fpath + ",,\n")
+
+path = os.path.join(f"{name}-{version}.dist-info", "WHEEL")
+with open(path, "w") as wheel:
+    wheel.write("Wheel-Version: 1.0\n")
+    wheel.write("Generator: custom\n")
+    wheel.write("Root-Is-Purelib: false\n")
+    wheel.write(f"Tag: {tag}\n")