142 lines
4.5 KiB
Python
Executable File
142 lines
4.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# Copyright 2022 The Matrix.org Foundation C.I.C.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
# Wraps `auditwheel repair` to first check if we're repairing a potentially abi3
|
|
# compatible wheel, if so rename the wheel before repairing it.
|
|
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
from typing import Optional
|
|
from zipfile import ZipFile
|
|
|
|
from packaging.tags import Tag
|
|
from packaging.utils import parse_wheel_filename
|
|
from packaging.version import Version
|
|
|
|
|
|
def check_is_abi3_compatible(wheel_file: str) -> None:
|
|
"""Check the contents of the built wheel for any `.so` files that are *not*
|
|
abi3 compatible.
|
|
"""
|
|
|
|
with ZipFile(wheel_file, "r") as wheel:
|
|
for file in wheel.namelist():
|
|
if not file.endswith(".so"):
|
|
continue
|
|
|
|
if not file.endswith(".abi3.so"):
|
|
raise Exception(f"Found non-abi3 lib: {file}")
|
|
|
|
|
|
def cpython(wheel_file: str, name: str, version: Version, tag: Tag) -> str:
|
|
"""Replaces the cpython wheel file with a ABI3 compatible wheel"""
|
|
|
|
if tag.abi == "abi3":
|
|
# Nothing to do.
|
|
return wheel_file
|
|
|
|
check_is_abi3_compatible(wheel_file)
|
|
|
|
# HACK: it seems that some older versions of pip will consider a wheel marked
|
|
# as macosx_11_0 as incompatible with Big Sur. I haven't done the full archaeology
|
|
# here; there are some clues in
|
|
# https://github.com/pantsbuild/pants/pull/12857
|
|
# https://github.com/pypa/pip/issues/9138
|
|
# https://github.com/pypa/packaging/pull/319
|
|
# Empirically this seems to work, note that macOS 11 and 10.16 are the same,
|
|
# both versions are valid for backwards compatibility.
|
|
platform = tag.platform.replace("macosx_11_0", "macosx_10_16")
|
|
abi3_tag = Tag(tag.interpreter, "abi3", platform)
|
|
|
|
dirname = os.path.dirname(wheel_file)
|
|
new_wheel_file = os.path.join(
|
|
dirname,
|
|
f"{name}-{version}-{abi3_tag}.whl",
|
|
)
|
|
|
|
os.rename(wheel_file, new_wheel_file)
|
|
|
|
print("Renamed wheel to", new_wheel_file)
|
|
|
|
return new_wheel_file
|
|
|
|
|
|
def main(wheel_file: str, dest_dir: str, archs: Optional[str]) -> None:
|
|
"""Entry point"""
|
|
|
|
# Parse the wheel file name into its parts. Note that `parse_wheel_filename`
|
|
# normalizes the package name (i.e. it converts matrix_synapse ->
|
|
# matrix-synapse), which is not what we want.
|
|
_, version, build, tags = parse_wheel_filename(os.path.basename(wheel_file))
|
|
name = os.path.basename(wheel_file).split("-")[0]
|
|
|
|
if len(tags) != 1:
|
|
# We expect only a wheel file with only a single tag
|
|
raise Exception(f"Unexpectedly found multiple tags: {tags}")
|
|
|
|
tag = next(iter(tags))
|
|
|
|
if build:
|
|
# We don't use build tags in Synapse
|
|
raise Exception(f"Unexpected build tag: {build}")
|
|
|
|
# If the wheel is for cpython then convert it into an abi3 wheel.
|
|
if tag.interpreter.startswith("cp"):
|
|
wheel_file = cpython(wheel_file, name, version, tag)
|
|
|
|
# Finally, repair the wheel.
|
|
if archs is not None:
|
|
# If we are given archs then we are on macos and need to use
|
|
# `delocate-listdeps`.
|
|
subprocess.run(["delocate-listdeps", wheel_file], check=True)
|
|
subprocess.run(
|
|
["delocate-wheel", "--require-archs", archs, "-w", dest_dir, wheel_file],
|
|
check=True,
|
|
)
|
|
else:
|
|
subprocess.run(["auditwheel", "repair", "-w", dest_dir, wheel_file], check=True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Tag wheel as abi3 and repair it.")
|
|
|
|
parser.add_argument(
|
|
"--wheel-dir",
|
|
"-w",
|
|
metavar="WHEEL_DIR",
|
|
help="Directory to store delocated wheels",
|
|
required=True,
|
|
)
|
|
|
|
parser.add_argument(
|
|
"--require-archs",
|
|
metavar="archs",
|
|
default=None,
|
|
)
|
|
|
|
parser.add_argument(
|
|
"wheel_file",
|
|
metavar="WHEEL_FILE",
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
wheel_file = args.wheel_file
|
|
wheel_dir = args.wheel_dir
|
|
archs = args.require_archs
|
|
|
|
main(wheel_file, wheel_dir, archs)
|