PYTHON-2663 Add manylinux* platform tags to linux wheels (#30)

Author: prashantmital

.github/workflows/release-python-osx.yml

@@ -63,12 +63,12 @@ jobs:
       - uses: actions/checkout@v2
       - name: Build wheels
         run: |
-          docker run --rm --volume `pwd`:/python --workdir /python --env PYTHON_BINARY=/opt/python/${{ matrix.python-version }}/bin/python quay.io/pypa/${{ matrix.container }} ./release.sh
+          docker run --rm --volume `pwd`:/python --workdir /python --env PLAT=${{ matrix.container }} --env PYTHON_BINARY=/opt/python/${{ matrix.python-version }}/bin/python quay.io/pypa/${{ matrix.container }} ./release.sh
         working-directory: ./bindings/python
       - uses: actions/upload-artifact@v2
         with:
           name: pymongoarrow-${{ matrix.python-version }}-${{ matrix.container }}-wheel
-          path: ./bindings/python/dist/*.whl
+          path: ./bindings/python/wheelhouse/*.whl
           if-no-files-found: error
   test-manylinux-wheels:
     runs-on: ubuntu-latest

bindings/python/THIRD-PARTY-NOTICES

@@ -0,0 +1,40 @@
+PyMongoArrow uses third-party libraries or other resources that may
+be distributed under licenses different than the PyMongoArrow software.
+
+In the event that we accidentally failed to list a required notice,
+please bring it to our attention through any of the ways detailed here:
+
+           [email protected]
+
+The attached notices are provided for information only.
+
+For any licenses that require disclosure of source, sources are available at
+https://github.com/mongodb-labs/mongo-arrow/.
+
+
+1) License Notice for addtags.py
+--------------------------------
+
+The addtags.py file was adapted from auditwheel (https://github.com/pypa/auditwheel).
+
+The MIT License
+
+Copyright (c) 2016 Robert T. McGibbon <[email protected]>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

bindings/python/addtags.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python3
+# Dependencies:
+# - auditwheel>=3,<4
+# Usage:
+# $ python addtags.py WHEEL_PATH TARGET_PLATFORM WHEEL_DIR
+import os
+from os.path import abspath, isfile, exists, basename, join as pjoin
+from sys import argv
+from auditwheel.policy import get_priority_by_name, get_replace_platforms
+from auditwheel.wheel_abi import analyze_wheel_abi
+from auditwheel.wheeltools import InWheelCtx, add_platforms
+
+
+def repair_wheel(wheel_path, abi, wheel_dir):
+    wheel_fname = basename(wheel_path)
+    with InWheelCtx(wheel_path) as ctx:
+        ctx.out_wheel = pjoin(wheel_dir, wheel_fname)
+        ctx.out_wheel = add_platforms(ctx, [abi],
+                                      get_replace_platforms(abi))
+    return ctx.out_wheel
+
+
+def main(wheel_path, abi, wheel_dir):
+    if not isfile(wheel_path):
+        raise FileNotFoundError('cannot access wheel file %s' % (wheel_path,))
+
+    if not exists(wheel_dir):
+        os.makedirs(wheel_dir)
+
+    reqd_tag = get_priority_by_name(abi)
+    out_wheel = repair_wheel(wheel_path, abi, wheel_dir)
+
+    if out_wheel is not None:
+        analyzed_tag = analyze_wheel_abi(out_wheel).overall_tag
+        if reqd_tag < get_priority_by_name(analyzed_tag):
+            print('Wheel is eligible for a higher priority tag. '
+                  'You requested %s but I have found this wheel is '
+                  'eligible for %s.' % (abi, analyzed_tag))
+            out_wheel = repair_wheel(wheel_path, analyzed_tag, wheel_dir)
+
+        print('Fixed-up wheel written to %s' % (out_wheel,))
+
+
+if __name__ == '__main__':
+    WHEEL_PATH, TARGET_PLATFORM, WHEEL_DIR = argv[1], argv[2], argv[3]
+    main(
+        wheel_path=abspath(WHEEL_PATH),
+        abi=TARGET_PLATFORM,
+        wheel_dir=abspath(WHEEL_DIR))

bindings/python/release.sh

@@ -46,3 +46,10 @@ $PYTHON -c "import pyarrow; pyarrow.create_library_symlinks()"
 
 # Build wheels in $(pwd)/dist/*.whl
 LIBBSON_INSTALL_DIR="$LIBBSON_INSTALL_DIR" $PYTHON setup.py bdist_wheel
+
+# Run auditwheel repair to set platform tags on Linux
+if [ "Linux" = "$(uname -s)" ]
+then
+  $PYTHON -m pip install auditwheel
+  $PYTHON addtags.py dist/*.whl "$PLAT" ./wheelhouse
+fi