Add option to build x86_64 only on macOS

Author: mayeut

cibuildwheel/macos.py

@@ -11,13 +11,13 @@
 from .util import prepare_command, get_build_verbosity_extra_flags
 
 
-def get_python_configurations(build_selector):
+def get_python_configurations(build_selector, platform='intel'):
     PythonConfiguration = namedtuple('PythonConfiguration', ['version', 'identifier', 'url'])
     python_configurations = [
-        PythonConfiguration(version='2.7', identifier='cp27-macosx_10_6_intel', url='https://www.python.org/ftp/python/2.7.17/python-2.7.17-macosx10.6.pkg'),
-        PythonConfiguration(version='3.5', identifier='cp35-macosx_10_6_intel', url='https://www.python.org/ftp/python/3.5.4/python-3.5.4-macosx10.6.pkg'),
-        PythonConfiguration(version='3.6', identifier='cp36-macosx_10_6_intel', url='https://www.python.org/ftp/python/3.6.8/python-3.6.8-macosx10.6.pkg'),
-        PythonConfiguration(version='3.7', identifier='cp37-macosx_10_6_intel', url='https://www.python.org/ftp/python/3.7.5/python-3.7.5-macosx10.6.pkg'),
+        PythonConfiguration(version='2.7', identifier='cp27-macosx_10_6_' + platform, url='https://www.python.org/ftp/python/2.7.17/python-2.7.17-macosx10.6.pkg'),
+        PythonConfiguration(version='3.5', identifier='cp35-macosx_10_6_' + platform, url='https://www.python.org/ftp/python/3.5.4/python-3.5.4-macosx10.6.pkg'),
+        PythonConfiguration(version='3.6', identifier='cp36-macosx_10_6_' + platform, url='https://www.python.org/ftp/python/3.6.8/python-3.6.8-macosx10.6.pkg'),
+        PythonConfiguration(version='3.7', identifier='cp37-macosx_10_6_' + platform, url='https://www.python.org/ftp/python/3.7.5/python-3.7.5-macosx10.6.pkg'),
         PythonConfiguration(version='3.8', identifier='cp38-macosx_10_9_x86_64', url='https://www.python.org/ftp/python/3.8.0/python-3.8.0-macosx10.9.pkg'),
     ]
 
@@ -26,7 +26,9 @@ def get_python_configurations(build_selector):
 
 
 def build(project_dir, output_dir, test_command, test_requires, test_extras, before_build, build_verbosity, build_selector, environment):
-    python_configurations = get_python_configurations(build_selector)
+    platform = os.environ.get('CIBW_MACOS_PLATFORM', 'intel')
+    assert platform in ['intel', 'x86_64']
+    python_configurations = get_python_configurations(build_selector, platform)
     get_pip_url = 'https://bootstrap.pypa.io/get-pip.py'
     get_pip_script = '/tmp/get-pip.py'
 
@@ -102,6 +104,15 @@ def call(args, env=None, cwd=None, shell=False):
             shutil.rmtree('/tmp/delocated_wheel')
         os.makedirs('/tmp/delocated_wheel')
 
+        # setup target platform
+        current_platform = subprocess.check_output(['python', '-c', 'import distutils.util; print(distutils.util.get_platform())'], env=env)
+        if sys.version_info[0] >= 3:
+            current_platform = current_platform.decode('utf8')
+        current_platform = current_platform.strip()
+        if platform == 'x86_64' and current_platform.endswith('intel'):
+            env['_PYTHON_HOST_PLATFORM'] = current_platform.replace('intel', 'x86_64')  # cross-compilation platform override
+            env['ARCHFLAGS'] = '-arch x86_64'  # https://github.com/python/cpython/blob/a5ed2fe0eedefa1649aa93ee74a0bafc8e628a10/Lib/_osx_support.py#L260
+
         # run the before_build command
         if before_build:
             before_build_prepared = prepare_command(before_build, project=abs_project_dir)

test/09_macos_x86_64/cibuildwheel_test.py

@@ -0,0 +1,19 @@
+import os, pytest
+import utils
+
+
+def test():
+    project_dir = os.path.dirname(__file__)
+
+    if utils.platform != 'macos':
+        pytest.skip('the test is only relevant to the macOS build')
+
+    # build the wheels
+    utils.cibuildwheel_run(project_dir, add_env={
+        'CIBW_MACOS_PLATFORM': 'x86_64',
+    })
+
+    # also check that we got the right wheels
+    expected_wheels = [w.replace('intel', 'x86_64') for w in utils.expected_wheels('spam', '0.1.0')]
+    actual_wheels = os.listdir('wheelhouse')
+    assert set(actual_wheels) == set(expected_wheels)

test/09_macos_x86_64/setup.py

@@ -0,0 +1,12 @@
+import os
+
+from setuptools import setup, Extension
+
+if os.environ.get('CIBUILDWHEEL', '0') != '1':
+    raise Exception('CIBUILDWHEEL environment variable is not set to 1')
+
+setup(
+    name="spam",
+    ext_modules=[Extension('spam', sources=['spam.c'])],
+    version="0.1.0",
+)

test/09_macos_x86_64/spam.c

@@ -0,0 +1,48 @@
+#include <Python.h>
+
+static PyObject *
+spam_system(PyObject *self, PyObject *args)
+{
+    const char *command;
+    int sts;
+
+    if (!PyArg_ParseTuple(args, "s", &command))
+        return NULL;
+    sts = system(command);
+    return PyLong_FromLong(sts);
+}
+
+/* Module initialization */
+
+#if PY_MAJOR_VERSION >= 3
+    #define MOD_INIT(name) PyMODINIT_FUNC PyInit_##name(void)
+    #define MOD_DEF(m, name, doc, methods, module_state_size) \
+        static struct PyModuleDef moduledef = { \
+            PyModuleDef_HEAD_INIT, name, doc, module_state_size, methods, }; \
+        m = PyModule_Create(&moduledef);
+    #define MOD_RETURN(m) return m;
+#else
+    #define MOD_INIT(name) PyMODINIT_FUNC init##name(void)
+    #define MOD_DEF(m, name, doc, methods, module_state_size) \
+        m = Py_InitModule3(name, methods, doc);
+    #define MOD_RETURN(m) return;
+#endif
+
+static PyMethodDef module_methods[] = {
+    {"system", (PyCFunction)spam_system, METH_VARARGS, 
+     "Execute a shell command."},
+    {NULL}  /* Sentinel */
+};
+
+MOD_INIT(spam)
+{
+    PyObject* m;
+
+    MOD_DEF(m, 
+            "spam", 
+            "Example module", 
+            module_methods,
+            -1)
+
+    MOD_RETURN(m)
+}