recipes/recipes/build_wheels.py

# Copyright 2020 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from contextlib import contextmanager

from recipe_engine.recipe_api import Property
from recipe_engine.config import List

DEPS = [
    'depot_tools/gclient',
    'depot_tools/windows_sdk',
    'depot_tools/osx_sdk',
    'recipe_engine/context',
    'recipe_engine/file',
    'recipe_engine/path',
    'recipe_engine/platform',
    'recipe_engine/properties',
    'recipe_engine/python',
]

PROPERTIES = {
    'platforms':
        Property(
            help=('The platforms to build wheels for. If empty, builds for all '
                  'platforms which are supported on this host.'),
            kind=List(str),
            default=(),
        ),
    'dry_run':
        Property(
            help='If true, do not upload wheels or source to CIPD.',
            kind=bool,
            default=False,
        ),
}


def RunSteps(api, platforms, dry_run):
  solution_path = api.path['cache'].join('builder', 'build_wheels')
  api.file.ensure_directory("init cache if it doesn't exist", solution_path)
  with api.context(cwd=solution_path):
    api.gclient.set_config('infra')
    api.gclient.c.solutions[0].revision = 'origin/master'
    api.gclient.checkout(timeout=10 * 60)
    api.gclient.runhooks()

  temp_path = api.path.mkdtemp('.dockerbuild')

  platform_args = []
  for p in platforms:
    platform_args.extend(['--platform', p])

  # DISTUTILS_USE_SDK and MSSdk are necessary for distutils to correctly locate
  # MSVC on Windows. They do nothing on other platforms, so we just set them
  # unconditionally.
  with PlatformSdk(api), api.context(
      cwd=solution_path.join('infra'),
      env={
          'DISTUTILS_USE_SDK': '1',
          'MSSdk': '1',
      }):
    args = [
        'infra.tools.dockerbuild',
        '--root',
        temp_path,
    ]
    if not dry_run:
      args.append('--upload-sources')
    args.append('wheel-build')
    if not dry_run:
      args.append('--upload')

    api.python('dockerbuild', solution_path.join('infra', 'run.py'),
               args + platform_args)


@contextmanager
def PlatformSdk(api):
  sdk = None
  if api.platform.is_win:
    sdk = api.windows_sdk()
  elif api.platform.is_mac:
    sdk = api.osx_sdk('mac')

  if sdk is None:
    yield
  else:
    with sdk:
      yield


def GenTests(api):
  yield api.test('success')
  yield api.test('win', api.platform('win', 64))
  yield api.test('mac', api.platform(
      'mac', 64)) + api.properties(platforms=['mac-x64', 'mac-x64-cp38'])
  yield api.test('dry-run', api.properties(dry_run=True))